Support for models with opset<=10

PINTO0309 · Nov 16, 2022 · b275740 · b275740
1 parent 58b0c6b
commit b275740
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -84,7 +84,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.1.22
+  ghcr.io/pinto0309/onnx2tf:1.1.23
 
   or
 

diff --git a/onnx2tf/__init__.py b/onnx2tf/__init__.py
@@ -1,3 +1,3 @@
 from onnx2tf.onnx2tf import convert, main
 
-__version__ = '1.1.22'
+__version__ = '1.1.23'
diff --git a/onnx2tf/ops/Resize.py b/onnx2tf/ops/Resize.py
@@ -47,17 +47,25 @@ def make_node(
     before_op_output_shape_trans = \
         before_op_output_shape_trans_1
 
+    opset = kwargs['opset']
+
     input_tensor = get_constant_or_variable(
         graph_node.inputs[0],
         before_op_output_shape_trans,
     )
     roi = None
-    if len(graph_node.inputs) >= 2:
-        roi = get_constant_or_variable(
-            graph_node.inputs[1],
-            before_op_output_shape_trans,
-        )
     scales = None
+    if len(graph_node.inputs) >= 2:
+        if opset > 10:
+            roi = get_constant_or_variable(
+                graph_node.inputs[1],
+                before_op_output_shape_trans,
+            )
+        else:
+            scales = get_constant_or_variable(
+                graph_node.inputs[1],
+                before_op_output_shape_trans,
+            )
     if len(graph_node.inputs) >= 3:
         scales = get_constant_or_variable(
             graph_node.inputs[2],