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]], 125, 255)
-            depth_gray[int(depth_gray.shape[0]/8*7):depth_gray.shape[0], 0:depth_gray.shape[1]][mask>0] = 125
+            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
 
             depth_color = cv2.cvtColor(depth_gray, cv2.COLOR_GRAY2BGR)
             # split_region = np.ones((frame_height, margin_width, 3), dtype=np.uint8) * 255
@@ -225,12 +225,18 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
 
             # encoding depth within original video
             dd = 5
+            cd = 255 / dd
+            mask = cv2.inRange(blur_frame, 255, 255)
+            blur_frame[mask>0] = np.array([254,254,254])
             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 * dd
+            
+            depth_comp = np.round(depth_gray / 2).astype(np.uint8) # 0-96
+            depth_r = np.floor(depth_comp / cd).astype(np.uint8)
+            depth_g = np.floor((depth_comp - depth_r * cd) / cd).astype(np.uint8)
+            depth_b = depth_comp - depth_r * cd - depth_g * cd
+            
             blur_frame[:,:,0] = blur_frame[:,:,0] + depth_r
-            # blur_frame[:,:,1] = blur_frame[:,:,1] + opacity_g
+            blur_frame[:,:,1] = blur_frame[:,:,1] + depth_g
             blur_frame[:,:,2] = blur_frame[:,:,2] + depth_b
             
             cv2.imwrite(f"f{count}.jpg", blur_frame)