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.
@@ -1,5 +1,5 @@
1
1
  MIT License
2
2
 
3
- Copyright (c) 2024 <Your Name>
3
+ Copyright (c) 2024 HARSHAL VILAS KALE
4
4
 
5
5
  Permission is hereby granted, free of charge, to any person obtaining a copy...
@@ -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
+ ![PyPI](https://img.shields.io/pypi/v/torch-anatomy)
28
+ ![License](https://img.shields.io/github/license/harsk03/torch-anatomy)
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
  ![PyPI](https://img.shields.io/pypi/v/torch-anatomy)
6
- ![License](https://img.shields.io/github/license/yourusername/torch-anatomy)
6
+ ![License](https://img.shields.io/github/license/harsk03/torch-anatomy)
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.0',
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
+ ![PyPI](https://img.shields.io/pypi/v/torch-anatomy)
28
+ ![License](https://img.shields.io/github/license/harsk03/torch-anatomy)
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
@@ -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