Merge pull request #334 from PINTO0309/fix_acc_check2

Fixed to never update batch size unless the batch size dimension is an undefined dimension

README.md

@@ -241,7 +241,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.9.16
+  ghcr.io/pinto0309/onnx2tf:1.9.17
 
   or
 

onnx2tf/__init__.py

@@ -1,3 +1,3 @@
 from onnx2tf.onnx2tf import convert, main
 
-__version__ = '1.9.16'
+__version__ = '1.9.17'

onnx2tf/utils/common_functions.py

@@ -3504,7 +3504,8 @@ def dummy_onnx_inference(
         for idx, dim in enumerate(input_size):
             if idx == 0 and input_sizes[0][0] is not None \
                 and not isinstance(input_sizes[0][0], str) \
-                and len(input_sizes[0]) == len(input_size):
+                and len(input_sizes[0]) == len(input_size) \
+                and (dim is None or isinstance(dim, str)):
                 # Batch size assignment for input OPs
                 new_input_size.append(input_sizes[0][0])
             elif dim is None or isinstance(dim, str):
@@ -3581,7 +3582,8 @@ def dummy_tf_inference(
         new_input_size = []
         for idx, dim in enumerate(input_size):
             if idx == 0 and input_sizes[0][0] is not None \
-                and len(input_sizes[0]) == len(input_size):
+                and len(input_sizes[0]) == len(input_size) \
+                and dim is None:
                 # Batch size assignment for input OPs
                 new_input_size.append(input_sizes[0][0])
             elif dim is None: