Update app.py

app.py

@@ -300,7 +300,7 @@ def get_mesh(image, depth, blur_data, loadall):
         i = i+1
 
     for i in range(0,len(verts)):
-        if verts[i] == [0.0,0.0,0.0]:
+        if verts[i][0] == 0.0 and verts[i][1] == 0.0 and verts[i][2] == 0.0:
             verts.pop(i)
             clrs.pop(i)
         i=i+1
@@ -374,7 +374,7 @@ def apply_mask(d):
     cv2.imwrite(f"f{frame_selected}_dmap.png", masks[frame_selected])
     return masks[frame_selected], f"f{frame_selected}_dmap.png"
 
-def draw_mask(l, v, d, evt: gr.EventData):
+def draw_mask(l, t, v, d, evt: gr.EventData):
     global params
     global frame_selected
     global masks
@@ -396,10 +396,10 @@ def draw_mask(l, v, d, evt: gr.EventData):
       bg = cv2.cvtColor(d["background"], cv2.COLOR_RGBA2GRAY)
 
       diff = (bg-cv2.cvtColor(gradient, cv2.COLOR_RGBA2GRAY)).astype(np.uint8)
-      mask = cv2.inRange(diff, -1, 1)
+      mask = cv2.inRange(diff, -t, t)
       #kernel = np.ones((c,c),np.float32)/(c*c)
       #mask = cv2.filter2D(mask,-1,kernel)
-      grad = cv2.convertScaleAbs(cv2.Sobel(mask, ddepth, 1, 1, ksize=3, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT))
+      grad = cv2.convertScaleAbs(cv2.Sobel(mask, ddepth, 1, 1, ksize=15-(t*2), scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT))
 
       mask = mask + cv2.inRange(grad, 1, 255)
 
@@ -425,7 +425,7 @@ def draw_mask(l, v, d, evt: gr.EventData):
     x = points[len(points)-1][0]
     y = points[len(points)-1][1]
     
-    mask = cv2.floodFill(bg, None, (x, y), 1, 32, 32, (4 | cv2.FLOODFILL_FIXED_RANGE))[2] #(4 | cv2.FLOODFILL_FIXED_RANGE | cv2.FLOODFILL_MASK_ONLY | 255 << 8)
+    mask = cv2.floodFill(bg, None, (x, y), 1, int(t*256/l), int(t*256/l), (4 | cv2.FLOODFILL_FIXED_RANGE))[2] #(4 | cv2.FLOODFILL_FIXED_RANGE | cv2.FLOODFILL_MASK_ONLY | 255 << 8)
     # 255 << 8 tells to fill with the value 255)
     mask = mask[1:mask.shape[0]-1, 1:mask.shape[1]-1]
     
@@ -488,8 +488,9 @@ with gr.Blocks(css=css) as demo:
             output_mask.apply(fn=apply_mask, inputs=[output_mask], outputs=[output_mask, depth_file])
             with gr.Accordion(label="Edge", open=False):
               levels = gr.Slider(label="Color levels", value=16, maximum=32, minimum=2, step=1)
+              tolerance = gr.Slider(label="Tolerance", value=1, maximum=7, minimum=0, step=1)
               mouse = gr.Textbox(elem_id="mouse", value="""[]""", interactive=False)
-              mouse.input(fn=draw_mask, show_progress="minimal", inputs=[levels, mouse, output_mask], outputs=[output_mask])
+              mouse.input(fn=draw_mask, show_progress="minimal", inputs=[levels, tolerance, mouse, output_mask], outputs=[output_mask])
             selector = gr.HTML(value="""
             <a href='#' id='selector' onclick='this.style.fontWeight=\"bold\";
             document.getElementById(\"image_edit\").getElementsByTagName(\"canvas\")[0].oncontextmenu = function(e){e.preventDefault();}