Update app.py

app.py

@@ -202,11 +202,17 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', remove_bg
                     mask_bg = cv2.inRange(diff_d, np.array([0,0,0]), np.array([16,16,16]))
                     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(raw_frame)
                     
                     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[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)
+                    shadowMask = cv2.inRange(fgMask, np.array([1,1,1]), np.array([254,254,254]))
+                    raw_frame[shadowMask>0] = (raw_frame[shadowMask>0] / 17 + 240).astype(np.uint8)
+                    raw_frame[fgMask==0] = (255,255,255)
             
             cv2.imwrite(f"f{count}.png", raw_frame)
             orig_frames.append(f"f{count}.png")