freealise commited on
Commit
e235e21
·
verified ·
1 Parent(s): 548376b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -200,7 +200,7 @@ def remove_bg(fl, count, mh, ms, md, lm, backSub):
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
- fgMask = backSub.apply(fr, learningRate=0)
204
  fr[fgMask==0] = (255,255,255)
205
 
206
  cv2.imwrite(fl, fr)
 
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
+ fgMask = backSub.apply(fr, learningRate=1)
204
  fr[fgMask==0] = (255,255,255)
205
 
206
  cv2.imwrite(fl, fr)