Update app.py

app.py

@@ -15,7 +15,6 @@ from vincenty import vincenty
 import json
 from collections import Counter
 import mediapy
-from pymatting import cutout
 
 #from depth_anything.dpt import DepthAnything
 #from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
@@ -234,26 +233,10 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
             depth_frames.append(f"f{count}_dmap.png")
 
             depth_gray = seg_frame(depth_gray, o, b) + 128
-            print(depth_gray[depth_gray>128]-128)
+            #print(depth_gray[depth_gray>128]-128)
 
             cv2.imwrite(f"f{count}_mask.png", depth_gray)
-            masks.append(f"f{count}_mask.png")
-            
-            depth_gray_trimap = np.zeros(depth_gray.shape[:2], np.uint8)
-            depth_gray_trimap[depth_gray==128] = 0
-            depth_gray_trimap[depth_gray==129] = 255
-            depth_gray_trimap[depth_gray==131] = 127
-
-            cv2.imwrite(f"f{count}_trimap.png", depth_gray_trimap)
-
-            cutout(
-   # input image path
-   f"f{count}.png",
-   # input trimap path
-   f"f{count}_trimap.png",
-   # output cutout path
-   f"f{count}.png")
-            
+            masks.append(f"f{count}_mask.png")            
             count += 1
 
         final_vid = create_video(orig_frames, frame_rate, "orig")
@@ -410,7 +393,7 @@ def get_mesh(image, depth, blur_data, loadall):
 
 def blur_image(image, depth, blur_data):
     blur_a = blur_data.split()
-    print(f'blur data {blur_data}')
+    #print(f'blur data {blur_data}')
 
     blur_frame = image.copy()
     j = 0