Update app.py
Browse files
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=
|
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)
|