Export static shape for detection/instance-segmentation models (#4113)

* Add onnxruntime dependency and update dynamic axes in detection/inst-seg models

Author: eugene123tw

pyproject.toml

@@ -55,6 +55,7 @@ dev = [
     "pytest-mock",
     "pytest-csv",
     "pytest-cov",
+    "onnxruntime==1.20.0",
 ]
 
 docs = [

src/otx/algo/detection/atss.py

@@ -183,7 +183,7 @@ def _exporter(self) -> OTXModelExporter:
                 "input_names": ["image"],
                 "output_names": ["boxes", "labels"],
                 "dynamic_axes": {
-                    "image": {0: "batch", 2: "height", 3: "width"},
+                    "image": {0: "batch"},
                     "boxes": {0: "batch", 1: "num_dets"},
                     "labels": {0: "batch", 1: "num_dets"},
                 },

src/otx/algo/detection/rtmdet.py

@@ -138,7 +138,7 @@ def _exporter(self) -> OTXModelExporter:
                 "input_names": ["image"],
                 "output_names": ["boxes", "labels"],
                 "dynamic_axes": {
-                    "image": {0: "batch", 2: "height", 3: "width"},
+                    "image": {0: "batch"},
                     "boxes": {0: "batch", 1: "num_dets"},
                     "labels": {0: "batch", 1: "num_dets"},
                 },

src/otx/algo/detection/ssd.py

@@ -346,7 +346,7 @@ def _exporter(self) -> OTXModelExporter:
                 "input_names": ["image"],
                 "output_names": ["boxes", "labels"],
                 "dynamic_axes": {
-                    "image": {0: "batch", 2: "height", 3: "width"},
+                    "image": {0: "batch"},
                     "boxes": {0: "batch", 1: "num_dets"},
                     "labels": {0: "batch", 1: "num_dets"},
                 },

src/otx/algo/detection/yolov9.py

@@ -125,7 +125,7 @@ def _exporter(self) -> OTXModelExporter:
                 "export_params": True,
                 "opset_version": 11,
                 "dynamic_axes": {
-                    "image": {0: "batch", 2: "height", 3: "width"},
+                    "image": {0: "batch"},
                     "boxes": {0: "batch", 1: "num_dets"},
                     "labels": {0: "batch", 1: "num_dets"},
                 },

src/otx/algo/detection/yolox.py

@@ -158,7 +158,7 @@ def _exporter(self) -> OTXModelExporter:
                 "export_params": True,
                 "opset_version": 11,
                 "dynamic_axes": {
-                    "image": {0: "batch", 2: "height", 3: "width"},
+                    "image": {0: "batch"},
                     "boxes": {0: "batch", 1: "num_dets"},
                     "labels": {0: "batch", 1: "num_dets"},
                 },

src/otx/algo/instance_segmentation/maskdino.py

@@ -190,7 +190,7 @@ def _exporter(self) -> OTXModelExporter:
             onnx_export_configuration={
                 "input_names": ["image"],
                 "dynamic_axes": {
-                    "image": {0: "batch", 2: "height", 3: "width"},
+                    "image": {0: "batch"},
                     "boxes": {0: "batch", 1: "num_dets"},
                     "labels": {0: "batch", 1: "num_dets"},
                     "masks": {0: "batch", 1: "num_dets", 2: "height", 3: "width"},

src/otx/algo/instance_segmentation/maskrcnn.py

@@ -319,7 +319,7 @@ def _exporter(self) -> OTXModelExporter:
                 "input_names": ["image"],
                 "output_names": ["boxes", "labels", "masks"],
                 "dynamic_axes": {
-                    "image": {0: "batch", 2: "height", 3: "width"},
+                    "image": {0: "batch"},
                     "boxes": {0: "batch", 1: "num_dets"},
                     "labels": {0: "batch", 1: "num_dets"},
                     "masks": {0: "batch", 1: "num_dets", 2: "height", 3: "width"},

src/otx/algo/instance_segmentation/maskrcnn_tv.py

@@ -214,7 +214,7 @@ def _exporter(self) -> OTXModelExporter:
                 "input_names": ["image"],
                 "output_names": ["boxes", "labels", "masks"],
                 "dynamic_axes": {
-                    "image": {0: "batch", 2: "height", 3: "width"},
+                    "image": {0: "batch"},
                     "boxes": {0: "batch", 1: "num_dets"},
                     "labels": {0: "batch", 1: "num_dets"},
                     "masks": {0: "batch", 1: "num_dets", 2: "height", 3: "width"},

src/otx/algo/instance_segmentation/rtmdet_inst.py

@@ -122,7 +122,7 @@ def _exporter(self) -> OTXModelExporter:
                 "input_names": ["image"],
                 "output_names": ["boxes", "labels", "masks"],
                 "dynamic_axes": {
-                    "image": {0: "batch", 2: "height", 3: "width"},
+                    "image": {0: "batch"},
                     "boxes": {0: "batch", 1: "num_dets"},
                     "labels": {0: "batch", 1: "num_dets"},
                     "masks": {0: "batch", 1: "num_dets", 2: "height", 3: "width"},