freealise commited on
Commit
549786e
1 Parent(s): 0455f94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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}.jpg", raw_frame)
163
- orig_frames.append(f"f{count}.jpg")
164
 
165
- cv2.imwrite(f"f{count}_dmap.jpg", depth_color)
166
- depth_frames.append(f"f{count}_dmap.jpg")
167
- masks.append(f"f{count}_dmap.jpg")
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")