Update app.py
Browse files
app.py
CHANGED
@@ -580,7 +580,7 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
|
|
580 |
#bg = cv2.cvtColor(bg, cv2.COLOR_RGB2HSV)
|
581 |
#bg[:,:,0] = clrs[:,:,0]
|
582 |
#bg = cv2.cvtColor(bg, cv2.COLOR_HSV2RGB)
|
583 |
-
bd = cv2.convertScaleAbs(cv2.Laplacian(bg, cv2.CV_64F,
|
584 |
|
585 |
diff = np.abs(cv2.cvtColor(d["background"], cv2.COLOR_RGBA2GRAY).astype(np.int16)-cv2.cvtColor(gradient, cv2.COLOR_RGBA2GRAY).astype(np.int16)).astype(np.uint8)
|
586 |
mask = cv2.inRange(diff, 0, t)
|
@@ -606,7 +606,7 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
|
|
606 |
res = center[label.flatten()]
|
607 |
bg = res.reshape((bg.shape))
|
608 |
|
609 |
-
bg[bd>
|
610 |
bg[mask>0] = 0
|
611 |
bg[bg==255] = 0
|
612 |
|
|
|
580 |
#bg = cv2.cvtColor(bg, cv2.COLOR_RGB2HSV)
|
581 |
#bg[:,:,0] = clrs[:,:,0]
|
582 |
#bg = cv2.cvtColor(bg, cv2.COLOR_HSV2RGB)
|
583 |
+
bd = cv2.convertScaleAbs(cv2.Laplacian(bg, cv2.CV_64F, 7)).astype(np.uint8)
|
584 |
|
585 |
diff = np.abs(cv2.cvtColor(d["background"], cv2.COLOR_RGBA2GRAY).astype(np.int16)-cv2.cvtColor(gradient, cv2.COLOR_RGBA2GRAY).astype(np.int16)).astype(np.uint8)
|
586 |
mask = cv2.inRange(diff, 0, t)
|
|
|
606 |
res = center[label.flatten()]
|
607 |
bg = res.reshape((bg.shape))
|
608 |
|
609 |
+
bg[bd>4] = 0
|
610 |
bg[mask>0] = 0
|
611 |
bg[bg==255] = 0
|
612 |
|