Update visualizer.py
Browse files- visualizer.py +5 -5
visualizer.py
CHANGED
@@ -286,12 +286,12 @@ class Visualizer:
|
|
286 |
# color_alpha=color_alpha,
|
287 |
# )
|
288 |
|
289 |
-
coord_ = coord[t,i]
|
290 |
# draw a red cross
|
291 |
# if gt_tracks[0] > 0 and gt_tracks[1] > 0:
|
292 |
length = self.linewidth * 3
|
293 |
-
coord_y = (int(
|
294 |
-
coord_x = (int(
|
295 |
rgb = draw_line(
|
296 |
img,
|
297 |
coord_y,
|
@@ -299,8 +299,8 @@ class Visualizer:
|
|
299 |
vector_colors[t, i].astype(int),
|
300 |
self.linewidth,
|
301 |
)
|
302 |
-
coord_y = (int(
|
303 |
-
coord_x = (int(
|
304 |
rgb = draw_line(
|
305 |
img,
|
306 |
coord_y,
|
|
|
286 |
# color_alpha=color_alpha,
|
287 |
# )
|
288 |
|
289 |
+
# coord_ = coord[t,i]
|
290 |
# draw a red cross
|
291 |
# if gt_tracks[0] > 0 and gt_tracks[1] > 0:
|
292 |
length = self.linewidth * 3
|
293 |
+
coord_y = (int(coord[0]) + length, int(coord[1]) + length)
|
294 |
+
coord_x = (int(coord[0]) - length, int(coord[1]) - length)
|
295 |
rgb = draw_line(
|
296 |
img,
|
297 |
coord_y,
|
|
|
299 |
vector_colors[t, i].astype(int),
|
300 |
self.linewidth,
|
301 |
)
|
302 |
+
coord_y = (int(coord[0]) - length, int(coord[1]) + length)
|
303 |
+
coord_x = (int(coord[0]) + length, int(coord[1]) - length)
|
304 |
rgb = draw_line(
|
305 |
img,
|
306 |
coord_y,
|