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")
@@ -388,9 +388,8 @@ def reset_mask():
     global masks
     global depths
 
-    masks[frame_selected] = cv2.imread(depths[frame_selected])
-    cv2.imwrite(f"f{frame_selected}_dmap.png", masks[frame_selected])
-    return masks[frame_selected], f"f{frame_selected}_dmap.png"
+    cv2.imwrite(masks[frame_selected], cv2.imread(depths[frame_selected]))
+    return masks[frame_selected], masks[frame_selected]
 
 def apply_mask(d, b):
     global frame_selected
@@ -402,13 +401,14 @@ def apply_mask(d, b):
     mask = cv2.dilate(mask, dilation)
     d["background"][mask==255] = (255,255,255,255)
 
-    for k, m in enumerate(masks):
+    for k, mk in enumerate(masks):
         if k != frame_selected:
-            masks[k][mask!=255] = (255,255,255)
+            m = cv2.imread(masks[k])
+            m[mask!=255] = (255,255,255)
+            cv2.imwrite(masks[k], m)
     
-    masks[frame_selected] = d["background"]
-    cv2.imwrite(f"f{frame_selected}_dmap.png", masks[frame_selected])
-    return masks[frame_selected], f"f{frame_selected}_dmap.png"
+    cv2.imwrite(masks[frame_selected], d["background"])
+    return masks[frame_selected], masks[frame_selected]
 
 def draw_mask(l, t, v, d, evt: gr.EventData):
     global params