torch-anatomy 0.1.0__tar.gz → 0.1.1__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/LICENSE +1 -1
- torch_anatomy-0.1.1/PKG-INFO +65 -0
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/README.md +1 -1
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/setup.py +6 -1
- torch_anatomy-0.1.1/torch_anatomy.egg-info/PKG-INFO +65 -0
- torch_anatomy-0.1.0/PKG-INFO +0 -18
- torch_anatomy-0.1.0/torch_anatomy.egg-info/PKG-INFO +0 -18
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/pyproject.toml +0 -0
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/setup.cfg +0 -0
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/tests/test_visualizer.py +0 -0
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/torch_anatomy/__init__.py +0 -0
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/torch_anatomy/cli.py +0 -0
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/torch_anatomy/utils.py +0 -0
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/torch_anatomy/visualizer.py +0 -0
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/torch_anatomy.egg-info/SOURCES.txt +0 -0
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/torch_anatomy.egg-info/dependency_links.txt +0 -0
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/torch_anatomy.egg-info/entry_points.txt +0 -0
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/torch_anatomy.egg-info/requires.txt +0 -0
- {torch_anatomy-0.1.0 → torch_anatomy-0.1.1}/torch_anatomy.egg-info/top_level.txt +0 -0
@@ -0,0 +1,65 @@
|
|
1
|
+
Metadata-Version: 2.4
|
2
|
+
Name: torch-anatomy
|
3
|
+
Version: 0.1.1
|
4
|
+
Summary: Layer-by-layer visualizer for PyTorch models
|
5
|
+
Author: Harshal Vilas Kale
|
6
|
+
License: MIT
|
7
|
+
Description-Content-Type: text/markdown
|
8
|
+
License-File: LICENSE
|
9
|
+
Requires-Dist: torch
|
10
|
+
Requires-Dist: torchvision
|
11
|
+
Requires-Dist: matplotlib
|
12
|
+
Requires-Dist: numpy
|
13
|
+
Requires-Dist: Pillow
|
14
|
+
Requires-Dist: click
|
15
|
+
Dynamic: author
|
16
|
+
Dynamic: description
|
17
|
+
Dynamic: description-content-type
|
18
|
+
Dynamic: license
|
19
|
+
Dynamic: license-file
|
20
|
+
Dynamic: requires-dist
|
21
|
+
Dynamic: summary
|
22
|
+
|
23
|
+
# torch-anatomy
|
24
|
+
|
25
|
+
**Layer-by-layer visualizer for PyTorch models — Understand what each layer actually does.**
|
26
|
+
|
27
|
+

|
28
|
+

|
29
|
+
|
30
|
+
## Install
|
31
|
+
|
32
|
+
```bash
|
33
|
+
pip install torch-anatomy
|
34
|
+
```
|
35
|
+
|
36
|
+
## Usage
|
37
|
+
|
38
|
+
```python
|
39
|
+
from torch_anatomy import visualize_layers
|
40
|
+
from torchvision import models
|
41
|
+
|
42
|
+
model = models.resnet18(pretrained=True)
|
43
|
+
visualize_layers(
|
44
|
+
model=model,
|
45
|
+
input_image='dog.jpg',
|
46
|
+
layers_to_show=['Conv2d', 'ReLU'],
|
47
|
+
channels_per_layer=6,
|
48
|
+
colormap='inferno',
|
49
|
+
show_colorbar=True
|
50
|
+
)
|
51
|
+
```
|
52
|
+
|
53
|
+
Or from CLI:
|
54
|
+
|
55
|
+
```bash
|
56
|
+
torch-anatomy --model resnet18 --image dog.jpg
|
57
|
+
```
|
58
|
+
|
59
|
+
## Features
|
60
|
+
- Plug-and-play for any PyTorch CNN
|
61
|
+
- Visualizes feature maps for key layers
|
62
|
+
- Customizable channels, colormap, and more
|
63
|
+
|
64
|
+
## License
|
65
|
+
MIT
|
@@ -3,7 +3,7 @@
|
|
3
3
|
**Layer-by-layer visualizer for PyTorch models — Understand what each layer actually does.**
|
4
4
|
|
5
5
|

|
6
|
-

|
7
7
|
|
8
8
|
## Install
|
9
9
|
|
@@ -1,9 +1,14 @@
|
|
1
1
|
from setuptools import setup, find_packages
|
2
2
|
|
3
|
+
with open("README.md", "r", encoding="utf-8") as fh:
|
4
|
+
long_description = fh.read()
|
5
|
+
|
3
6
|
setup(
|
4
7
|
name='torch-anatomy',
|
5
|
-
version='0.1.
|
8
|
+
version='0.1.1',
|
6
9
|
description='Layer-by-layer visualizer for PyTorch models',
|
10
|
+
long_description=long_description,
|
11
|
+
long_description_content_type="text/markdown",
|
7
12
|
author='Harshal Vilas Kale',
|
8
13
|
packages=find_packages(),
|
9
14
|
install_requires=[
|
@@ -0,0 +1,65 @@
|
|
1
|
+
Metadata-Version: 2.4
|
2
|
+
Name: torch-anatomy
|
3
|
+
Version: 0.1.1
|
4
|
+
Summary: Layer-by-layer visualizer for PyTorch models
|
5
|
+
Author: Harshal Vilas Kale
|
6
|
+
License: MIT
|
7
|
+
Description-Content-Type: text/markdown
|
8
|
+
License-File: LICENSE
|
9
|
+
Requires-Dist: torch
|
10
|
+
Requires-Dist: torchvision
|
11
|
+
Requires-Dist: matplotlib
|
12
|
+
Requires-Dist: numpy
|
13
|
+
Requires-Dist: Pillow
|
14
|
+
Requires-Dist: click
|
15
|
+
Dynamic: author
|
16
|
+
Dynamic: description
|
17
|
+
Dynamic: description-content-type
|
18
|
+
Dynamic: license
|
19
|
+
Dynamic: license-file
|
20
|
+
Dynamic: requires-dist
|
21
|
+
Dynamic: summary
|
22
|
+
|
23
|
+
# torch-anatomy
|
24
|
+
|
25
|
+
**Layer-by-layer visualizer for PyTorch models — Understand what each layer actually does.**
|
26
|
+
|
27
|
+

|
28
|
+

|
29
|
+
|
30
|
+
## Install
|
31
|
+
|
32
|
+
```bash
|
33
|
+
pip install torch-anatomy
|
34
|
+
```
|
35
|
+
|
36
|
+
## Usage
|
37
|
+
|
38
|
+
```python
|
39
|
+
from torch_anatomy import visualize_layers
|
40
|
+
from torchvision import models
|
41
|
+
|
42
|
+
model = models.resnet18(pretrained=True)
|
43
|
+
visualize_layers(
|
44
|
+
model=model,
|
45
|
+
input_image='dog.jpg',
|
46
|
+
layers_to_show=['Conv2d', 'ReLU'],
|
47
|
+
channels_per_layer=6,
|
48
|
+
colormap='inferno',
|
49
|
+
show_colorbar=True
|
50
|
+
)
|
51
|
+
```
|
52
|
+
|
53
|
+
Or from CLI:
|
54
|
+
|
55
|
+
```bash
|
56
|
+
torch-anatomy --model resnet18 --image dog.jpg
|
57
|
+
```
|
58
|
+
|
59
|
+
## Features
|
60
|
+
- Plug-and-play for any PyTorch CNN
|
61
|
+
- Visualizes feature maps for key layers
|
62
|
+
- Customizable channels, colormap, and more
|
63
|
+
|
64
|
+
## License
|
65
|
+
MIT
|
torch_anatomy-0.1.0/PKG-INFO
DELETED
@@ -1,18 +0,0 @@
|
|
1
|
-
Metadata-Version: 2.4
|
2
|
-
Name: torch-anatomy
|
3
|
-
Version: 0.1.0
|
4
|
-
Summary: Layer-by-layer visualizer for PyTorch models
|
5
|
-
Author: Harshal Vilas Kale
|
6
|
-
License: MIT
|
7
|
-
License-File: LICENSE
|
8
|
-
Requires-Dist: torch
|
9
|
-
Requires-Dist: torchvision
|
10
|
-
Requires-Dist: matplotlib
|
11
|
-
Requires-Dist: numpy
|
12
|
-
Requires-Dist: Pillow
|
13
|
-
Requires-Dist: click
|
14
|
-
Dynamic: author
|
15
|
-
Dynamic: license
|
16
|
-
Dynamic: license-file
|
17
|
-
Dynamic: requires-dist
|
18
|
-
Dynamic: summary
|
@@ -1,18 +0,0 @@
|
|
1
|
-
Metadata-Version: 2.4
|
2
|
-
Name: torch-anatomy
|
3
|
-
Version: 0.1.0
|
4
|
-
Summary: Layer-by-layer visualizer for PyTorch models
|
5
|
-
Author: Harshal Vilas Kale
|
6
|
-
License: MIT
|
7
|
-
License-File: LICENSE
|
8
|
-
Requires-Dist: torch
|
9
|
-
Requires-Dist: torchvision
|
10
|
-
Requires-Dist: matplotlib
|
11
|
-
Requires-Dist: numpy
|
12
|
-
Requires-Dist: Pillow
|
13
|
-
Requires-Dist: click
|
14
|
-
Dynamic: author
|
15
|
-
Dynamic: license
|
16
|
-
Dynamic: license-file
|
17
|
-
Dynamic: requires-dist
|
18
|
-
Dynamic: summary
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|