Fix tiling IS tests

openvinotoolkit · May 17, 2023 · 0ecddd7 · 0ecddd7
1 parent b1760f8
commit 0ecddd7
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/otx/algorithms/common/adapters/mmdeploy/apis.py b/otx/algorithms/common/adapters/mmdeploy/apis.py
@@ -226,6 +226,7 @@ def helper(*args, **kwargs):
                     input_data,
                     cfg,
                     deploy_cfg,
+                    export_type,
                     model_name=model_name,
                 )
 
@@ -255,6 +256,7 @@ def extract_partition(
             input_data: Any,
             cfg: mmcv.Config,
             deploy_cfg: mmcv.Config,
+            export_type: str,
             *,
             model_name: str = "model",
         ):
@@ -276,15 +278,16 @@ def extract_partition(
                 partition_cfgs,
             )
 
-            deploy_cfg_ = deepcopy(deploy_cfg)
-            update_deploy_cfg(partition_onnx[0], deploy_cfg_)
-            MMdeployExporter.onnx2openvino(
-                output_dir,
-                partition_onnx[0],
-                deploy_cfg_,
-            )
-            deploy_cfg["partition_config"]["apply_marks"] = False
-            reset_mark_function_count()
+            if "ONNX" not in export_type:
+                deploy_cfg_ = deepcopy(deploy_cfg)
+                update_deploy_cfg(partition_onnx[0], deploy_cfg_)
+                MMdeployExporter.onnx2openvino(
+                    output_dir,
+                    partition_onnx[0],
+                    deploy_cfg_,
+                )
+                deploy_cfg["partition_config"]["apply_marks"] = False
+                reset_mark_function_count()
 
         @staticmethod
         def torch2onnx(