Update app.py

app.py

@@ -111,7 +111,6 @@ 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()
@@ -155,8 +154,6 @@ 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")
@@ -165,7 +162,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         # out.release()
         cv2.destroyAllWindows()
         
-        return final_vid, final_zip, orig_frames[0], gdepth_frames[0] #output_path
+        return final_vid, final_zip, orig_frames[0], depth_frames[0] #output_path
 
 def depth_edges_mask(depth):
     """Returns a mask of edges in the depth map.
@@ -210,8 +207,11 @@ def pano_depth_to_world_points(depth):
 
     return pts3d
 
+def rgb2gray(rgb):
+    return np.dot(rgb[...,:3], [0.33, 0.33, 0.33])
+
 def get_mesh(image, depth, keep_edges=False):
-    pts3d = pano_depth_to_world_points(depth)
+    pts3d = pano_depth_to_world_points(rgb2gray(depth))
     print('radius from depth - ok')
 
     # Create a trimesh mesh from the points