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