ultralytics-thop 0.0.2__tar.gz → 0.0.3__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.
Files changed (27) hide show
  1. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/PKG-INFO +10 -9
  2. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/README.md +8 -8
  3. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/pyproject.toml +2 -1
  4. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/profile.py +3 -3
  5. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/ultralytics_thop.egg-info/PKG-INFO +10 -9
  6. ultralytics_thop-0.0.3/ultralytics_thop.egg-info/requires.txt +2 -0
  7. ultralytics_thop-0.0.2/ultralytics_thop.egg-info/requires.txt +0 -1
  8. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/LICENSE +0 -0
  9. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/setup.cfg +0 -0
  10. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/tests/test_conv2d.py +0 -0
  11. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/tests/test_matmul.py +0 -0
  12. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/tests/test_relu.py +0 -0
  13. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/tests/test_utils.py +0 -0
  14. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/__init__.py +0 -0
  15. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/__version__.py +0 -0
  16. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/fx_profile.py +0 -0
  17. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/onnx_profile.py +0 -0
  18. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/rnn_hooks.py +0 -0
  19. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/utils.py +0 -0
  20. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/vision/__init__.py +0 -0
  21. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/vision/basic_hooks.py +0 -0
  22. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/vision/calc_func.py +0 -0
  23. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/vision/efficientnet.py +0 -0
  24. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/vision/onnx_counter.py +0 -0
  25. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/ultralytics_thop.egg-info/SOURCES.txt +0 -0
  26. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/ultralytics_thop.egg-info/dependency_links.txt +0 -0
  27. {ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/ultralytics_thop.egg-info/top_level.txt +0 -0
{ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ultralytics-thop
3
- Version: 0.0.2
3
+ Version: 0.0.3
4
4
  Summary: A tool to count the FLOPs of PyTorch model.
5
5
  Author-email: Ligeng Zhu <ligeng.zhu+github@gmail.com>
6
6
  Maintainer-email: Ligeng Zhu <ligeng.zhu+github@gmail.com>
@@ -688,6 +688,7 @@ Classifier: Operating System :: Microsoft :: Windows
688
688
  Requires-Python: >=3.8
689
689
  Description-Content-Type: text/markdown
690
690
  License-File: LICENSE
691
+ Requires-Dist: packaging
691
692
  Requires-Dist: torch
692
693
 
693
694
  <br>
@@ -697,7 +698,7 @@ Requires-Dist: torch
697
698
 
698
699
  Welcome to the [THOP](https://github.com/ultralytics/thop) repository, your comprehensive solution for profiling PyTorch models by computing the number of Multiply-Accumulate Operations (MACs) and parameters. This tool is essential for deep learning practitioners to evaluate model efficiency and performance.
699
700
 
700
- [![GitHub Actions](https://github.com/ultralytics/thop/actions/workflows/format.yml/badge.svg)](https://github.com/ultralytics/thop/actions/workflows/main.yml) [![PyPI version](https://badge.fury.io/py/ultralytics-.svg)](https://badge.fury.io/py/ultralytics-thop) <a href="https://ultralytics.com/discord"><img alt="Discord" src="https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue"></a>
701
+ [![GitHub Actions](https://github.com/ultralytics/thop/actions/workflows/format.yml/badge.svg)](https://github.com/ultralytics/thop/actions/workflows/main.yml) [![PyPI version](https://badge.fury.io/py/ultralytics-thop.svg)](https://badge.fury.io/py/ultralytics-thop) <a href="https://ultralytics.com/discord"><img alt="Discord" src="https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue"></a>
701
702
 
702
703
  ## 📄 Description
703
704
 
@@ -830,17 +831,17 @@ For bugs or feature requests, please open an issue on [GitHub Issues](https://gi
830
831
 
831
832
  <br>
832
833
  <div align="center">
833
- <a href="https://github.com/ultralytics/thop"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-github.png" width="3%" alt="THOP GitHub"></a>
834
+ <a href="https://github.com/ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-github.png" width="3%" alt="Ultralytics GitHub"></a>
834
835
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
835
- <a href="https://www.linkedin.com/company/pytorch/"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-linkedin.png" width="3%" alt="PyTorch LinkedIn"></a>
836
+ <a href="https://www.linkedin.com/company/ultralytics/"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-linkedin.png" width="3%" alt="Ultralytics LinkedIn"></a>
836
837
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
837
- <a href="https://twitter.com/pytorch"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-twitter.png" width="3%" alt="PyTorch Twitter"></a>
838
+ <a href="https://twitter.com/ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-twitter.png" width="3%" alt="Ultralytics Twitter"></a>
838
839
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
839
- <a href="https://youtube.com/pytorch?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="3%" alt="PyTorch YouTube"></a>
840
+ <a href="https://youtube.com/ultralytics?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="3%" alt="Ultralytics YouTube"></a>
840
841
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
841
- <a href="https://www.tiktok.com/@pytorch"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-tiktok.png" width="3%" alt="PyTorch TikTok"></a>
842
+ <a href="https://www.tiktok.com/@ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-tiktok.png" width="3%" alt="Ultralytics TikTok"></a>
842
843
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
843
- <a href="https://www.instagram.com/pytorch/"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-instagram.png" width="3%" alt="PyTorch Instagram"></a>
844
+ <a href="https://www.instagram.com/ultralytics/"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-instagram.png" width="3%" alt="Ultralytics Instagram"></a>
844
845
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
845
- <a href="https://discord.com/invite/pytorch"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-discord.png" width="3%" alt="PyTorch Discord"></a>
846
+ <a href="https://ultralytics.com/discord"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-discord.png" width="3%" alt="Ultralytics Discord"></a>
846
847
  </div>
{ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/README.md RENAMED
@@ -5,7 +5,7 @@
5
5
 
6
6
  Welcome to the [THOP](https://github.com/ultralytics/thop) repository, your comprehensive solution for profiling PyTorch models by computing the number of Multiply-Accumulate Operations (MACs) and parameters. This tool is essential for deep learning practitioners to evaluate model efficiency and performance.
7
7
 
8
- [![GitHub Actions](https://github.com/ultralytics/thop/actions/workflows/format.yml/badge.svg)](https://github.com/ultralytics/thop/actions/workflows/main.yml) [![PyPI version](https://badge.fury.io/py/ultralytics-.svg)](https://badge.fury.io/py/ultralytics-thop) <a href="https://ultralytics.com/discord"><img alt="Discord" src="https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue"></a>
8
+ [![GitHub Actions](https://github.com/ultralytics/thop/actions/workflows/format.yml/badge.svg)](https://github.com/ultralytics/thop/actions/workflows/main.yml) [![PyPI version](https://badge.fury.io/py/ultralytics-thop.svg)](https://badge.fury.io/py/ultralytics-thop) <a href="https://ultralytics.com/discord"><img alt="Discord" src="https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue"></a>
9
9
 
10
10
  ## 📄 Description
11
11
 
@@ -138,17 +138,17 @@ For bugs or feature requests, please open an issue on [GitHub Issues](https://gi
138
138
 
139
139
  <br>
140
140
  <div align="center">
141
- <a href="https://github.com/ultralytics/thop"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-github.png" width="3%" alt="THOP GitHub"></a>
141
+ <a href="https://github.com/ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-github.png" width="3%" alt="Ultralytics GitHub"></a>
142
142
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
143
- <a href="https://www.linkedin.com/company/pytorch/"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-linkedin.png" width="3%" alt="PyTorch LinkedIn"></a>
143
+ <a href="https://www.linkedin.com/company/ultralytics/"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-linkedin.png" width="3%" alt="Ultralytics LinkedIn"></a>
144
144
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
145
- <a href="https://twitter.com/pytorch"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-twitter.png" width="3%" alt="PyTorch Twitter"></a>
145
+ <a href="https://twitter.com/ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-twitter.png" width="3%" alt="Ultralytics Twitter"></a>
146
146
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
147
- <a href="https://youtube.com/pytorch?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="3%" alt="PyTorch YouTube"></a>
147
+ <a href="https://youtube.com/ultralytics?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="3%" alt="Ultralytics YouTube"></a>
148
148
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
149
- <a href="https://www.tiktok.com/@pytorch"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-tiktok.png" width="3%" alt="PyTorch TikTok"></a>
149
+ <a href="https://www.tiktok.com/@ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-tiktok.png" width="3%" alt="Ultralytics TikTok"></a>
150
150
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
151
- <a href="https://www.instagram.com/pytorch/"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-instagram.png" width="3%" alt="PyTorch Instagram"></a>
151
+ <a href="https://www.instagram.com/ultralytics/"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-instagram.png" width="3%" alt="Ultralytics Instagram"></a>
152
152
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
153
- <a href="https://discord.com/invite/pytorch"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-discord.png" width="3%" alt="PyTorch Discord"></a>
153
+ <a href="https://ultralytics.com/discord"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-discord.png" width="3%" alt="Ultralytics Discord"></a>
154
154
  </div>
{ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/pyproject.toml RENAMED
@@ -25,7 +25,7 @@ build-backend = "setuptools.build_meta"
25
25
 
26
26
  [project]
27
27
  name = "ultralytics-thop"
28
- version = "0.0.2" # Placeholder version, needs to be dynamically set
28
+ version = "0.0.3" # Placeholder version, needs to be dynamically set
29
29
  description = "A tool to count the FLOPs of PyTorch model."
30
30
  readme = "README.md"
31
31
  requires-python = ">=3.8"
@@ -57,6 +57,7 @@ classifiers = [
57
57
  "Operating System :: Microsoft :: Windows",
58
58
  ]
59
59
  dependencies = [
60
+ "packaging",
60
61
  "torch",
61
62
  ]
62
63
 
{ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/thop/profile.py RENAMED
@@ -1,4 +1,4 @@
1
- from distutils.version import LooseVersion
1
+ from packaging.version import Version
2
2
 
3
3
  from thop.rnn_hooks import *
4
4
  from thop.vision.basic_hooks import *
@@ -7,7 +7,7 @@ from thop.vision.basic_hooks import *
7
7
  # logger.setLevel(logging.INFO)
8
8
  from .utils import prGreen, prRed, prYellow
9
9
 
10
- if LooseVersion(torch.__version__) < LooseVersion("1.0.0"):
10
+ if Version(torch.__version__) < Version("1.0.0"):
11
11
  logging.warning(
12
12
  "You are using an old version PyTorch {version}, which THOP does NOT support.".format(version=torch.__version__)
13
13
  )
@@ -61,7 +61,7 @@ register_hooks = {
61
61
  nn.PixelShuffle: zero_ops,
62
62
  }
63
63
 
64
- if LooseVersion(torch.__version__) >= LooseVersion("1.1.0"):
64
+ if Version(torch.__version__) >= Version("1.1.0"):
65
65
  register_hooks.update({nn.SyncBatchNorm: count_normalization})
66
66
 
67
67
 
{ultralytics_thop-0.0.2 → ultralytics_thop-0.0.3}/ultralytics_thop.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ultralytics-thop
3
- Version: 0.0.2
3
+ Version: 0.0.3
4
4
  Summary: A tool to count the FLOPs of PyTorch model.
5
5
  Author-email: Ligeng Zhu <ligeng.zhu+github@gmail.com>
6
6
  Maintainer-email: Ligeng Zhu <ligeng.zhu+github@gmail.com>
@@ -688,6 +688,7 @@ Classifier: Operating System :: Microsoft :: Windows
688
688
  Requires-Python: >=3.8
689
689
  Description-Content-Type: text/markdown
690
690
  License-File: LICENSE
691
+ Requires-Dist: packaging
691
692
  Requires-Dist: torch
692
693
 
693
694
  <br>
@@ -697,7 +698,7 @@ Requires-Dist: torch
697
698
 
698
699
  Welcome to the [THOP](https://github.com/ultralytics/thop) repository, your comprehensive solution for profiling PyTorch models by computing the number of Multiply-Accumulate Operations (MACs) and parameters. This tool is essential for deep learning practitioners to evaluate model efficiency and performance.
699
700
 
700
- [![GitHub Actions](https://github.com/ultralytics/thop/actions/workflows/format.yml/badge.svg)](https://github.com/ultralytics/thop/actions/workflows/main.yml) [![PyPI version](https://badge.fury.io/py/ultralytics-.svg)](https://badge.fury.io/py/ultralytics-thop) <a href="https://ultralytics.com/discord"><img alt="Discord" src="https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue"></a>
701
+ [![GitHub Actions](https://github.com/ultralytics/thop/actions/workflows/format.yml/badge.svg)](https://github.com/ultralytics/thop/actions/workflows/main.yml) [![PyPI version](https://badge.fury.io/py/ultralytics-thop.svg)](https://badge.fury.io/py/ultralytics-thop) <a href="https://ultralytics.com/discord"><img alt="Discord" src="https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue"></a>
701
702
 
702
703
  ## 📄 Description
703
704
 
@@ -830,17 +831,17 @@ For bugs or feature requests, please open an issue on [GitHub Issues](https://gi
830
831
 
831
832
  <br>
832
833
  <div align="center">
833
- <a href="https://github.com/ultralytics/thop"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-github.png" width="3%" alt="THOP GitHub"></a>
834
+ <a href="https://github.com/ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-github.png" width="3%" alt="Ultralytics GitHub"></a>
834
835
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
835
- <a href="https://www.linkedin.com/company/pytorch/"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-linkedin.png" width="3%" alt="PyTorch LinkedIn"></a>
836
+ <a href="https://www.linkedin.com/company/ultralytics/"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-linkedin.png" width="3%" alt="Ultralytics LinkedIn"></a>
836
837
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
837
- <a href="https://twitter.com/pytorch"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-twitter.png" width="3%" alt="PyTorch Twitter"></a>
838
+ <a href="https://twitter.com/ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-twitter.png" width="3%" alt="Ultralytics Twitter"></a>
838
839
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
839
- <a href="https://youtube.com/pytorch?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="3%" alt="PyTorch YouTube"></a>
840
+ <a href="https://youtube.com/ultralytics?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="3%" alt="Ultralytics YouTube"></a>
840
841
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
841
- <a href="https://www.tiktok.com/@pytorch"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-tiktok.png" width="3%" alt="PyTorch TikTok"></a>
842
+ <a href="https://www.tiktok.com/@ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-tiktok.png" width="3%" alt="Ultralytics TikTok"></a>
842
843
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
843
- <a href="https://www.instagram.com/pytorch/"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-instagram.png" width="3%" alt="PyTorch Instagram"></a>
844
+ <a href="https://www.instagram.com/ultralytics/"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-instagram.png" width="3%" alt="Ultralytics Instagram"></a>
844
845
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
845
- <a href="https://discord.com/invite/pytorch"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-discord.png" width="3%" alt="PyTorch Discord"></a>
846
+ <a href="https://ultralytics.com/discord"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-discord.png" width="3%" alt="Ultralytics Discord"></a>
846
847
  </div>
ultralytics_thop-0.0.3/ultralytics_thop.egg-info/requires.txt ADDED
@@ -0,0 +1,2 @@
1
+ packaging
2
+ torch