ultralytics-thop 2.0.16__py3-none-any.whl → 2.0.18__py3-none-any.whl

thop/__init__.py

@@ -1,7 +1,6 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
-__version__ = "2.0.16"
-
+__version__ = "2.0.18"
 
 import torch
 
@@ -9,3 +8,5 @@ from .profile import profile, profile_origin
 from .utils import clever_format
 
 default_dtype = torch.float64
+
+__all__ = ["clever_format", "default_dtype", "profile", "profile_origin"]

thop/fx_profile.py

@@ -6,6 +6,11 @@ from distutils.version import LooseVersion
 import torch
 import torch as th
 import torch.nn as nn
+from torch.fx import symbolic_trace
+from torch.fx.passes.shape_prop import ShapeProp
+
+from .utils import prRed, prYellow
+from .vision.calc_func import calculate_conv
 
 if LooseVersion(torch.__version__) < LooseVersion("1.8.0"):
     logging.warning(
@@ -40,18 +45,15 @@ def count_fn_linear(input_shapes, output_shapes, *args, **kwargs):
     return flops
 
 
-from .vision.calc_func import calculate_conv
-
-
 def count_fn_conv2d(input_shapes, output_shapes, *args, **kwargs):
     """Calculates total operations (FLOPs) for a 2D conv layer based on input and output shapes using
     `calculate_conv`.
     """
-    inputs, weight, bias, stride, padding, dilation, groups = args
+    _inputs, _weight, _bias, _stride, _padding, _dilation, groups = args
     if len(input_shapes) == 2:
         x_shape, k_shape = input_shapes
     elif len(input_shapes) == 3:
-        x_shape, k_shape, b_shape = input_shapes
+        x_shape, k_shape, _b_shape = input_shapes
     out_shape = output_shapes[0]
 
     kernel_parameters = k_shape[2:].numel()
@@ -118,11 +120,6 @@ for k in zero_ops:
 
 missing_maps = {}
 
-from torch.fx import symbolic_trace
-from torch.fx.passes.shape_prop import ShapeProp
-
-from .utils import prRed, prYellow
-
 
 def null_print(*args, **kwargs):
     """A no-op print function that takes any arguments without performing any actions."""
@@ -211,11 +208,15 @@ def fx_profile(mod: nn.Module, input: th.Tensor, verbose=False):
 if __name__ == "__main__":
 
     class MyOP(nn.Module):
+        """Custom operator that performs a simple forward pass dividing input by 1."""
+
         def forward(self, input):
             """Performs forward pass on given input data."""
             return input / 1
 
     class MyModule(torch.nn.Module):
+        """Neural network module with two linear layers and a custom MyOP operator."""
+
         def __init__(self):
             """Initializes MyModule with two linear layers and a custom MyOP operator."""
             super().__init__()

thop/profile.py

@@ -1,7 +1,30 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
-from thop.rnn_hooks import *
-from thop.vision.basic_hooks import *
+from thop.rnn_hooks import (
+    count_gru,
+    count_gru_cell,
+    count_lstm,
+    count_lstm_cell,
+    count_rnn,
+    count_rnn_cell,
+    torch,
+)
+from thop.vision.basic_hooks import (
+    count_adap_avgpool,
+    count_avgpool,
+    count_convNd,
+    count_convtNd,
+    count_linear,
+    count_normalization,
+    count_parameters,
+    count_prelu,
+    count_relu,
+    count_softmax,
+    count_upsample,
+    logging,
+    nn,
+    zero_ops,
+)
 
 from .utils import prRed
 
@@ -71,7 +94,7 @@ def profile_origin(model, inputs, custom_ops=None, verbose=True, report_missing=
 
         if hasattr(m, "total_ops") or hasattr(m, "total_params"):
             logging.warning(
-                f"Either .total_ops or .total_params is already defined in {str(m)}. "
+                f"Either .total_ops or .total_params is already defined in {m!s}. "
                 "Be careful, it might change your code's behavior."
             )
 

thop/vision/basic_hooks.py

@@ -2,10 +2,24 @@
 
 import logging
 
+import torch
 import torch.nn as nn
 from torch.nn.modules.conv import _ConvNd
 
-from .calc_func import *
+from thop.vision.calc_func import (
+    calculate_adaptive_avg,
+    calculate_avgpool,
+    calculate_conv,
+    calculate_conv2d_flops,
+    calculate_linear,
+    calculate_norm,
+    calculate_parameters,
+    calculate_relu,
+    calculate_relu_flops,
+    calculate_softmax,
+    calculate_upsample,
+    calculate_zero_ops,
+)
 
 multiply_adds = 1
 

{ultralytics_thop-2.0.16.dist-info → ultralytics_thop-2.0.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-thop
-Version: 2.0.16
+Version: 2.0.18
 Summary: Ultralytics THOP package for fast computation of PyTorch model FLOPs and parameters.
 Author-email: Ligeng Zhu <ligeng.zhu+github@gmail.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>
@@ -56,7 +56,7 @@ THOP offers an intuitive API designed to profile PyTorch models by calculating t
 
 Get started with THOP quickly by installing it via pip:
 
-[![PyPI - Version](https://img.shields.io/pypi/v/ultralytics-thop?logo=pypi&logoColor=white)](https://pypi.org/project/ultralytics-thop/) [![Downloads](https://static.pepy.tech/badge/ultralytics-thop)](https://www.pepy.tech/projects/ultralytics-thop) [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/ultralytics-thop?logo=python&logoColor=gold)](https://pypi.org/project/ultralytics-thop/)
+[![PyPI - Version](https://img.shields.io/pypi/v/ultralytics-thop?logo=pypi&logoColor=white)](https://pypi.org/project/ultralytics-thop/) [![Downloads](https://static.pepy.tech/badge/ultralytics-thop)](https://clickpy.clickhouse.com/dashboard/ultralytics-thop) [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/ultralytics-thop?logo=python&logoColor=gold)](https://pypi.org/project/ultralytics-thop/)
 
 ```bash
 pip install ultralytics-thop
@@ -74,7 +74,7 @@ This ensures you have the most recent version, incorporating the latest improvem
 
 ### Basic Usage
 
-Profiling a standard PyTorch model like [ResNet50](https://pytorch.org/vision/main/models/generated/torchvision.models.resnet50.html) is straightforward. Import the necessary libraries, load your model and a sample input tensor, then use the `profile` function:
+Profiling a standard PyTorch model like [ResNet50](https://docs.pytorch.org/vision/main/models/generated/torchvision.models.resnet50.html) is straightforward. Import the necessary libraries, load your model and a sample input tensor, then use the `profile` function:
 
 ```python
 import torch
@@ -230,7 +230,7 @@ The table below showcases the parameters and MACs for several popular [computer
 
 ## 🙌 Contribute
 
-We actively welcome and encourage community contributions to make THOP even better! Whether it's adding support for new [PyTorch layers](https://pytorch.org/docs/stable/nn.html), improving existing calculations, enhancing documentation, or fixing bugs, your input is valuable. Please see our [Contributing Guide](https://docs.ultralytics.com/help/contributing/) for detailed instructions on how to participate. Together, we can ensure THOP remains a state-of-the-art tool for the [machine learning](https://www.ultralytics.com/glossary/machine-learning-ml) community. Don't hesitate to share your feedback and suggestions!
+We actively welcome and encourage community contributions to make THOP even better! Whether it's adding support for new [PyTorch layers](https://docs.pytorch.org/docs/stable/nn.html), improving existing calculations, enhancing documentation, or fixing bugs, your input is valuable. Please see our [Contributing Guide](https://docs.ultralytics.com/help/contributing/) for detailed instructions on how to participate. Together, we can ensure THOP remains a state-of-the-art tool for the [machine learning](https://www.ultralytics.com/glossary/machine-learning-ml) community. Don't hesitate to share your feedback and suggestions!
 
 ## 📜 License
 

ultralytics_thop-2.0.18.dist-info/RECORD

@@ -0,0 +1,13 @@
+thop/__init__.py,sha256=EBsQOr5rVWk_i2g7NxrycotRXfKmpgcUu1tDBMtJ0Jw,293
+thop/fx_profile.py,sha256=uQETQqeLlOmx_AVunvdZ27xX13a8qVHk7zmjQrd-Yec,8411
+thop/profile.py,sha256=D8OpFcM66YKthAkxnEgdOquuUdsXKnyKbxnOJE9GBM4,8354
+thop/rnn_hooks.py,sha256=JKZ2eSCvIKvhvCDqM4oWPZjmBkdyJ4R2Q7XSn63lsX0,6503
+thop/utils.py,sha256=IwFJQ1v-SLyhm-313Li535R6fhtomkm8Fem1Kfe6G_U,1484
+thop/vision/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXppv1-QUM,70
+thop/vision/basic_hooks.py,sha256=tvr8P4tRJpP9Qyk-tnuXnBPGaMGBj6qlDpGI-uJh5mE,5466
+thop/vision/calc_func.py,sha256=2-LcoFg8ODy8EMb-xEsgLVPigLTki3dXnAg58dxVq80,4530
+ultralytics_thop-2.0.18.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+ultralytics_thop-2.0.18.dist-info/METADATA,sha256=bvyzh4AhrPj8Q8yrbPSNBe4nDGwTzZ9HSlgKDGNZvzc,14942
+ultralytics_thop-2.0.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ultralytics_thop-2.0.18.dist-info/top_level.txt,sha256=HQ7D0gSvDJ31CNR-f0EuXNVve05RYBmwyIkHQKiEhU8,5
+ultralytics_thop-2.0.18.dist-info/RECORD,,

ultralytics_thop-2.0.16.dist-info/RECORD

@@ -1,13 +0,0 @@
-thop/__init__.py,sha256=uIP-Z82T22WjV4ZABASgzJpdE7KrxsGHmoAMbe9Dv1Y,219
-thop/fx_profile.py,sha256=ACI2RLyDYBLB7Cru9y2IAx_YCFX_uhYEArDR0np_tFc,8231
-thop/profile.py,sha256=CsS9mFIW3VWUA15pHQbGDZX-N624u10LZs5FtJOW_Fg,7983
-thop/rnn_hooks.py,sha256=JKZ2eSCvIKvhvCDqM4oWPZjmBkdyJ4R2Q7XSn63lsX0,6503
-thop/utils.py,sha256=IwFJQ1v-SLyhm-313Li535R6fhtomkm8Fem1Kfe6G_U,1484
-thop/vision/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXppv1-QUM,70
-thop/vision/basic_hooks.py,sha256=jrq6HQtyMjO0PouGxiUZEehMVTozlUvslqMyOdoBcnQ,5156
-thop/vision/calc_func.py,sha256=2-LcoFg8ODy8EMb-xEsgLVPigLTki3dXnAg58dxVq80,4530
-ultralytics_thop-2.0.16.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-ultralytics_thop-2.0.16.dist-info/METADATA,sha256=EWJhKFCiDgN0tuXepxy0mF9Z_bxbzOYGYzddFWxqK_k,14922
-ultralytics_thop-2.0.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ultralytics_thop-2.0.16.dist-info/top_level.txt,sha256=HQ7D0gSvDJ31CNR-f0EuXNVve05RYBmwyIkHQKiEhU8,5
-ultralytics_thop-2.0.16.dist-info/RECORD,,