freealise commited on
Commit
597c171
·
verified ·
1 Parent(s): b33f76e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -198,7 +198,7 @@ def remove_bg(fl, count, mh, ms, md, lm):
198
  #fr[fr_fg==0] = (255,255,255)
199
 
200
  fr_fg[fr_fg>0] = 3 #probable fg
201
- mask, bgdModel, fgdModel = cv2.grabCut(fr, fr_fg, None,None,None,15, cv2.GC_INIT_WITH_MASK)
202
  mask = np.where((mask==2)|(mask==0),0,1).astype('uint8')
203
  fr[mask==0] = (255,255,255)
204
 
 
198
  #fr[fr_fg==0] = (255,255,255)
199
 
200
  fr_fg[fr_fg>0] = 3 #probable fg
201
+ mask, bgdModel, fgdModel = cv2.grabCut(fr, fr_fg, None,None,None,31, cv2.GC_INIT_WITH_MASK)
202
  mask = np.where((mask==2)|(mask==0),0,1).astype('uint8')
203
  fr[mask==0] = (255,255,255)
204