From c792d06706e972cd614660c19a6274c0a22fefff Mon Sep 17 00:00:00 2001
From: Chen Xin <xinchen.tju@gmail.com>
Date: Wed, 29 Jun 2022 15:19:39 +0800
Subject: [PATCH] fix export sdk info of input shape (#667)

---
 mmdeploy/codebase/mmpose/deploy/pose_detection.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mmdeploy/codebase/mmpose/deploy/pose_detection.py b/mmdeploy/codebase/mmpose/deploy/pose_detection.py
index c6d8f0d643..fde76460be 100644
--- 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):