Update app.py

app.py

@@ -387,8 +387,10 @@ def reset_mask():
     global frame_selected
     global masks
     global depths
+    global edge
 
-    masks[frame_selected] = cv2.imread(depths[frame_selected])
+    edge = []
+    masks[frame_selected] = cv2.imread(depths[frame_selected]).astype(np.uint8)
     cv2.imwrite(f"f{frame_selected}_dmap.png", masks[frame_selected])
     return masks[frame_selected], f"f{frame_selected}_dmap.png"
 
@@ -405,7 +407,7 @@ def apply_mask(d, b):
 
     for k, mk in enumerate(masks):
         if k != frame_selected:
-            masks[k] = cv2.imread(depths[k])
+            masks[k] = cv2.imread(depths[k]).astype(np.uint8)
             masks[k][mask==0] = (255,255,255)
 
     masks[frame_selected] = d["background"]
@@ -450,7 +452,7 @@ def draw_mask(l, t, v, d, evt: gr.EventData):
       #for i in range(l):
       #    bg[(bg >= i*255/l) & (bg < (i+1)*255/l)] = i*255/(l-1)
       bg = cv2.convertScaleAbs(palette[bg]).astype(np.uint8) # Converting image back to uint
-      bg[mask>0] = 254
+      bg[mask>0] = 0
       bg[bg==255] = 254
 
       params["fnum"] = frame_selected