Update app.py

app.py

@@ -197,11 +197,27 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', remove_bg
                     
                     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(np.average(diff_d))
-                    print(np.average(diff_c))
+                    avg_d = int(np.average(diff_d))
+                    avg_c = int(np.average(diff_c))
+                    print('-average')
+                    print(avg_d)
+                    print(avg_c)
+                    md_d = int(np.median(diff_d))
+                    md_c = int(np.median(diff_c))
+                    print('-median')
+                    print(md_d)
+                    print(md_c)
+                    print('-most common')
+                    c = Counter(diff_d)
+                    value, cc = c.most_common()[0]
+                    print(value)
+                    c = Counter(diff_c)
+                    value, cc = c.most_common()[0]
+                    print(value)
+                    print('-')
                     
-                    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_bg = cv2.inRange(diff_d, np.array([0,0,0]), np.array([avg_d,avg_d,avg_d]))
+                    mask_no_shadow = cv2.inRange(diff_c, np.array([0,0,0]), np.array([avg_c,avg_c,avg_c]))
                     mask_shadow = cv2.bitwise_not(mask_no_shadow)
 
                     #stereo = cv2.StereoBM.create(numDisparities=16, blockSize=15)