fix export sdk info of input shape (#667)

open-mmlab · Jun 29, 2022 · c792d06 · c792d06
1 parent 5858488
commit c792d06
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mmdeploy/codebase/mmpose/deploy/pose_detection.py b/mmdeploy/codebase/mmpose/deploy/pose_detection.py
@@ -38,6 +38,8 @@ def process_model_config(
     sdk_pipeline = []
     color_type = 'color'
     channel_order = 'rgb'
+    if input_shape is None:
+        input_shape = np.array(cfg.data_cfg['image_size'])
 
     idx = 0
     while idx < len(test_pipeline):