Update app.py

app.py

@@ -159,8 +159,8 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             # change image to black where we found white
             depth_gray[mask>0] = 0
 
-            mask = cv2.inRange(depth_gray[int(depth_gray.shape[0]/8*6.5):depth_gray.shape[0]-1, 0:depth_gray.shape[1]], 160, 255)
-            depth_gray[int(depth_gray.shape[0]/8*6.5):depth_gray.shape[0]-1, 0:depth_gray.shape[1]][mask>0] = 160
+            mask = cv2.inRange(depth_gray[int(depth_gray.shape[0]/8*6.5):depth_gray.shape[0], 0:depth_gray.shape[1]], 160, 255)
+            depth_gray[int(depth_gray.shape[0]/8*6.5):depth_gray.shape[0], 0:depth_gray.shape[1]][mask>0] = 160
 
             depth_color = cv2.cvtColor(depth_gray, cv2.COLOR_GRAY2BGR)
             # split_region = np.ones((frame_height, margin_width, 3), dtype=np.uint8) * 255
@@ -549,8 +549,9 @@ def draw_mask(l, t, v, d, evt: gr.EventData):
 
     x = points[len(points)-1][0]
     y = points[len(points)-1][1]
-    
-    mask = cv2.floodFill(bg, None, (x, y), 1, 0, int(t*256/l), (4 | cv2.FLOODFILL_FIXED_RANGE))[2] #(4 | cv2.FLOODFILL_FIXED_RANGE | cv2.FLOODFILL_MASK_ONLY | 255 << 8)
+
+                                                #int(t*256/l)
+    mask = cv2.floodFill(bg, None, (x, y), 1, 0, 256, (4 | cv2.FLOODFILL_FIXED_RANGE))[2] #(4 | cv2.FLOODFILL_FIXED_RANGE | cv2.FLOODFILL_MASK_ONLY | 255 << 8)
     # 255 << 8 tells to fill with the value 255)
     mask = mask[1:mask.shape[0]-1, 1:mask.shape[1]-1]