Update app.py

app.py

@@ -134,6 +134,9 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', remove_bg
         depth_frames = []
         orig_frames = []
         thumbnail_old = []
+
+        backSub = cv2.createBackgroundSubtractorMOG2()
+
         while raw_video.isOpened():
             ret, raw_frame = raw_video.read()
             if not ret:
@@ -203,8 +206,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', remove_bg
                     mask_no_shadow = cv2.inRange(diff_c, np.array([0,0,0]), np.array([16,16,16]))
                     mask_shadow = cv2.bitwise_not(mask_no_shadow)
 
-                    backSub = cv2.createBackgroundSubtractorMOG2()
-                    fgMask = backSub.apply(cv2.cvtColor(raw_frame, cv2.COLOR_BGR2RGB))
+                    fgMask = backSub.apply(raw_frame)
                     
                     m = cv2.inRange(raw_frame, np.array([255,255,255]), np.array([255,255,255]))
                     raw_frame[m>0] = (239,239,239)