Spaces:
Runtime error
Runtime error
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 |
|