From 7796eed13e0fca65f7d8ece883dbfd2c5ffdb460 Mon Sep 17 00:00:00 2001 From: Zheng-LinXiao <951310862@qq.com> Date: Wed, 12 Apr 2023 23:37:18 +0800 Subject: [PATCH] fix-visualizer --- mmpose/visualization/local_visualizer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmpose/visualization/local_visualizer.py b/mmpose/visualization/local_visualizer.py index 0d6fcc61f4..b19e89dea6 100644 --- 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)