Spaces:
Running
on
Zero
Running
on
Zero
deleted temp files
Browse files
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.
|
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 |
|