Update app.py

app.py

@@ -87,18 +87,10 @@ def get_frames(video_in, step, name, n):
 
 
 def create_video(frames, fps, type):
-
-    backSub = cv2.createBackgroundSubtractorMOG2()
-    #backSub = cv2.createBackgroundSubtractorKNN()
-    
     print("building video result")
     imgs = []
     for j, img in enumerate(frames):
-        frame = cv2.cvtColor(cv2.imread(img).astype(np.uint8), cv2.COLOR_BGR2RGB)
-        fgMask = backSub.apply(frame)
-        frame[fgMask==255] = (255,255,255)
-        frame[fgMask==0] = (0,0,0)
-        imgs.append(frame)
+        imgs.append(cv2.cvtColor(cv2.imread(img).astype(np.uint8), cv2.COLOR_BGR2RGB))
 
     mediapy.write_video(type + "_result.mp4", imgs, fps=fps)
     return type + "_result.mp4"
@@ -206,9 +198,9 @@ def remove_bg(fl, count, mh, ms, md, lm):
         #fr[fr_fg==0] = (255,255,255)
 
         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, bgdModel, fgdModel = cv2.grabCut(fr, fr_fg, None,None,None,15, cv2.GC_INIT_WITH_MASK)
         mask = np.where((mask==2)|(mask==0),0,1).astype('uint8')
-        #fr[mask==0] = (255,255,255)
+        fr[mask==0] = (255,255,255)
         
         cv2.imwrite(fl, fr)
         return fl