onnxslim 0.1.74__tar.gz → 0.1.76__tar.gz

{onnxslim-0.1.74/onnxslim.egg-info → onnxslim-0.1.76}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxslim
-Version: 0.1.74
+Version: 0.1.76
 Summary: OnnxSlim: A Toolkit to Help Optimize Onnx Model
 Home-page: https://github.com/inisis/OnnxSlim
 Author: inisis
@@ -15,7 +15,7 @@ Requires-Python: >=3.6
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: onnx
-Requires-Dist: sympy>=1.13.3
+Requires-Dist: sympy>=1.13.1
 Requires-Dist: packaging
 Requires-Dist: colorama
 Requires-Dist: ml_dtypes
@@ -36,7 +36,7 @@ Dynamic: summary
 
 <p align="center">
     <a href="https://pypi.org/project/onnxslim">
-        <img src="https://badgen.net/pypi/v/onnxslim?color=blue" />
+        <img src="https://img.shields.io/pypi/v/onnxslim?color=blue" />
     </a>
     <a href="https://pypi.org/project/onnxslim">
         <img src="https://static.pepy.tech/badge/onnxslim/week" />
@@ -131,6 +131,7 @@ For more usage, see onnxslim -h or refer to our [examples](./examples)
 - <img src="https://avatars.githubusercontent.com/u/126587470?s=48&v=4" width="22" height="22"/>[deepghs/imgutils](https://github.com/deepghs/imgutils)
 - <img src="https://avatars.githubusercontent.com/u/48153283?s=48&v=4" width="22" height="22"/>[sunsmarterjie/yolov12](https://github.com/sunsmarterjie/yolov12)
 - <img src="https://avatars.githubusercontent.com/u/147458884?s=48&v=4" width="22" height="22"/>[nndeploy/nndeploy](https://github.com/nndeploy/nndeploy)
+- <img src="https://avatars.githubusercontent.com/u/111754012?s=48&v=4" width="22" height="22"/>[CVCUDA/CV-CUDA](https://github.com/CVCUDA/CV-CUDA)
 
 # References
 

{onnxslim-0.1.74 → onnxslim-0.1.76}/README.md

@@ -2,7 +2,7 @@
 
 <p align="center">
     <a href="https://pypi.org/project/onnxslim">
-        <img src="https://badgen.net/pypi/v/onnxslim?color=blue" />
+        <img src="https://img.shields.io/pypi/v/onnxslim?color=blue" />
     </a>
     <a href="https://pypi.org/project/onnxslim">
         <img src="https://static.pepy.tech/badge/onnxslim/week" />
@@ -97,6 +97,7 @@ For more usage, see onnxslim -h or refer to our [examples](./examples)
 - <img src="https://avatars.githubusercontent.com/u/126587470?s=48&v=4" width="22" height="22"/>[deepghs/imgutils](https://github.com/deepghs/imgutils)
 - <img src="https://avatars.githubusercontent.com/u/48153283?s=48&v=4" width="22" height="22"/>[sunsmarterjie/yolov12](https://github.com/sunsmarterjie/yolov12)
 - <img src="https://avatars.githubusercontent.com/u/147458884?s=48&v=4" width="22" height="22"/>[nndeploy/nndeploy](https://github.com/nndeploy/nndeploy)
+- <img src="https://avatars.githubusercontent.com/u/111754012?s=48&v=4" width="22" height="22"/>[CVCUDA/CV-CUDA](https://github.com/CVCUDA/CV-CUDA)
 
 # References
 

onnxslim-0.1.76/VERSION

@@ -0,0 +1 @@
+0.1.76

{onnxslim-0.1.74 → onnxslim-0.1.76}/onnxslim/core/pattern/fusion/convbn.py

@@ -36,21 +36,21 @@ class ConvBatchNormMatcher(PatternMatcher):
             bn_bias = bn_node.inputs[2].values
             bn_running_mean = bn_node.inputs[3].values
             bn_running_var = bn_node.inputs[4].values
-            bn_eps = bn_node.attrs["epsilon"]
+            bn_eps = bn_node.attrs.get("epsilon", 1.0e-5)
 
             if len(conv_transpose_node.inputs) == 2:
                 conv_transpose_bias = np.zeros_like(bn_running_mean)
             else:
                 conv_transpose_bias = conv_transpose_node.inputs[2].values
 
-            bn_var_rsqrt = 1.0 / np.sqrt(bn_running_var + bn_eps)
+            bn_var_rsqrt = bn_scale / np.sqrt(bn_running_var + bn_eps)
             shape = [1] * len(conv_transpose_weight.shape)
             if bn_node.i(0).op == "Conv":
                 shape[0] = -1
             else:
                 shape[1] = -1
-            conv_w = conv_transpose_weight * (bn_scale * bn_var_rsqrt).reshape(shape)
-            conv_b = (conv_transpose_bias - bn_running_mean) * bn_var_rsqrt * bn_scale + bn_bias
+            conv_w = conv_transpose_weight * bn_var_rsqrt.reshape(shape)
+            conv_b = (conv_transpose_bias - bn_running_mean) * bn_var_rsqrt + bn_bias
 
             inputs = []
             inputs.append(next(iter(conv_transpose_node.inputs)))

{onnxslim-0.1.74 → onnxslim-0.1.76}/onnxslim/third_party/symbolic_shape_infer.py

@@ -359,6 +359,7 @@ class SymbolicShapeInference:
                 for i in self.out_mp_.graph.initializer
             }
         )
+        self.known_vi_.update({i.name: i for i in list(self.out_mp_.graph.output)})
 
     def _merge_symbols(self, dims):
         """Merge dimension symbols, handling automatic merging and validation of symbolic dimensions."""
@@ -582,9 +583,13 @@ class SymbolicShapeInference:
         for i_o in range(len(node.output)):
             o = node.output[i_o]
             if o:  # skip optional output
+                out = model.graph.output[i_o]
+                if not out.type.WhichOneof("value") and o in self.known_vi_: # if empty and already had
+                    continue
+
                 vi = self.out_mp_.graph.value_info.add()
                 if not skip_infer:
-                    vi.CopyFrom(model.graph.output[i_o])
+                    vi.CopyFrom(out)
                 else:
                     vi.name = o
                 self.known_vi_[o] = vi
@@ -1844,7 +1849,7 @@ class SymbolicShapeInference:
             scales = self._try_get_value(node, 2)
             sizes = self._try_get_value(node, 3)
             if sizes is not None:
-                new_sympy_shape = [sympy.simplify(sympy.floor(s)) for s in sizes]
+                new_sympy_shape = [sympy.simplify(round(s)) for s in sizes]
                 self._update_computed_dims(new_sympy_shape)
             elif scales is not None:
                 rank = len(scales)
@@ -1860,7 +1865,7 @@ class SymbolicShapeInference:
                 else:
                     scales = list(scales)
                 new_sympy_shape = [
-                    (sympy.floor(d * (end - start) * scale))
+                    (round(d * (end - start) * scale))
                     for d, start, end, scale in zip(input_sympy_shape, roi_start, roi_end, scales)
                 ]
                 self._update_computed_dims(new_sympy_shape)

onnxslim-0.1.76/onnxslim/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.76"

{onnxslim-0.1.74 → onnxslim-0.1.76/onnxslim.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxslim
-Version: 0.1.74
+Version: 0.1.76
 Summary: OnnxSlim: A Toolkit to Help Optimize Onnx Model
 Home-page: https://github.com/inisis/OnnxSlim
 Author: inisis
@@ -15,7 +15,7 @@ Requires-Python: >=3.6
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: onnx
-Requires-Dist: sympy>=1.13.3
+Requires-Dist: sympy>=1.13.1
 Requires-Dist: packaging
 Requires-Dist: colorama
 Requires-Dist: ml_dtypes
@@ -36,7 +36,7 @@ Dynamic: summary
 
 <p align="center">
     <a href="https://pypi.org/project/onnxslim">
-        <img src="https://badgen.net/pypi/v/onnxslim?color=blue" />
+        <img src="https://img.shields.io/pypi/v/onnxslim?color=blue" />
     </a>
     <a href="https://pypi.org/project/onnxslim">
         <img src="https://static.pepy.tech/badge/onnxslim/week" />
@@ -131,6 +131,7 @@ For more usage, see onnxslim -h or refer to our [examples](./examples)
 - <img src="https://avatars.githubusercontent.com/u/126587470?s=48&v=4" width="22" height="22"/>[deepghs/imgutils](https://github.com/deepghs/imgutils)
 - <img src="https://avatars.githubusercontent.com/u/48153283?s=48&v=4" width="22" height="22"/>[sunsmarterjie/yolov12](https://github.com/sunsmarterjie/yolov12)
 - <img src="https://avatars.githubusercontent.com/u/147458884?s=48&v=4" width="22" height="22"/>[nndeploy/nndeploy](https://github.com/nndeploy/nndeploy)
+- <img src="https://avatars.githubusercontent.com/u/111754012?s=48&v=4" width="22" height="22"/>[CVCUDA/CV-CUDA](https://github.com/CVCUDA/CV-CUDA)
 
 # References
 

{onnxslim-0.1.74 → onnxslim-0.1.76}/onnxslim.egg-info/requires.txt

@@ -1,5 +1,5 @@
 onnx
-sympy>=1.13.3
+sympy>=1.13.1
 packaging
 colorama
 ml_dtypes

{onnxslim-0.1.74 → onnxslim-0.1.76}/setup.py

@@ -25,7 +25,7 @@ setup(
         "Topic :: Scientific/Engineering :: Artificial Intelligence",
     ],
     license="MIT",
-    install_requires=["onnx", "sympy>=1.13.3", "packaging", "colorama", "ml_dtypes"],
+    install_requires=["onnx", "sympy>=1.13.1", "packaging", "colorama", "ml_dtypes"],
     packages=find_packages(exclude=("tests", "tests.*")),
     entry_points={"console_scripts": ["onnxslim=onnxslim.cli:main"]},
     zip_safe=True,

onnxslim-0.1.74/VERSION

@@ -1 +0,0 @@
-0.1.74

onnxslim-0.1.74/onnxslim/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.74"