Update app.py

app.py

@@ -193,11 +193,14 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', remove_bg
                     n = 0 #n = count-int(cframes/2)
                     depth_color_bg = cv2.imread(f"f{n}_dmap.png").astype(np.uint8)
                     raw_frame_bg = cv2.imread(f"f{n}.png").astype(np.uint8)
-                    print(np.abs(depth_color.astype(np.int16)-depth_color_bg.astype(np.int16)))
-                    print(np.abs(raw_frame_bg.astype(np.int16)-raw_frame.astype(np.int16)))
                     
-                    mask_bg = cv2.inRange(np.abs(depth_color.astype(np.int16)-depth_color_bg.astype(np.int16)), np.array([0,0,0]), np.array([32,32,32]))
-                    mask_no_shadow = cv2.inRange(np.abs(raw_frame_bg.astype(np.int16)-raw_frame.astype(np.int16)), np.array([0,0,0]), np.array([4,4,4]))
+                    diff_d = np.abs(depth_color.astype(np.int16)-depth_color_bg.astype(np.int16))
+                    diff_c = np.abs(raw_frame.astype(np.int16)-raw_frame_bg.astype(np.int16))
+                    print(diff_d)
+                    print(diff_c)
+                    
+                    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([2,2,2]))
                     
                     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)