onnx2tf 1.29.19__tar.gz → 1.29.21__tar.gz

{onnx2tf-1.29.19 → onnx2tf-1.29.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnx2tf
-Version: 1.29.19
+Version: 1.29.21
 Summary: Self-Created Tools to convert ONNX files (NCHW) to TensorFlow/TFLite/Keras format (NHWC). The purpose of this tool is to solve the massive Transpose extrapolation problem in onnx-tensorflow (onnx-tf).
 Keywords: onnx,tensorflow,tflite,keras,deep-learning,machine-learning
 Author: Katsuya Hyodo
@@ -18,7 +18,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Dist: requests==2.32.5
 Requires-Dist: numpy==1.26.4
-Requires-Dist: onnx==1.19.0
+Requires-Dist: onnx==1.19.1
 Requires-Dist: onnxruntime==1.23.0
 Requires-Dist: opencv-python==4.11.0.86
 Requires-Dist: onnxsim==0.4.36
@@ -365,7 +365,7 @@ Video speed is adjusted approximately 50 times slower than actual speed.
   docker run --rm -it \
   -v `pwd`:/workdir \
   -w /workdir \
-  ghcr.io/pinto0309/onnx2tf:1.29.19
+  ghcr.io/pinto0309/onnx2tf:1.29.21
 
   or
 
@@ -373,18 +373,18 @@ Video speed is adjusted approximately 50 times slower than actual speed.
   docker run --rm -it \
   -v `pwd`:/workdir \
   -w /workdir \
-  docker.io/pinto0309/onnx2tf:1.29.19
+  docker.io/pinto0309/onnx2tf:1.29.21
 
   or
 
-  pip install -U onnx==1.19.0 \
+  pip install -U onnx==1.19.1 \
   && pip install -U onnx-graphsurgeon==0.5.8 \
   && pip install -U onnxruntime==1.23.0 \
   && pip install -U onnxsim==0.4.36 \
   && pip install -U onnxoptimizer==0.4.2 \
   && pip install -U simple_onnx_processing_tools==1.1.32 \
-  && pip install -U sne4onnx>=1.0.13 \
-  && pip install -U sng4onnx>=1.0.4 \
+  && pip install -U sne4onnx==1.0.15 \
+  && pip install -U sng4onnx==1.0.5 \
   && pip install -U ai_edge_litert==1.2.0 \
   && pip install -U tensorflow==2.19.0 \
   && pip install -U protobuf==3.20.3 \
@@ -630,7 +630,7 @@ After many upgrades, the need for JSON parameter correction has become much less
 
 `-ois` an option to overwrite the input OP to a static size if it has undefined dimensions. `-cotof` option checks the accuracy of all OPs one by one. `-cotoa` is the error value of the threshold for determining an accuracy error. If there are undefined dimensions in the input OP, it is better to fix them to the static geometry to improve the accuracy of the accuracy measurement.
 
-Also, you can use the `-cind` option to specify custom input for `-cotof`, instead of using the default dummy input. Otherwise, all input values will be set to 1. For more information about the `-cind` option, please refer to [here](#cli-parameter).
+Also, you can use the `-cind` option to specify custom input for `-cotof`, instead of using the default dummy input. Otherwise, all input values will be set to 1. You can override the dummy input values with `--value_hints` (scalar only, `*:default` supported). For more information about the `-cind` option, please refer to [here](#cli-parameter).
 
 The `-cotof` option only compares the original ONNX and converted TensorFlow (Keras) models at Float32 precision, not at Float16 or INT8 precision.
 
@@ -644,6 +644,10 @@ onnx2tf -i mobilenetv2-12.onnx -b 1 -cotof -cotoa 1e-1
 or
 
 onnx2tf -i mobilenetv2-12.onnx -cotof -cotoa 1e-1 -cind "input" "/your/path/x.npy"
+
+or
+
+onnx2tf -i mobilenetv2-12.onnx -cotof -cotoa 1e-1 --value_hints "input:0.5" "*:1.0"
 ```
 ![image](https://user-images.githubusercontent.com/33194443/216901668-5fdb1e38-8670-46a4-b4b9-8a774fa7545e.png)
 
@@ -1826,6 +1830,14 @@ optional arguments:
     A value of 1 or more must be specified.
     Numerical values other than dynamic dimensions are ignored.
 
+  -vh VALUE_HINTS [VALUE_HINTS ...], \
+      --value_hints VALUE_HINTS [VALUE_HINTS ...]
+    Value hints for dummy inference input tensors.
+    The format is
+    "input_name_1:value" "input_name_2:value" "*:default_value"
+    "*" applies to all inputs not explicitly specified.
+    Values are scalar only.
+
   -nlt, --no_large_tensor
     Suppresses constant bloat caused by Tile OP when optimizing models in onnxsim.
     See: https://github.com/daquexian/onnx-simplifier/issues/178
@@ -1887,6 +1899,15 @@ optional arguments:
     model partitioned into subgraphs.
     e.g. --output_names_to_interrupt_model_conversion "output0" "output1" "output2"
 
+  -easm, --enable_auto_split_model
+    Force auto split regardless of the ONNX file size.
+    Uses --auto_split_max_size_mb as the target partition size.
+
+  -asmsm AUTO_SPLIT_MAX_SIZE_MB, --auto_split_max_size_mb AUTO_SPLIT_MAX_SIZE_MB
+    Target maximum size per partition in MB based on ONNX initializer sizes.
+    Used when auto-split is triggered or forced.
+    Default: 1024
+
   -dgc, --disable_group_convolution
     Disable GroupConvolution and replace it with SeparableConvolution for
     output to saved_model format.
@@ -2148,6 +2169,7 @@ convert(
   batch_size: Union[int, NoneType] = None,
   overwrite_input_shape: Union[List[str], NoneType] = None,
   shape_hints: Union[List[str], NoneType] = None,
+  value_hints: Union[List[str], NoneType] = None,
   no_large_tensor: Optional[bool] = False,
   output_nms_with_dynamic_tensor: Optional[bool] = False,
   switch_nms_version: Optional[str] = 'v4',
@@ -2156,6 +2178,8 @@ convert(
   keep_shape_absolutely_input_names: Optional[List[str]] = None,
   input_names_to_interrupt_model_conversion: Union[List[str], NoneType] = None,
   output_names_to_interrupt_model_conversion: Union[List[str], NoneType] = None,
+  enable_auto_split_model: Optional[bool] = False,
+  auto_split_max_size_mb: Optional[int] = 1024,
   disable_group_convolution: Union[bool, NoneType] = False,
   enable_batchmatmul_unfold: Optional[bool] = False,
   enable_rnn_unroll: Optional[bool] = False,
@@ -2366,6 +2390,13 @@ convert(
       A value of 1 or more must be specified.
       Numerical values other than dynamic dimensions are ignored.
 
+    value_hints: Optional[List[str]]
+      Value hints for dummy inference input tensors.
+      The format is
+      ['input_name_1:value', 'input_name_2:value', '*:default_value']
+      "*" applies to all inputs not explicitly specified.
+      Values are scalar only.
+
     no_large_tensor: Optional[bool]
       Suppresses constant bloat caused by Tile OP when optimizing models in onnxsim.
       See: https://github.com/daquexian/onnx-simplifier/issues/178
@@ -2424,6 +2455,17 @@ convert(
       e.g.
       output_names_to_interrupt_model_conversion=['output0','output1','output2']
 
+    enable_auto_split_model: Optional[bool]
+      Force auto split regardless of the ONNX file size.
+      Uses auto_split_max_size_mb as the target partition size.
+      Short option: -easm
+      Default: False
+
+    auto_split_max_size_mb: Optional[int]
+      Target maximum size per partition in MB based on ONNX initializer sizes.
+      Used when auto-split is triggered or forced.
+      Default: 1024
+
     disable_group_convolution: Optional[bool]
       Disable GroupConvolution and replace it with SeparableConvolution for
       output to saved_model format.
@@ -3010,6 +3052,7 @@ The above differences often cannot be dealt with by simply converting the model
 14. [nobuco](https://github.com/AlexanderLutsenko/nobuco)
 15. [onnx2torch](https://github.com/ENOT-AutoDL/onnx2torch)
 16. [ai-edge-torch](https://github.com/google-ai-edge/ai-edge-torch)
+17. [LiteRT.js](https://ai.google.dev/edge/litert/web)
 
 ## Acknowledgement
 1. https://github.com/onnx/models

{onnx2tf-1.29.19 → onnx2tf-1.29.21}/README.md

@@ -323,7 +323,7 @@ Video speed is adjusted approximately 50 times slower than actual speed.
   docker run --rm -it \
   -v `pwd`:/workdir \
   -w /workdir \
-  ghcr.io/pinto0309/onnx2tf:1.29.19
+  ghcr.io/pinto0309/onnx2tf:1.29.21
 
   or
 
@@ -331,18 +331,18 @@ Video speed is adjusted approximately 50 times slower than actual speed.
   docker run --rm -it \
   -v `pwd`:/workdir \
   -w /workdir \
-  docker.io/pinto0309/onnx2tf:1.29.19
+  docker.io/pinto0309/onnx2tf:1.29.21
 
   or
 
-  pip install -U onnx==1.19.0 \
+  pip install -U onnx==1.19.1 \
   && pip install -U onnx-graphsurgeon==0.5.8 \
   && pip install -U onnxruntime==1.23.0 \
   && pip install -U onnxsim==0.4.36 \
   && pip install -U onnxoptimizer==0.4.2 \
   && pip install -U simple_onnx_processing_tools==1.1.32 \
-  && pip install -U sne4onnx>=1.0.13 \
-  && pip install -U sng4onnx>=1.0.4 \
+  && pip install -U sne4onnx==1.0.15 \
+  && pip install -U sng4onnx==1.0.5 \
   && pip install -U ai_edge_litert==1.2.0 \
   && pip install -U tensorflow==2.19.0 \
   && pip install -U protobuf==3.20.3 \
@@ -588,7 +588,7 @@ After many upgrades, the need for JSON parameter correction has become much less
 
 `-ois` an option to overwrite the input OP to a static size if it has undefined dimensions. `-cotof` option checks the accuracy of all OPs one by one. `-cotoa` is the error value of the threshold for determining an accuracy error. If there are undefined dimensions in the input OP, it is better to fix them to the static geometry to improve the accuracy of the accuracy measurement.
 
-Also, you can use the `-cind` option to specify custom input for `-cotof`, instead of using the default dummy input. Otherwise, all input values will be set to 1. For more information about the `-cind` option, please refer to [here](#cli-parameter).
+Also, you can use the `-cind` option to specify custom input for `-cotof`, instead of using the default dummy input. Otherwise, all input values will be set to 1. You can override the dummy input values with `--value_hints` (scalar only, `*:default` supported). For more information about the `-cind` option, please refer to [here](#cli-parameter).
 
 The `-cotof` option only compares the original ONNX and converted TensorFlow (Keras) models at Float32 precision, not at Float16 or INT8 precision.
 
@@ -602,6 +602,10 @@ onnx2tf -i mobilenetv2-12.onnx -b 1 -cotof -cotoa 1e-1
 or
 
 onnx2tf -i mobilenetv2-12.onnx -cotof -cotoa 1e-1 -cind "input" "/your/path/x.npy"
+
+or
+
+onnx2tf -i mobilenetv2-12.onnx -cotof -cotoa 1e-1 --value_hints "input:0.5" "*:1.0"
 ```
 ![image](https://user-images.githubusercontent.com/33194443/216901668-5fdb1e38-8670-46a4-b4b9-8a774fa7545e.png)
 
@@ -1784,6 +1788,14 @@ optional arguments:
     A value of 1 or more must be specified.
     Numerical values other than dynamic dimensions are ignored.
 
+  -vh VALUE_HINTS [VALUE_HINTS ...], \
+      --value_hints VALUE_HINTS [VALUE_HINTS ...]
+    Value hints for dummy inference input tensors.
+    The format is
+    "input_name_1:value" "input_name_2:value" "*:default_value"
+    "*" applies to all inputs not explicitly specified.
+    Values are scalar only.
+
   -nlt, --no_large_tensor
     Suppresses constant bloat caused by Tile OP when optimizing models in onnxsim.
     See: https://github.com/daquexian/onnx-simplifier/issues/178
@@ -1845,6 +1857,15 @@ optional arguments:
     model partitioned into subgraphs.
     e.g. --output_names_to_interrupt_model_conversion "output0" "output1" "output2"
 
+  -easm, --enable_auto_split_model
+    Force auto split regardless of the ONNX file size.
+    Uses --auto_split_max_size_mb as the target partition size.
+
+  -asmsm AUTO_SPLIT_MAX_SIZE_MB, --auto_split_max_size_mb AUTO_SPLIT_MAX_SIZE_MB
+    Target maximum size per partition in MB based on ONNX initializer sizes.
+    Used when auto-split is triggered or forced.
+    Default: 1024
+
   -dgc, --disable_group_convolution
     Disable GroupConvolution and replace it with SeparableConvolution for
     output to saved_model format.
@@ -2106,6 +2127,7 @@ convert(
   batch_size: Union[int, NoneType] = None,
   overwrite_input_shape: Union[List[str], NoneType] = None,
   shape_hints: Union[List[str], NoneType] = None,
+  value_hints: Union[List[str], NoneType] = None,
   no_large_tensor: Optional[bool] = False,
   output_nms_with_dynamic_tensor: Optional[bool] = False,
   switch_nms_version: Optional[str] = 'v4',
@@ -2114,6 +2136,8 @@ convert(
   keep_shape_absolutely_input_names: Optional[List[str]] = None,
   input_names_to_interrupt_model_conversion: Union[List[str], NoneType] = None,
   output_names_to_interrupt_model_conversion: Union[List[str], NoneType] = None,
+  enable_auto_split_model: Optional[bool] = False,
+  auto_split_max_size_mb: Optional[int] = 1024,
   disable_group_convolution: Union[bool, NoneType] = False,
   enable_batchmatmul_unfold: Optional[bool] = False,
   enable_rnn_unroll: Optional[bool] = False,
@@ -2324,6 +2348,13 @@ convert(
       A value of 1 or more must be specified.
       Numerical values other than dynamic dimensions are ignored.
 
+    value_hints: Optional[List[str]]
+      Value hints for dummy inference input tensors.
+      The format is
+      ['input_name_1:value', 'input_name_2:value', '*:default_value']
+      "*" applies to all inputs not explicitly specified.
+      Values are scalar only.
+
     no_large_tensor: Optional[bool]
       Suppresses constant bloat caused by Tile OP when optimizing models in onnxsim.
       See: https://github.com/daquexian/onnx-simplifier/issues/178
@@ -2382,6 +2413,17 @@ convert(
       e.g.
       output_names_to_interrupt_model_conversion=['output0','output1','output2']
 
+    enable_auto_split_model: Optional[bool]
+      Force auto split regardless of the ONNX file size.
+      Uses auto_split_max_size_mb as the target partition size.
+      Short option: -easm
+      Default: False
+
+    auto_split_max_size_mb: Optional[int]
+      Target maximum size per partition in MB based on ONNX initializer sizes.
+      Used when auto-split is triggered or forced.
+      Default: 1024
+
     disable_group_convolution: Optional[bool]
       Disable GroupConvolution and replace it with SeparableConvolution for
       output to saved_model format.
@@ -2968,6 +3010,7 @@ The above differences often cannot be dealt with by simply converting the model
 14. [nobuco](https://github.com/AlexanderLutsenko/nobuco)
 15. [onnx2torch](https://github.com/ENOT-AutoDL/onnx2torch)
 16. [ai-edge-torch](https://github.com/google-ai-edge/ai-edge-torch)
+17. [LiteRT.js](https://ai.google.dev/edge/litert/web)
 
 ## Acknowledgement
 1. https://github.com/onnx/models

{onnx2tf-1.29.19 → onnx2tf-1.29.21}/onnx2tf/__init__.py

@@ -1,3 +1,3 @@
 from onnx2tf.onnx2tf import convert, main
 
-__version__ = '1.29.19'
+__version__ = '1.29.21'