Update app.py

app.py

@@ -114,6 +114,15 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             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)
+
+            # Remove white border around map:
+            # define lower and upper limits of white
+            white_lo = np.array([255,255,255])
+            white_hi = np.array([255,255,255])
+            # mask image to only select white
+            mask = cv2.inRange(depth_color, white_lo, white_hi)
+            # change image to black where we found white
+            depth_color[mask>0] = (0,0,0)
             
             # split_region = np.ones((frame_height, margin_width, 3), dtype=np.uint8) * 255
             # combined_frame = cv2.hconcat([raw_frame, split_region, depth_color])