Update app.py

app.py

@@ -180,14 +180,14 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         depths = depth_frames
 
         if depth_color.shape[0] == 4096:
-            masks[len(masks)-1] = cv2.imread('./gradient_large.png')
-            depth_frames[len(masks)-1] = cv2.imread('./gradient_large.png')
+            masks[len(masks)-1] = './gradient_large.png'
+            depth_frames[len(masks)-1] = './gradient_large.png'
         elif depth_color.shape[0] == 2048:
-            masks[len(masks)-1] = cv2.imread('./gradient.png')
-            depth_frames[len(masks)-1] = cv2.imread('./gradient.png')
+            masks[len(masks)-1] = './gradient.png'
+            depth_frames[len(masks)-1] = './gradient.png'
         else:
-            masks[len(masks)-1] = cv2.imread('./gradient_small.png')
-            depth_frames[len(masks)-1] = cv2.imread('./gradient_small.png')
+            masks[len(masks)-1] = './gradient_small.png'
+            depth_frames[len(masks)-1] = './gradient_small.png'
         
         return final_vid, final_zip, frames, masks[frame_selected] #output_path
 
@@ -354,7 +354,7 @@ def draw_mask(th, v, d, evt: gr.EventData):
     pts = pts.reshape((-1,1,2))
 
     bg = cv2.cvtColor(d["background"], cv2.COLOR_RGB2GRAY)
-    gd = ((cv2.cvtColor(masks[len(masks)-1], cv2.COLOR_RGB2GRAY)-th)*2).astype(np.uint8)
+    gd = ((cv2.cvtColor(cv2.imread(masks[len(masks)-1]).astype(np.uint8), cv2.COLOR_RGB2GRAY)-th)*2)
     diff = abs(bg-gd).astype(np.uint8)
     mask = cv2.inRange(diff, 0, 5)
     d["background"][mask>0] = (0,0,0,255)