fix empty person results (#569)

open-mmlab · Apr 13, 2021 · 0274ecf · 0274ecf
1 parent dc7adc2
commit 0274ecf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/mmpose/apis/inference.py b/mmpose/apis/inference.py
@@ -362,6 +362,9 @@ def inference_top_down_pose_model(model,
     pose_results = []
     returned_outputs = []
 
+    if len(person_results) == 0:
+        return pose_results, returned_outputs
+
     # Change for-loop preprocess each bbox to preprocess all bboxes at once.
     bboxes = np.array([box['bbox'] for box in person_results])