Update app.py

app.py

@@ -114,7 +114,6 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         #fourcc = cv2.VideoWriter_fourcc(*'mp4v')
         #out = cv2.VideoWriter(output_path, fourcc, frame_rate, (output_width, frame_height))
         count=0
-        depth_frames_gray = []
         depth_frames = []
         orig_frames = []
         while raw_video.isOpened():
@@ -137,7 +136,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             depth = depth.cpu().numpy().astype(np.uint8)
             depth_color = cv2.applyColorMap(depth, cv2.COLORMAP_BONE)
             depth_gray = cv2.cvtColor(depth_color, cv2.COLOR_RGBA2GRAY)
-            depth_color = cv2.cvtColor(depth_gray, cv2.COLOR_GRAY2RGB)
+            depth_color = cv2.cvtColor(depth_gray, cv2.COLOR_GRAY2BGR)
 
             # Remove white border around map:
             # define lower and upper limits of white
@@ -147,7 +146,6 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             mask = cv2.inRange(depth_color, white_lo, white_hi)
             # change image to black where we found white
             depth_color[mask>0] = (0,0,0)
-            depth_gray = cv2.cvtColor(depth_color, cv2.COLOR_RGB2GRAY)
             
             # split_region = np.ones((frame_height, margin_width, 3), dtype=np.uint8) * 255
             # combined_frame = cv2.hconcat([raw_frame, split_region, depth_color])
@@ -159,13 +157,11 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             cv2.imwrite(f"f{count}.jpg", raw_frame)
             orig_frames.append(f"f{count}.jpg")
             cv2.imwrite(f"f{count}_dmap.jpg", depth_color)
-            cv2.imwrite(f"f{count}_dmap_gray.jpg", depth_gray)
             depth_frames.append(f"f{count}_dmap.jpg")
-            depth_frames_gray.append(f"f{count}_dmap_gray.jpg")
             count += 1
 
         final_vid = create_video(depth_frames, frame_rate, "depth")
-        final_zip = zip_files(orig_frames, depth_frames_gray)
+        final_zip = zip_files(orig_frames, depth_frames)
         raw_video.release()
         # out.release()
         cv2.destroyAllWindows()