Update app.py

app.py

@@ -195,8 +195,8 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
             # change image to black where we found white
             depth_gray[0:int(depth_gray.shape[0]/8*7)-1, 0:depth_gray.shape[1]][mask>0] = 0
 
-            mask = cv2.inRange(depth_gray[int(depth_gray.shape[0]/8*7):depth_gray.shape[0], 0:depth_gray.shape[1]], 192, 255)
-            depth_gray[int(depth_gray.shape[0]/8*7):depth_gray.shape[0], 0:depth_gray.shape[1]][mask>0] = 192
+            mask = cv2.inRange(depth_gray[int(depth_gray.shape[0]/8*7):depth_gray.shape[0], 0:depth_gray.shape[1]], 125, 255)
+            depth_gray[int(depth_gray.shape[0]/8*7):depth_gray.shape[0], 0:depth_gray.shape[1]][mask>0] = 125
 
             depth_color = cv2.cvtColor(depth_gray, cv2.COLOR_GRAY2BGR)
             # split_region = np.ones((frame_height, margin_width, 3), dtype=np.uint8) * 255
@@ -224,10 +224,11 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
             blur_frame = blur_image(raw_frame, depth_color, blur_data)
 
             # encoding depth within original video
-            blur_frame = (np.floor(blur_frame / 6) * 6).astype(np.uint8)
-            depth_r = np.floor(depth_gray / 6).astype(np.uint8)
+            dd = 5
+            blur_frame = (np.floor(blur_frame / dd) * dd).astype(np.uint8)
+            depth_r = np.floor(depth_gray / dd).astype(np.uint8)
             # may use green channel for 16 levels of opacity
-            depth_b = depth_gray - depth_r * 6
+            depth_b = depth_gray - depth_r * dd
             blur_frame[:,:,0] = blur_frame[:,:,0] + depth_r
             # blur_frame[:,:,1] = blur_frame[:,:,1] + opacity_g
             blur_frame[:,:,2] = blur_frame[:,:,2] + depth_b