fix

open-mmlab · Jun 15, 2023 · fd09517 · fd09517
1 parent 113aa8c
commit fd09517
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/mmdeploy/codebase/mmpose/deploy/pose_detection.py b/mmdeploy/codebase/mmpose/deploy/pose_detection.py
@@ -247,10 +247,15 @@ def create_input(self,
                 bbox_score = np.array([bbox[4] if len(bbox) == 5 else 1
                                        ])  # shape (1,)
                 data = {
-                    'img': imgs[i],
-                    'bbox_score': bbox_score,
-                    'bbox': [] if cfg.model.bbox_head.type == 'YOLOXPoseHead' else bbox[None],
-                    'img_path': img_path[i]
+                    'img':
+                    imgs[i],
+                    'bbox_score':
+                    bbox_score,
+                    'bbox': [] if hasattr(cfg.model, 'bbox_head')
+                    and cfg.model.bbox_head.type == 'YOLOXPoseHead' else
+                    bbox[None],
+                    'img_path':
+                    img_path[i]
                 }
                 data.update(meta_data)
                 data = test_pipeline(data)