Hugging Face's logo

Spaces:
nikkar
/
cotracker
Sleeping

App Files Community
nikkar commited on
Commit
098b174
·
verified ·
1 Parent(s): dfc9682

Update visualizer.py

Browse files
Files changed (1) hide show
  1. visualizer.py +20 -19
visualizer.py CHANGED
@@ -289,25 +289,26 @@ class Visualizer:
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,
298
- coord_x,
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,
307
- coord_x,
308
- vector_colors[t, i].astype(int),
309
- self.linewidth,
310
- )
 
311
  res_video[t] = np.array(img)
312
 
313
  # construct the final rgb sequence
 
289
  # coord_ = coord[t,i]
290
  # draw a red cross
291
  # if gt_tracks[0] > 0 and gt_tracks[1] > 0:
292
+ if visibile:
293
+ length = self.linewidth * 3
294
+ coord_y = (int(coord[0]) + length, int(coord[1]) + length)
295
+ coord_x = (int(coord[0]) - length, int(coord[1]) - length)
296
+ rgb = draw_line(
297
+ img,
298
+ coord_y,
299
+ coord_x,
300
+ vector_colors[t, i].astype(int),
301
+ self.linewidth,
302
+ )
303
+ coord_y = (int(coord[0]) - length, int(coord[1]) + length)
304
+ coord_x = (int(coord[0]) + length, int(coord[1]) - length)
305
+ rgb = draw_line(
306
+ img,
307
+ coord_y,
308
+ coord_x,
309
+ vector_colors[t, i].astype(int),
310
+ self.linewidth,
311
+ )
312
  res_video[t] = np.array(img)
313
 
314
  # construct the final rgb sequence