iicd/cotracker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

readme.md update, demo flexible save path (#83)

Browse Source
This commit is contained in:
Iurii Makarov
2024-05-11 15:34:09 +01:00
committed by GitHub
parent 0f9d32869a
commit e29e938311
3 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
online_demo.py
View File

@@ -97,7 +97,7 @@ if __name__ == "__main__":
print("Tracks are computed")
# save a video with predicted tracks
seq_name = args.video_path.split("/")[-1]
seq_name = os.path.splitext(args.video_path.split("/")[-1])[0]
video = torch.tensor(np.stack(window_frames), device=DEFAULT_DEVICE).permute(0, 3, 1, 2)[None]
vis = Visualizer(save_dir="./saved_videos", pad_value=120, linewidth=3)
vis.visualize(video, pred_tracks, pred_visibility, query_frame=args.grid_query_frame)
vis.visualize(video, pred_tracks, pred_visibility, query_frame=args.grid_query_frame, filename=seq_name)