[Fix] Align visualization color of 3d demo (#2734)

open-mmlab · Oct 7, 2023 · 451a17a · 451a17a
1 parent a7f7939
commit 451a17a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmpose/visualization/local_visualizer_3d.py b/mmpose/visualization/local_visualizer_3d.py
@@ -199,7 +199,7 @@ def _draw_3d_instances_kpts(keypoints,
 
                 x_3d, y_3d, z_3d = np.split(kpts_valid[:, :3], [1, 2], axis=1)
 
-                kpt_color = kpt_color[valid][..., ::-1] / 255.
+                kpt_color = kpt_color[valid] / 255.
 
                 ax.scatter(x_3d, y_3d, z_3d, marker='o', c=kpt_color)
 
@@ -230,7 +230,7 @@ def _draw_3d_instances_kpts(keypoints,
                         if kpt_score.min() > kpt_thr and kpt_score_2d.min(
                         ) > kpt_thr:
                             # matplotlib uses RGB color in [0, 1] value range
-                            _color = link_color[sk_id][::-1] / 255.
+                            _color = link_color[sk_id] / 255.
                             ax.plot(
                                 xs_3d, ys_3d, zs_3d, color=_color, zdir='z')