Fix inference on list of ndarrays (#2506)

open-mmlab · Dec 21, 2023 · 97c1dc5 · 97c1dc5
1 parent 7079ec2
commit 97c1dc5
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/mmdeploy/codebase/mmpose/deploy/pose_detection.py b/mmdeploy/codebase/mmpose/deploy/pose_detection.py
@@ -209,10 +209,13 @@ def create_input(self,
             img_path = [imgs]
         else:
             raise AssertionError('imgs must be strings or numpy arrays')
-        if isinstance(imgs, (list, tuple)) and isinstance(imgs[0], str):
-            img_path = imgs
-            img_data = [mmcv.imread(img) for img in imgs]
-            imgs = img_data
+        if isinstance(imgs, (list, tuple)):
+            if isinstance(imgs[0], str):
+                img_path = imgs
+                img_data = [mmcv.imread(img) for img in imgs]
+                imgs = img_data
+            else:
+                img_path = [''] * len(imgs)
         person_results = []
         bboxes = []
         for img in imgs: