Update app.py
Browse files
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=
|
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)
|