Update app.py

app.py

@@ -145,9 +145,11 @@ def logscale(linear):
 def linscale(linear):
     return int(math.log2(linear))
 
-def remove_bg(fl, count, mh, ms, md, lm):
+def remove_bg(fl, count, mh, ms, md, lm, backSub):
     global fl_
     fr = cv2.imread(fl).astype(np.uint8)
+
+    fgMask = backSub.apply(fr)
     
     #b = 3
     #element = cv2.getStructuringElement(cv2.MORPH_RECT, (2 * b + 1, 2 * b + 1), (b, b))
@@ -200,7 +202,7 @@ def remove_bg(fl, count, mh, ms, md, lm):
         fr_fg[fr_fg>0] = 3 #probable fg
         mask, bgdModel, fgdModel = cv2.grabCut(fr, fr_fg, None,None,None,31, cv2.GC_INIT_WITH_MASK)
         mask = np.where((mask==2)|(mask==0),0,1).astype('uint8')
-        fr[mask==0] = (255,255,255)
+        fr[fgMask==0] = (255,255,255)
         
         cv2.imwrite(fl, fr)
         return fl
@@ -257,6 +259,9 @@ def loadf(f, r_bg, mh, ms, md, lm):
         f.sort(key=sortFiles)
         fnew = []
 
+        backSub = cv2.createBackgroundSubtractorMOG2()
+        #backSub = cv2.createBackgroundSubtractorKNN()
+
         for i, fl in enumerate(f):
             ftype = fl.split('/')
             if ftype[len(ftype)-1].split('.')[1] == 'mp4':
@@ -265,7 +270,7 @@ def loadf(f, r_bg, mh, ms, md, lm):
                 fl = denoise(fl)
 
             if r_bg == True:
-                fl = remove_bg(fl, i, mh, ms, md, lm)
+                fl = remove_bg(fl, i, mh, ms, md, lm, backSub)
                 if i % 2: # odd: is photo without the flash
                     fnew.append(fl)
             else: