Update app.py

app.py

@@ -170,7 +170,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             
             cv2.imwrite(f"f{count}_dmap.png", depth_color)
             depth_frames.append(f"f{count}_dmap.png")
-            masks.append(depth_color.copy())
+            masks.append(f"f{count}_dmap.png")
             count += 1
 
         final_vid = create_video(depth_frames, frame_rate, "depth")
@@ -401,12 +401,12 @@ def apply_mask(d, b):
     dilation = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * b + 1, 2 * b + 1), (b, b))
     mask = cv2.dilate(mask, dilation)
     d["background"][mask==255] = (255,255,255,255)
+    mask = cv2.cvtColor(d["background"], cv2.COLOR_RGBA2GRAY)
 
     for k, mk in enumerate(masks):
         if k != frame_selected:
-            m = masks[k].copy()
-            m[mask!=255] = (255,255,255)
-            masks[k] = m
+            masks[k] = cv2.imread(depths[k])
+            masks[k][mask!=255] = (255,255,255)
 
     masks[frame_selected] = d["background"]
     cv2.imwrite(f"f{frame_selected}_dmap.png", masks[frame_selected])