Update app.py

app.py

@@ -222,6 +222,15 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
               thumbnail_old = thumbnail
 
             blur_frame = blur_image(raw_frame, depth_color, blur_data)
+
+            # encoding depth within original video
+            blur_frame = (round(blur_frame / 17) * 17).astype(np.uint8)
+            depth_r = round(depth_gray / 17).astype(np.uint8)
+            # may use green channel for 16 levels of opacity
+            depth_b = depth_gray - depth_r * 17
+            blur_frame[:,:,0] = blur_frame[:,:,0] + depth_r
+            # blur_frame[:,:,1] = blur_frame[:,:,1] + opacity_g
+            blur_frame[:,:,2] = blur_frame[:,:,2] + depth_b
             
             cv2.imwrite(f"f{count}.jpg", blur_frame)
             orig_frames.append(f"f{count}.jpg")