i

main.py

@@ -5,15 +5,22 @@ from cv2.ximgproc import guidedFilter
 import gradio as gr
 
 
-def clean_image(input_image: Image) -> Image:
+def clean_image(
+    input_image: Image,
+    diameter: float = 5,
+    sigma_color: float = 8,
+    sigma_space: float = 8,
+    radius: float = 4,
+    eps: float = 16,
+) -> Image:
     img = np.array(input_image).astype(np.float32)
     y = img.copy()
 
     for _ in range(64):
-        y = cv2.bilateralFilter(y, 5, 8, 8)
+        y = cv2.bilateralFilter(y, diameter, sigma_color, sigma_space)
 
     for _ in range(4):
-        y = guidedFilter(img, y, 4, 16)
+        y = guidedFilter(img, y, radius, eps)
 
     output_image = Image.fromarray(y.clip(0, 255).astype(np.uint8))
     return output_image
@@ -34,13 +41,66 @@ def ui():
                 input_image = gr.Image(type="pil", label="Input Image")
                 start_btn = gr.Button(value="Start", variant="primary")
 
+                with gr.Accordion("Advanced Config", open=False):
+                    # ref: https://github.com/gogodr/AdverseCleanerExtension
+                    gr.Markdown("#### Bilateral Filter")
+                    diameter_slider = gr.Slider(
+                        minimum=1,
+                        maximum=30,
+                        step=1,
+                        value=5,
+                        label="Diameter",
+                        interactive=True,
+                    )
+                    sigma_color_slider = gr.Slider(
+                        minimum=1,
+                        maximum=30,
+                        step=1,
+                        value=8,
+                        label="SigmaColor",
+                        interactive=True,
+                    )
+                    sigma_space_slider = gr.Slider(
+                        minimum=1,
+                        maximum=30,
+                        step=1,
+                        value=8,
+                        label="SigmaSpace",
+                        interactive=True,
+                    )
+
+                    gr.Markdown("#### Guided Filter")
+                    radius_slider = gr.Slider(
+                        minimum=1,
+                        maximum=30,
+                        step=1,
+                        value=4,
+                        label="Radius",
+                        interactive=True,
+                    )
+                    eps_slider = gr.Slider(
+                        minimum=1,
+                        maximum=30,
+                        step=1,
+                        value=16,
+                        label="Accuracy",
+                        interactive=True,
+                    )
+
                 gr.Examples(
                     examples=[
                         ["./examples/sample1.jpg"],
                         ["./examples/sample2.jpg"],
                         ["./examples/sample3.jpg"],
                     ],
-                    inputs=[input_image],
+                    inputs=[
+                        input_image,
+                        diameter_slider,
+                        sigma_color_slider,
+                        sigma_space_slider,
+                        radius_slider,
+                        eps_slider,
+                    ],
                     outputs=[],
                     fn=example,
                     cache_examples=True,