nikkar commited on
Commit
8417f13
·
verified ·
1 Parent(s): 098b174

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -397,7 +397,7 @@ def track(
397
 
398
  pred_tracks = (pred_tracks * torch.tensor([video_preview.shape[2], video_preview.shape[1]]).to(device) / torch.tensor([VIDEO_INPUT_RESO[1], VIDEO_INPUT_RESO[0]]).to(device))
399
 
400
- vis = Visualizer(save_dir="./saved_videos", pad_value=0, linewidth=1, tracks_leave_trace=0)
401
  # segm_mask = torch.zeros(queries.shape[1])
402
  # segm_mask[:num_tracks] = 1
403
  # print('segm_mask',segm_mask.shape, segm_mask)
 
397
 
398
  pred_tracks = (pred_tracks * torch.tensor([video_preview.shape[2], video_preview.shape[1]]).to(device) / torch.tensor([VIDEO_INPUT_RESO[1], VIDEO_INPUT_RESO[0]]).to(device))
399
 
400
+ vis = Visualizer(save_dir="./saved_videos", pad_value=0, linewidth=2, tracks_leave_trace=0)
401
  # segm_mask = torch.zeros(queries.shape[1])
402
  # segm_mask[:num_tracks] = 1
403
  # print('segm_mask',segm_mask.shape, segm_mask)