diff --git a/otx/algorithms/common/adapters/mmdeploy/apis.py b/otx/algorithms/common/adapters/mmdeploy/apis.py index 3cb27904abd..53c4a36e7dc 100644 --- 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(