freealise commited on
Commit
1814958
1 Parent(s): a6a60c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)