Update app.py

app.py

@@ -12,19 +12,22 @@ from utils.dc_utils import read_video_frames, save_video
 from huggingface_hub import hf_hub_download
 
 # Examples for the Gradio Demo.
-# Each example now contains 8 parameters in the following order:
+# Each example now contains 8 parameters:
 # [video_path, max_len, target_fps, max_res, stitch, grayscale, convert_from_color, blur]
 examples = [
-    ['assets/example_videos/davis_rollercoaster.mp4', -1, -1, 1280, True, True, True, 0.5],
-    ['assets/example_videos/Tokyo-Walk_rgb.mp4', -1, -1, 1280, True, True, True, 0.5],
-    ['assets/example_videos/4158877-uhd_3840_2160_30fps_rgb.mp4', -1, -1, 1280, True, True, True, 0.5],
-    ['assets/example_videos/4511004-uhd_3840_2160_24fps_rgb.mp4', -1, -1, 1280, True, True, True, 0.5],
-    ['assets/example_videos/1753029-hd_1920_1080_30fps.mp4', -1, -1, 1280, True, True, True, 0.5],
-    ['assets/example_videos/davis_burnout.mp4', -1, -1, 1280, True, True, True, 0.5],
-    ['assets/example_videos/example_5473765-l.mp4', -1, -1, 1280, True, True, True, 0.5],
-    ['assets/example_videos/Istanbul-26920.mp4', -1, -1, 1280, True, True, True, 0.5],
-    ['assets/example_videos/obj_1.mp4', -1, -1, 1280, True, True, True, 0.5],
-    ['assets/example_videos/sheep_cut1.mp4', -1, -1, 1280, True, True, True, 0.5],
+    ['assets/example_videos/octopus_01.mp4', -1, -1, 1280, True, True, True, 0.3],
+    ['assets/example_videos/chicken_01.mp4', -1, -1, 1280, True, True, True, 0.3],
+    ['assets/example_videos/gorilla_01.mp4', -1, -1, 1280, True, True, True, 0.3],
+    ['assets/example_videos/davis_rollercoaster.mp4', -1, -1, 1280, True, True, True, 0.3],
+    ['assets/example_videos/Tokyo-Walk_rgb.mp4', -1, -1, 1280, True, True, True, 0.3],
+    ['assets/example_videos/4158877-uhd_3840_2160_30fps_rgb.mp4', -1, -1, 1280, True, True, True, 0.3],
+    ['assets/example_videos/4511004-uhd_3840_2160_24fps_rgb.mp4', -1, -1, 1280, True, True, True, 0.3],
+    ['assets/example_videos/1753029-hd_1920_1080_30fps.mp4', -1, -1, 1280, True, True, True, 0.3],
+    ['assets/example_videos/davis_burnout.mp4', -1, -1, 1280, True, True, True, 0.3],
+    ['assets/example_videos/example_5473765-l.mp4', -1, -1, 1280, True, True, True, 0.3],
+    ['assets/example_videos/Istanbul-26920.mp4', -1, -1, 1280, True, True, True, 0.3],
+    ['assets/example_videos/obj_1.mp4', -1, -1, 1280, True, True, True, 0.3],
+    ['assets/example_videos/sheep_cut1.mp4', -1, -1, 1280, True, True, True, 0.3],
 ]
 
 # Use GPU if available; otherwise, use CPU.
@@ -64,7 +67,7 @@ def infer_video_depth(
     stitch: bool = True,
     grayscale: bool = True,
     convert_from_color: bool = True,
-    blur: float = 0.5,
+    blur: float = 0.3,
     output_dir: str = './outputs',
     input_size: int = 518,
 ):
@@ -104,7 +107,7 @@ def infer_video_depth(
                     depth_gray = cv2.cvtColor(depth_color, cv2.COLOR_RGB2GRAY)
                     depth_vis = np.stack([depth_gray] * 3, axis=-1)
                 else:
-                    # Directly generate a grayscale image from the normalized depth values.
+                    # Directly generate a grayscale image from normalized depth.
                     depth_vis = np.stack([depth_norm] * 3, axis=-1)
             else:
                 # Generate a color depth image using the inferno colormap.
@@ -175,7 +178,7 @@ def construct_demo():
                     stitch_option = gr.Checkbox(label="Stitch RGB & Depth Videos", value=True)
                     grayscale_option = gr.Checkbox(label="Output Depth as Grayscale", value=True)
                     convert_from_color_option = gr.Checkbox(label="Convert Grayscale from Color", value=True)
-                    blur_slider = gr.Slider(minimum=0, maximum=1, step=0.01, label="Depth Blur (can reduce edge artifacts)", value=0.5)
+                    blur_slider = gr.Slider(minimum=0, maximum=1, step=0.01, label="Depth Blur (can reduce edge artifacts)", value=0.3)
                 generate_btn = gr.Button("Generate")
             with gr.Column(scale=2):
                 pass
@@ -200,4 +203,4 @@ def construct_demo():
 if __name__ == "__main__":
     demo = construct_demo()
     demo.queue()  # Enable asynchronous processing.
-    demo.launch(share=True)
+    demo.launch(share=True)