fix visualizer

LareinaM · Sep 20, 2023 · 69c9d17 · 69c9d17
1 parent a9251b7
commit 69c9d17
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/mmpose/visualization/local_visualizer_3d.py b/mmpose/visualization/local_visualizer_3d.py
@@ -608,9 +608,8 @@ def add_datasample(self,
             det_img_data = cv2.copyMakeBorder(
                 det_img_data,
                 height // 2,
-                height // 2,
-                width // 2,
-                width // 2,
+                (height // 2 + 1) if height % 2 == 1 else height // 2,
+                width // 2, (width // 2 + 1) if width % 2 == 1 else width // 2,
                 cv2.BORDER_CONSTANT,
                 value=(255, 255, 255))
             drawn_img = np.concatenate((det_img_data, pred_img_data), axis=1)