Update app.py

app.py

@@ -201,13 +201,14 @@ 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 = mask_bg[mask_no_shadow==0]
                     
                     depth_gray[cv2.bitwise_and(mask_no_shadow, mask_bg)>0] = 0
                     depth_color[cv2.bitwise_and(mask_no_shadow, mask_bg)>0] = (0,0,0)
                     
                     m = cv2.inRange(raw_frame, np.array([255,255,255]), np.array([255,255,255]))
-                    raw_frame[m>0] = (253,253,253)
-                    raw_frame[mask_bg>0] = (254,254,254)
+                    raw_frame[m>0] = (239,239,239)
+                    raw_frame[mask_shadow>0] = int(raw_frame / 17 + 240)
                     raw_frame[cv2.bitwise_and(mask_no_shadow, mask_bg)>0] = (255,255,255)
             
             cv2.imwrite(f"f{count}.png", raw_frame)