Update app.py

app.py

@@ -341,15 +341,20 @@ def draw_mask(v, d, evt: gr.EventData):
     w = int(vv[2])
     h = int(vv[3])
 
-    mask = np.zeros(d["background"].shape[:2], np.uint8)
-    bgdModel = np.zeros((1,65), np.float64)
-    fgdModel = np.zeros((1,65), np.float64)
-    rect = (x, y, w, h)
-    mask = cv2.grabCut(d["background"], mask, rect, bgdModel, fgdModel, 5, cv2.GC_INIT_WITH_RECT)
-    print(mask)
-    mask0 = mask[0]
-    newmask = np.where((mask0==2)|(mask0==0),1,0).astype('uint8')
-    d["composite"] = rgb*newmask[:,:,np.newaxis]
+    scale = 1
+    delta = 0
+    ddepth = cv.CV_16S
+
+    gray = cv.cvtColor(d["background"], cv.COLOR_BGR2GRAY)
+    grad_x = cv.Sobel(gray, ddepth, 1, 0, ksize=5, scale=scale, delta=delta, borderType=cv.BORDER_DEFAULT)
+    # Gradient-Y
+    # grad_y = cv.Scharr(gray,ddepth,0,1)
+    grad_y = cv.Sobel(gray, ddepth, 0, 1, ksize=5, scale=scale, delta=delta, borderType=cv.BORDER_DEFAULT)
+    abs_grad_x = cv.convertScaleAbs(grad_x)
+    abs_grad_y = cv.convertScaleAbs(grad_y)
+
+    grad = cv.addWeighted(abs_grad_x, 0.5, abs_grad_y, 0.5, 0)
+    d["composite"] = cv.cvtColor(255-grad, cv.COLOR_GRAY2RGB)
     
     return gr.ImageEditor(value=d)