Update visualizer.py

visualizer.py

@@ -269,37 +269,46 @@ class Visualizer:
         for t in range(T):
             img = Image.fromarray(np.uint8(res_video[t]))
             for i in range(N):
-                coord = (tracks[t, i, 0], tracks[t, i, 1])
-                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,
-                        )
+                coord = (round(tracks[t, i, 0]), round(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,
+                            )
             res_video[t] = np.array(img)
 
         #  construct the final rgb sequence