nikkar commited on
Commit
293db1f
·
verified ·
1 Parent(s): 842281d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -390,7 +390,10 @@ def track(
390
  colors.extend(frame_colors)
391
  colors = np.array(colors)
392
 
393
- painted_video = paint_point_track(video_preview,tracks,pred_occ,colors)
 
 
 
394
 
395
  # save video
396
  video_file_name = uuid.uuid4().hex + ".mp4"
 
390
  colors.extend(frame_colors)
391
  colors = np.array(colors)
392
 
393
+ vis = Visualizer(save_dir="./saved_videos", pad_value=120, linewidth=3)
394
+ painted_video = vis.visualize(video_preview, tracks, pred_occ, save_video=False)
395
+
396
+ # painted_video = paint_point_track(video_preview,tracks,pred_occ,colors)
397
 
398
  # save video
399
  video_file_name = uuid.uuid4().hex + ".mp4"