Update app.py

app.py

@@ -111,15 +111,15 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         #out = cv2.VideoWriter(output_path, fourcc, frame_rate, (output_width, frame_height))
         count=0
         depth_frames = []
+        gdepth_frames = []
         orig_frames = []
         while raw_video.isOpened():
             ret, raw_frame = raw_video.read()
             if not ret:
                 break
 
-            raw_frame = cv2.cvtColor(raw_frame, cv2.COLOR_BGR2RGB).astype(np.uint8)
-            frame = raw_frame / 255.0
-            frame_pil =  Image.fromarray(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']
             
             frame = torch.from_numpy(frame).unsqueeze(0).to(DEVICE)
@@ -132,8 +132,8 @@ 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_color = cv2.cvtColor(depth_color, cv2.COLOR_RGBA2GRAY)
-            depth_color = cv2.cvtColor(depth_color, cv2.COLOR_GRAY2RGB)
+            depth_gray = cv2.cvtColor(depth_color, cv2.COLOR_RGBA2GRAY)
+            depth_color = cv2.cvtColor(depth_gray, cv2.COLOR_GRAY2RGB)
 
             # Remove white border around map:
             # define lower and upper limits of white
@@ -155,6 +155,8 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             orig_frames.append(f"f{count}.jpg")
             cv2.imwrite(f"f{count}_dmap.jpg", depth_color)
             depth_frames.append(f"f{count}_dmap.jpg")
+            cv2.imwrite(f"f{count}_gmap.jpg", depth_gray)
+            gdepth_frames.append(f"f{count}_gmap.jpg")
             count += 1
 
         final_vid = create_video(depth_frames, frame_rate, "depth")
@@ -163,7 +165,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         # out.release()
         cv2.destroyAllWindows()
         
-        return final_vid, final_zip, orig_frames[0], depth_frames[0] #output_path
+        return final_vid, final_zip, orig_frames[0], gdepth_frames[0] #output_path
 
 def depth_edges_mask(depth):
     """Returns a mask of edges in the depth map.