[Fix] fix a bug when visualize keypoint indices (#2631)

open-mmlab · Aug 18, 2023 · 14e0d67 · 14e0d67
1 parent 91c0a84
commit 14e0d67
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/mmpose/visualization/local_visualizer.py b/mmpose/visualization/local_visualizer.py
@@ -345,11 +345,10 @@ def _draw_instances_kpts(self,
                         alpha=transparency,
                         line_widths=self.radius)
                     if show_kpt_idx:
-                        kpt[0] += self.radius
-                        kpt[1] -= self.radius
+                        kpt_idx_coords = kpt + [self.radius, -self.radius]
                         self.draw_texts(
                             str(kid),
-                            kpt,
+                            kpt_idx_coords,
                             colors=color,
                             font_sizes=self.radius * 3,
                             vertical_alignments='bottom',