Update visualizer.py

visualizer.py

@@ -270,45 +270,36 @@ class Visualizer:
             img = Image.fromarray(np.uint8(res_video[t]))
             for i in range(N):
                 coord = (tracks[t, i, 0], tracks[t, i, 1])
-                visibile = True
-                if visibility is not None:
-                    visibile = visibility[0, t, i]
-                if coord[0] != 0 and coord[1] != 0:
-                    if not compensate_for_camera_motion or (
-                        compensate_for_camera_motion and segm_mask[i] > 0
-                    ):
-                        # img = draw_circle(
-                        #     img,
-                        #     coord=coord,
-                        #     radius=int(self.linewidth * 2),
-                        #     color=vector_colors[t, i].astype(int),
-                        #     visible=visibile,
-                        #     color_alpha=color_alpha,
-                        # )
-
-                        # coord_ = coord[t,i]
-                        #  draw a red cross
-                        # if gt_tracks[0] > 0 and gt_tracks[1] > 0:
-                        if visibile:
-                            length = self.linewidth * 3
-                            coord_y = ((coord[0]) + length, (coord[1]) + length)
-                            coord_x = ((coord[0]) - length, (coord[1]) - length)
-                            rgb = draw_line(
-                                img,
-                                coord_y,
-                                coord_x,
-                                vector_colors[t, i].astype(int),
-                                self.linewidth,
-                            )
-                            coord_y = ((coord[0]) - length, (coord[1]) + length)
-                            coord_x = ((coord[0]) + length, (coord[1]) - length)
-                            rgb = draw_line(
-                                img,
-                                coord_y,
-                                coord_x,
-                                vector_colors[t, i].astype(int),
-                                self.linewidth,
-                            )
+                visible = True if visibility is None else bool(visibility[0, t, i])
+                
+                # Skip invalid coordinates
+                if not (0 <= coord[0] < W and 0 <= coord[1] < H):
+                    continue
+                    
+                if visible and (not compensate_for_camera_motion or (
+                    compensate_for_camera_motion and segm_mask is not None and segm_mask[i] > 0
+                )):
+                    length = self.linewidth * 3
+                    # Draw cross with anti-aliasing effect
+                    for offset in [-0.5, 0, 0.5]:
+                        coord_y = (coord[0] + length + offset, coord[1] + length + offset)
+                        coord_x = (coord[0] - length + offset, coord[1] - length + offset)
+                        rgb = draw_line(
+                            img,
+                            coord_y,
+                            coord_x,
+                            vector_colors[t, i].astype(int),
+                            self.linewidth,
+                        )
+                        coord_y = (coord[0] - length + offset, coord[1] + length + offset)
+                        coord_x = (coord[0] + length + offset, coord[1] - length + offset)
+                        rgb = draw_line(
+                            img,
+                            coord_y,
+                            coord_x,
+                            vector_colors[t, i].astype(int),
+                            self.linewidth,
+                        )
             res_video[t] = np.array(img)
 
         #  construct the final rgb sequence