Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -432,6 +432,7 @@ def draw_mask(l, c, k, v, d, evt: gr.EventData):
|
|
432 |
|
433 |
grad = cv2.addWeighted(grad_x, 0.5, grad_y, 0.5, 0)
|
434 |
mask[cv2.inRange(grad, 1, 255)>0] = 1
|
|
|
435 |
|
436 |
d["layers"][1][mask>0] = (0,0,0,255)
|
437 |
d["background"][mask>0] = (0,0,0,255)
|
|
|
432 |
|
433 |
grad = cv2.addWeighted(grad_x, 0.5, grad_y, 0.5, 0)
|
434 |
mask[cv2.inRange(grad, 1, 255)>0] = 1
|
435 |
+
mask = mask[1:mask.shape[0]-1, 1:mask.shape[1]-1]
|
436 |
|
437 |
d["layers"][1][mask>0] = (0,0,0,255)
|
438 |
d["background"][mask>0] = (0,0,0,255)
|