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

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

View File

@@ -83,11 +83,12 @@ if __name__ == "__main__":
print("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]
vis = Visualizer(save_dir="./saved_videos", pad_value=120, linewidth=3)
vis.visualize(
video,
pred_tracks,
pred_visibility,
query_frame=0 if args.backward_tracking else args.grid_query_frame,
filename=seq_name,
)