Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -159,12 +159,12 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
|
|
159 |
# frame_path = os.path.join(temp_frame_dir, f"frame_{count:05d}.png")
|
160 |
# cv2.imwrite(frame_path, combined_frame)
|
161 |
|
162 |
-
cv2.imwrite(f"f{count}.
|
163 |
-
orig_frames.append(f"f{count}.
|
164 |
|
165 |
-
cv2.imwrite(f"f{count}_dmap.
|
166 |
-
depth_frames.append(f"f{count}_dmap.
|
167 |
-
masks.append(f"f{count}_dmap.
|
168 |
count += 1
|
169 |
|
170 |
final_vid = create_video(depth_frames, frame_rate, "depth")
|
|
|
159 |
# frame_path = os.path.join(temp_frame_dir, f"frame_{count:05d}.png")
|
160 |
# cv2.imwrite(frame_path, combined_frame)
|
161 |
|
162 |
+
cv2.imwrite(f"f{count}.png", raw_frame)
|
163 |
+
orig_frames.append(f"f{count}.png")
|
164 |
|
165 |
+
cv2.imwrite(f"f{count}_dmap.png", depth_color)
|
166 |
+
depth_frames.append(f"f{count}_dmap.png")
|
167 |
+
masks.append(f"f{count}_dmap.png")
|
168 |
count += 1
|
169 |
|
170 |
final_vid = create_video(depth_frames, frame_rate, "depth")
|