Update app.py

app.py

@@ -135,8 +135,6 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', remove_bg
         orig_frames = []
         thumbnail_old = []
 
-        backSub = cv2.createBackgroundSubtractorKNN()
-
         while raw_video.isOpened():
             ret, raw_frame = raw_video.read()
             if not ret:
@@ -206,16 +204,13 @@ 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)
 
-                    fgMask = backSub.apply(raw_frame)
+                    #stereo = cv2.StereoBM.create(numDisparities=16, blockSize=15)
+                    #disparity = stereo.compute(raw_frame_l, raw_frame_r)
                     
                     m = cv2.inRange(raw_frame, np.array([255,255,255]), np.array([255,255,255]))
                     raw_frame[m>0] = (239,239,239)
-                    #raw_frame[cv2.bitwise_and(mask_shadow, mask_bg)>0] = (raw_frame[cv2.bitwise_and(mask_shadow, mask_bg)>0] / 17 + 240).astype(np.uint8)
-                    #raw_frame[cv2.bitwise_and(mask_no_shadow, mask_bg)>0] = (255,255,255)
-                    raw_frame[fgMask==127] = (raw_frame[fgMask==127] / 17 + 240).astype(np.uint8)
-                    raw_frame[fgMask==0] = (255,255,255)
-                    depth_gray = fgMask
-                    depth_color = cv2.cvtColor(fgMask, cv2.COLOR_GRAY2BGR)
+                    raw_frame[cv2.bitwise_and(mask_shadow, mask_bg)>0] = (raw_frame[cv2.bitwise_and(mask_shadow, mask_bg)>0] / 17 + 240).astype(np.uint8)
+                    raw_frame[cv2.bitwise_and(mask_no_shadow, mask_bg)>0] = (255,255,255)
             
             cv2.imwrite(f"f{count}.png", raw_frame)
             orig_frames.append(f"f{count}.png")