Fix crash when nposes > 1

google-coral · Feb 3, 2021 · 655a354 · 655a354
1 parent 39d29ac
commit 655a354
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pose_engine.py b/pose_engine.py
@@ -211,7 +211,7 @@ def ParseOutputs(self, outputs):
     def _parse_poses(self, outputs):
         keypoints = outputs[0].reshape(-1, len(KEYPOINTS), 2)
         keypoint_scores = outputs[1].reshape(-1, len(KEYPOINTS))
-        pose_scores = outputs[2]
+        pose_scores = outputs[2].flatten()
         nposes = int(outputs[3][0])
 
         # Convert the poses to a friendlier format of keypoints with associated