Update app.py

app.py

@@ -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)
+    mask = cv2.inRange(d["background"], np.array([255,255,255,255]), np.array([255,255,255,255]))
 
     for k, mk in enumerate(masks):
         if k != frame_selected:
             masks[k] = cv2.imread(depths[k])
-            masks[k][mask!=255] = (255,255,255)
+            masks[k][mask==0] = (255,255,255)
 
     masks[frame_selected] = d["background"]
     cv2.imwrite(f"f{frame_selected}_dmap.png", masks[frame_selected])