Update visualizer.py

visualizer.py

@@ -255,7 +255,7 @@ class Visualizer:
                     curr_tracks = curr_tracks[:, segm_mask > 0]
                     curr_colors = curr_colors[:, segm_mask > 0]
 
-                res_video[t] = self._draw_gt_tracks(
+                res_video[t] = self._draw_pred_tracks(
                     res_video[t],
                     curr_tracks,
                     curr_colors,
@@ -277,13 +277,36 @@ class Visualizer:
                     if not compensate_for_camera_motion or (
                         compensate_for_camera_motion and segm_mask[i] > 0
                     ):
-                        img = draw_circle(
+                        # 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:
+                        length = self.linewidth * 3
+                        coord_y = (int(coord_[0]) + length, int(coord_[1]) + length)
+                        coord_x = (int(coord_[0]) - length, int(coord_[1]) - length)
+                        rgb = draw_line(
                             img,
-                            coord=coord,
-                            radius=int(self.linewidth * 2),
-                            color=vector_colors[t, i].astype(int),
-                            visible=visibile,
-                            color_alpha=color_alpha,
+                            coord_y,
+                            coord_x,
+                            vector_colors[t, i].astype(int),
+                            self.linewidth,
+                        )
+                        coord_y = (int(coord_[0]) - length, int(coord_[1]) + length)
+                        coord_x = (int(coord_[0]) + length, int(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)