aharley commited on
Commit
49aebdf
·
1 Parent(s): 91848f9

deleted temp files

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -609,6 +609,8 @@ def paint_video(video_preview, query_frame, video_fps, tracks, visibs, rate=1, s
609
  video_path = os.path.join(os.path.dirname(__file__), "tmp")
610
  video_file_path = os.path.join(video_path, video_file_name)
611
  os.makedirs(video_path, exist_ok=True)
 
 
612
  if False:
613
  mediapy.write_video(video_file_path, painted_video, fps=video_fps)
614
  else:
@@ -786,7 +788,7 @@ def track(
786
  tracks = traj_maps_e.permute(0,3,4,1,2).reshape(-1,T,2).numpy()
787
  visibs = visconf_maps_e.permute(0,3,4,1,2).reshape(-1,T,2)[:,:,0].numpy()
788
  confs = visconf_maps_e.permute(0,3,4,1,2).reshape(-1,T,2)[:,:,0].numpy()
789
- # visibs = (visibs * confs) > 0.2 # N,T
790
  visibs = (confs) > 0.1 # N,T
791
  # visibs = (confs) > 0.1 # N,T
792
 
 
609
  video_path = os.path.join(os.path.dirname(__file__), "tmp")
610
  video_file_path = os.path.join(video_path, video_file_name)
611
  os.makedirs(video_path, exist_ok=True)
612
+ for f in glob.glob(os.path.join(video_path, "*.jpg")):
613
+ os.remove(f)
614
  if False:
615
  mediapy.write_video(video_file_path, painted_video, fps=video_fps)
616
  else:
 
788
  tracks = traj_maps_e.permute(0,3,4,1,2).reshape(-1,T,2).numpy()
789
  visibs = visconf_maps_e.permute(0,3,4,1,2).reshape(-1,T,2)[:,:,0].numpy()
790
  confs = visconf_maps_e.permute(0,3,4,1,2).reshape(-1,T,2)[:,:,0].numpy()
791
+ # visibs = (visibs * confs) > 0.1 # N,T
792
  visibs = (confs) > 0.1 # N,T
793
  # visibs = (confs) > 0.1 # N,T
794