Update app.py

app.py

@@ -219,7 +219,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
                       continue
               thumbnail_old = thumbnail
 
-            blur_frame = cv2.cvtColor(blur_image(raw_frame, depth_color, blur_data), cv2.COLOR_BGR2BGRA)
+            blur_frame = blur_image(cv2.cvtColor(raw_frame, cv2.COLOR_BGR2BGRA), depth_color, blur_data)
             comb_frame = np.concatenate((blur_frame, depth_color), axis=0)
 
             cv2.imwrite(f"f{count}_comb.png", comb_frame)
@@ -289,8 +289,8 @@ def blur_image(image, depth, blur_data):
     j = 0
     while j < 256:
         i = 255 - j
-        blur_lo = np.array([i,i,i])
-        blur_hi = np.array([i+1,i+1,i+1])
+        blur_lo = np.array([i,i,i,255])
+        blur_hi = np.array([i+1,i+1,i+1,255])
         blur_mask = cv2.inRange(depth, blur_lo, blur_hi)
         
         #print(f'kernel size {int(blur_a[j])}')
@@ -298,9 +298,6 @@ def blur_image(image, depth, blur_data):
                 
         blur_frame[blur_mask>0] = blur[blur_mask>0]
         j = j + 1
-
-    white = cv2.inRange(blur_frame, np.array([255,255,255]), np.array([255,255,255]))
-    blur_frame[white>0] = (254,254,254)
     
     return blur_frame
 
@@ -448,8 +445,11 @@ def draw_mask(o, b, v, d, evt: gr.EventData):
     frame = cv2.imread(frames[frame_selected], cv2.IMREAD_UNCHANGED).astype(np.uint8)
     frame[mask>0] = (0,0,0,0)
     cv2.imwrite(frames[frame_selected], frame)
+
+    depth = cv2.imread(depths[frame_selected], cv2.IMREAD_UNCHANGED).astype(np.uint8)
+    depth[mask>0] = (0,0,0,0)
+    cv2.imwrite(depths[frame_selected], depth)
     
-    switch_rows(False)
     return gr.ImageEditor(value=d)
 
 js = """