From a4bbbd59397e16d403b1c60bfdb062122c87ffb9 Mon Sep 17 00:00:00 2001
From: Songki Choi <songki.choi@intel.com>
Date: Tue, 23 May 2023 23:10:23 +0900
Subject: [PATCH] Fix deploy patch bug (w,h) -> (h,w)
---
.../detection/adapters/mmdet/utils/config_utils.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/otx/algorithms/detection/adapters/mmdet/utils/config_utils.py b/otx/algorithms/detection/adapters/mmdet/utils/config_utils.py
index 006559b48c7..3fd49f978bf 100644
--- a/otx/algorithms/detection/adapters/mmdet/utils/config_utils.py
+++ b/otx/algorithms/detection/adapters/mmdet/utils/config_utils.py
@@ -449,9 +449,10 @@ def patch_input_shape(cfg: ConfigDict, deploy_cfg: ConfigDict):
assert all(isinstance(i, int) and i > 0 for i in size)
# default is static shape to prevent an unexpected error
# when converting to OpenVINO IR
+ w, h = size
logger.info(f"Patching OpenVINO IR input shape: {size}")
- deploy_cfg.ir_config.input_shape = size
- deploy_cfg.backend_config.model_inputs = [ConfigDict(opt_shapes=ConfigDict(input=[1, 3, *size]))]
+ deploy_cfg.ir_config.input_shape = (w, h)
+ deploy_cfg.backend_config.model_inputs = [ConfigDict(opt_shapes=ConfigDict(input=[1, 3, h, w]))]
def patch_ir_scale_factor(deploy_cfg: ConfigDict, hyper_parameters: DetectionConfig):