nikkar commited on
Commit
dfc9682
·
verified ·
1 Parent(s): 7da67b4

Update visualizer.py

Browse files
Files changed (1) hide show
  1. 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(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,8 +299,8 @@ class Visualizer:
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,
 
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,