Update app.py

app.py

@@ -114,7 +114,6 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             if not ret:
                 break
 
-            orig_frames.append(raw_frame)
             frame = cv2.cvtColor(raw_frame, cv2.COLOR_BGR2RGB) / 255.0
             frame_pil =  Image.fromarray((frame * 255).astype(np.uint8))
             frame = transform({'image': frame})['image']
@@ -147,8 +146,12 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             # out.write(combined_frame)
             # frame_path = os.path.join(temp_frame_dir, f"frame_{count:05d}.png")
             # cv2.imwrite(frame_path, combined_frame)
-            cv2.imwrite(f"frame{count}.jpg", depth_color)
-            depth_frames.append(f"frame{count}.jpg")
+            orig_frame = cv2.cvtColor(raw_frame, cv2.COLOR_BGR2RGB)
+            
+            cv2.imwrite(f"vframe{count}.jpg", orig_frame)
+            orig_frames.append(f"vframe{count}.jpg")
+            cv2.imwrite(f"dframe{count}.jpg", depth_color)
+            depth_frames.append(f"dframe{count}.jpg")
             count += 1
 
         final_vid = create_video(depth_frames, frame_rate, "depth")