Update app.py

app.py

@@ -344,20 +344,31 @@ def draw_mask(ksize, th, v, d, evt: gr.EventData):
     w = int(vv[2])
     h = int(vv[3])
 
-    scale = 1
-    delta = 0
-    ddepth = cv2.CV_16S
+    src = d["background"]
+    src_gray = cv2.cvtColor(src, cv2.COLOR_BGR2GRAY)
+    ratio = 3
+    kernel_size = ksize
+    low_threshold = th
+    img_blur = cv.blur(src_gray, (3,3))
+    detected_edges = cv.Canny(img_blur, low_threshold, low_threshold*ratio, kernel_size)
+    mask = detected_edges != 0
+    dst = src * (mask[:,:,None].astype(src.dtype))
+    
+
+    #scale = 1
+    #delta = 0
+    #ddepth = cv2.CV_16S
 
-    gray = cv2.cvtColor(d["background"], cv2.COLOR_BGR2GRAY)
-    grad_x = cv2.Sobel(gray, ddepth, 1, 0, ksize=ksize, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT)
+    #gray = cv2.cvtColor(d["background"], cv2.COLOR_BGR2GRAY)
+    #grad_x = cv2.Sobel(gray, ddepth, 1, 0, ksize=ksize, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT)
     # Gradient-Y
     # grad_y = cv2.Scharr(gray,ddepth,0,1)
-    grad_y = cv2.Sobel(gray, ddepth, 0, 1, ksize=ksize, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT)
-    abs_grad_x = cv2.convertScaleAbs(grad_x)
-    abs_grad_y = cv2.convertScaleAbs(grad_y)
+    #grad_y = cv2.Sobel(gray, ddepth, 0, 1, ksize=ksize, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT)
+    #abs_grad_x = cv2.convertScaleAbs(grad_x)
+    #abs_grad_y = cv2.convertScaleAbs(grad_y)
 
-    grad = cv2.addWeighted(abs_grad_x, 0.5, abs_grad_y, 0.5, 0)
-    bw = cv2.threshold(255-grad, th, 255, cv2.THRESH_BINARY)[1]
+    #grad = cv2.addWeighted(abs_grad_x, 0.5, abs_grad_y, 0.5, 0)
+    bw = cv2.threshold(255-dst, th, 255, cv2.THRESH_BINARY)[1]
 
     d["background"] = cv2.cvtColor(bw, cv2.COLOR_GRAY2RGB)
     d["composite"] = cv2.cvtColor(bw, cv2.COLOR_GRAY2RGB)