freealise commited on
Commit
8e1625f
·
verified ·
1 Parent(s): 79610f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, 5)).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,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>b] = 0
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