upload

app.py

@@ -361,15 +361,6 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                                 value="Estimate Depth", variant="primary"
                             )
                             depth_image_reset_btn = gr.Button(value="Reset")
-                        with gr.Accordion("Advanced options", open=False):
-                            image_processing_res = gr.Radio(
-                                [
-                                    ("Native", 0),
-                                    ("Recommended", 768),
-                                ],
-                                label="Processing resolution",
-                                value=default_image_processing_res,
-                            )
                     with gr.Column():
                         depth_image_output_slider = ImageSlider(
                             label="Predicted depth of gray / color (red-near, blue-far)",
@@ -417,15 +408,6 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                                 value="Estimate Normal", variant="primary"
                             )
                             normal_image_reset_btn = gr.Button(value="Reset")
-                        with gr.Accordion("Advanced options", open=False):
-                            image_processing_res = gr.Radio(
-                                [
-                                    ("Native", 0),
-                                    ("Recommended", 768),
-                                ],
-                                label="Processing resolution",
-                                value=default_image_processing_res,
-                            )
                     with gr.Column():
                         # normal_image_output_slider = ImageSlider(
                         #     label="Predicted surface normal",
@@ -477,15 +459,6 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                                 value="Estimate Dichotomous Segmentation", variant="primary"
                             )
                             dis_image_reset_btn = gr.Button(value="Reset")
-                        with gr.Accordion("Advanced options", open=False):
-                            image_processing_res = gr.Radio(
-                                [
-                                    ("Native", 0),
-                                    ("Recommended", 768),
-                                ],
-                                label="Processing resolution",
-                                value=default_image_processing_res,
-                            )
                     with gr.Column():
                         # dis_image_output_slider = ImageSlider(
                         #     label="Predicted dichotomous image segmentation",
@@ -534,15 +507,6 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                                 value="Estimate Matting", variant="primary"
                             )
                             matting_image_reset_btn = gr.Button(value="Reset")
-                        with gr.Accordion("Advanced options", open=False):
-                            image_processing_res = gr.Radio(
-                                [
-                                    ("Native", 0),
-                                    ("Recommended", 768),
-                                ],
-                                label="Processing resolution",
-                                value=default_image_processing_res,
-                            )
                     with gr.Column():
                         # matting_image_output_slider = ImageSlider(
                         #     label="Predicted matting image",
@@ -561,7 +525,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                         )
 
                 filenames = []
-                filenames.extend(["matting_%d.jpg" %(i+1) for i in range(4)])
+                filenames.extend(["matting_%d.jpg" %i for i in range(5)])
                 example_folder = os.path.join(os.path.dirname(__file__), "matting_images")
                 # print(example_folder)
                 Examples(
@@ -590,15 +554,6 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                                 value="Estimate Segmentation", variant="primary"
                             )
                             seg_image_reset_btn = gr.Button(value="Reset")
-                        with gr.Accordion("Advanced options", open=False):
-                            image_processing_res = gr.Radio(
-                                [
-                                    ("Native", 0),
-                                    ("Recommended", 768),
-                                ],
-                                label="Processing resolution",
-                                value=default_image_processing_res,
-                            )
                     with gr.Column():
                         seg_image_output_slider = ImageSlider(
                             label="Predicted segmentation results",
@@ -616,9 +571,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                         )
 
                 filenames = []
-                filenames.extend(["seg_anime_%d.jpg" %(i+1) for i in range(7)])
-                filenames.extend(["seg_line_%d.jpg" %(i+1) for i in range(6)])
-                filenames.extend(["seg_real_%d.jpg" %(i+1) for i in range(24)])
+                filenames.extend(["seg_%d.jpg" %(i+1) for i in range(5)])
 
                 example_folder = os.path.join(os.path.dirname(__file__), "seg_images")
                 Examples(
@@ -643,19 +596,10 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                             # type="pil",
                         )
                         with gr.Row():
-                            disparity_image_submit_btn = gr.Button(
+                            disparity_imfage_submit_btn = gr.Button(
                                 value="Estimate Disparity", variant="primary"
                             )
                             disparity_image_reset_btn = gr.Button(value="Reset")
-                        with gr.Accordion("Advanced options", open=False):
-                            image_processing_res = gr.Radio(
-                                [
-                                    ("Native", 0),
-                                    ("Recommended", 768),
-                                ],
-                                label="Processing resolution",
-                                value=default_image_processing_res,
-                            )
                     with gr.Column():
                         disparity_image_output_slider = ImageSlider(
                             label="Predicted disparity results",
@@ -673,9 +617,9 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                         )
 
                 filenames = []
-                filenames.extend(["disparity_anime_%d.jpg" %(i+1) for i in range(7)])
-                filenames.extend(["disparity_line_%d.jpg" %(i+1) for i in range(6)])
-                filenames.extend(["disparity_real_%d.jpg" %(i+1) for i in range(24)])
+                filenames.extend(["depth_anime_%d.jpg" %(i+1) for i in range(7)])
+                filenames.extend(["depth_line_%d.jpg" %(i+1) for i in range(6)])
+                filenames.extend(["depth_real_%d.jpg" %(i+1) for i in range(24)])
 
                 example_folder = os.path.join(os.path.dirname(__file__), "depth_images")
                 Examples(
@@ -703,7 +647,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
             fn=process_pipe_depth,
             inputs=[
                 depth_image_input,
-                image_processing_res,
+                default_image_processing_res,
             ],
             outputs=[depth_image_output_slider, depth_image_output_files],
             concurrency_limit=1,
@@ -721,7 +665,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                 depth_image_input,
                 depth_image_output_slider,
                 depth_image_output_files,
-                image_processing_res,
+                default_image_processing_res,
             ],
             queue=False,
         )
@@ -736,7 +680,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
             fn=process_pipe_normal,
             inputs=[
                 normal_image_input,
-                image_processing_res,
+                default_image_processing_res,
             ],
             outputs=[normal_image_output, normal_image_output_files],
             concurrency_limit=1,
@@ -754,7 +698,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                 normal_image_input,
                 normal_image_output,
                 normal_image_output_files,
-                image_processing_res,
+                default_image_processing_res,
             ],
             queue=False,
         )
@@ -769,7 +713,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
             fn=process_pipe_dis,
             inputs=[
                 dis_image_input,
-                image_processing_res,
+                default_image_processing_res,
             ],
             outputs=[dis_image_output, dis_image_output_files],
             concurrency_limit=1,
@@ -787,7 +731,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                 dis_image_input,
                 dis_image_output,
                 dis_image_output_files,
-                image_processing_res,
+                default_image_processing_res,
             ],
             queue=False,
         )
@@ -802,7 +746,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
             fn=process_pipe_dis,
             inputs=[
                 matting_image_input,
-                image_processing_res,
+                default_image_processing_res,
             ],
             outputs=[matting_image_output, matting_image_output_files],
             concurrency_limit=1,
@@ -820,7 +764,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                 matting_image_input,
                 matting_image_output,
                 matting_image_output_files,
-                image_processing_res,
+                default_image_processing_res,
             ],
             queue=False,
         )
@@ -835,7 +779,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
             fn=process_pipe_seg,
             inputs=[
                 seg_image_input,
-                image_processing_res,
+                default_image_processing_res,
             ],
             outputs=[seg_image_output_slider, seg_image_output_files],
             concurrency_limit=1,
@@ -853,7 +797,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                 seg_image_input,
                 seg_image_output_slider,
                 seg_image_output_files,
-                image_processing_res,
+                default_image_processing_res,
             ],
             queue=False,
         )
@@ -868,7 +812,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
             fn=process_pipe_disparity,
             inputs=[
                 disparity_image_input,
-                image_processing_res,
+                default_image_processing_res,
             ],
             outputs=[disparity_image_output_slider, disparity_image_output_files],
             concurrency_limit=1,
@@ -886,7 +830,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                 disparity_image_input,
                 disparity_image_output_slider,
                 disparity_image_output_files,
-                image_processing_res,
+                default_image_processing_res,
             ],
             queue=False,
         )