fix-visualizer (open-mmlab#2210)

shuheilocale · Apr 12, 2023 · 3fe9fba · 3fe9fba
1 parent c8ec77b
commit 3fe9fba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmpose/visualization/local_visualizer.py b/mmpose/visualization/local_visualizer.py
@@ -265,8 +265,8 @@ def _draw_instances_kpts(self,
                 neck = np.mean(keypoints_info[:, [5, 6]], axis=1)
                 # neck score when visualizing pred
                 neck[:, 2:4] = np.logical_and(
-                    keypoints_info[:, 5, 2:4] < kpt_thr,
-                    keypoints_info[:, 6, 2:4] < kpt_thr).astype(int)
+                    keypoints_info[:, 5, 2:4] > kpt_thr,
+                    keypoints_info[:, 6, 2:4] > kpt_thr).astype(int)
                 new_keypoints_info = np.insert(
                     keypoints_info, 17, neck, axis=1)