add dynamic

app.py

@@ -1,4 +1,5 @@
 import os
+import random
 
 import cv2
 import gradio as gr
@@ -9,21 +10,33 @@ from imgutils.data import load_image
 from imgutils.restore import restore_with_nafnet, restore_with_scunet
 
 
-def clean_adverse(
+def dynamic_clean_adverse(
         input_image: Image.Image,
-        diameter: float = 5,
-        sigma_color: float = 8,
-        sigma_space: float = 8,
-        radius: float = 4,
-        eps: float = 16,
-) -> Image.Image:
+        diameter_min: int = 4,
+        diameter_max: int = 6,
+        sigma_color_min: float = 6.0,
+        sigma_color_max: float = 10.0,
+        sigma_space_min: float = 6.0,
+        sigma_space_max: float = 10.0,
+        radius_min: int = 3,
+        radius_max: int = 6,
+        eps_min: float = 16.0,
+        eps_max: float = 24.0,
+        b_iters: int = 64,
+        g_iters: int = 4,
+):
     img = np.array(input_image).astype(np.float32)
     y = img.copy()
 
-    for _ in range(64):
+    for _ in range(b_iters):
+        diameter = random.randint(diameter_min, diameter_max)
+        sigma_color = random.random() * (sigma_color_max - sigma_color_min) + sigma_color_min
+        sigma_space = random.random() * (sigma_space_max - sigma_space_min) + sigma_space_min
         y = cv2.bilateralFilter(y, diameter, sigma_color, sigma_space)
 
-    for _ in range(4):
+    for _ in range(g_iters):
+        radius = random.randint(radius_min, radius_max)
+        eps = random.random() * (eps_max - eps_min) + eps_min
         y = guidedFilter(img, y, radius, eps)
 
     output_image = Image.fromarray(y.clip(0, 255).astype(np.uint8))
@@ -33,18 +46,33 @@ def clean_adverse(
 def clean(
         image: Image.Image,
 
-        diameter: float = 5,
-        sigma_color: float = 8,
-        sigma_space: float = 8,
-        radius: float = 4,
-        eps: float = 16,
+        diameter_min: int = 4,
+        diameter_max: int = 6,
+        sigma_color_min: float = 6.0,
+        sigma_color_max: float = 10.0,
+        sigma_space_min: float = 6.0,
+        sigma_space_max: float = 10.0,
+        radius_min: int = 3,
+        radius_max: int = 6,
+        eps_min: float = 16.0,
+        eps_max: float = 24.0,
+        b_iters: int = 64,
+        g_iters: int = 4,
 
         use_scunet_clean: bool = False,
         use_nafnet_clean: bool = False
 ) -> Image.Image:
     image = load_image(image)
 
-    image = clean_adverse(image, diameter, sigma_color, sigma_space, radius, eps)
+    image = dynamic_clean_adverse(
+        image,
+        diameter_min, diameter_max,
+        sigma_color_min, sigma_color_max,
+        sigma_space_min, sigma_space_max,
+        radius_min, radius_max,
+        eps_min, eps_max,
+        b_iters, g_iters
+    )
     if use_scunet_clean:
         image = restore_with_scunet(image)
     if use_nafnet_clean:
@@ -59,51 +87,70 @@ if __name__ == '__main__':
                 gr_input_image = gr.Image(label='Input Image', type="pil")
                 gr_submit = gr.Button(value='MIST = MIST is Stupid Trash')
                 with gr.Accordion("Advanced Config", open=False):
-                    diameter_slider = gr.Slider(
-                        minimum=1,
-                        maximum=30,
-                        step=1,
-                        value=5,
-                        label="Diameter (default = 5)",
-                        interactive=True,
-                    )
-                    sigma_color_slider = gr.Slider(
-                        minimum=1,
-                        maximum=30,
-                        step=1,
-                        value=8,
-                        label="SigmaColor (default = 8)",
-                        interactive=True,
-                    )
-                    sigma_space_slider = gr.Slider(
-                        minimum=1,
-                        maximum=30,
-                        step=1,
-                        value=8,
-                        label="SigmaSpace (default = 8)",
-                        interactive=True,
-                    )
-
-                    radius_slider = gr.Slider(
-                        minimum=1,
-                        maximum=30,
-                        step=1,
-                        value=4,
-                        label="Radius (default = 4)",
-                        interactive=True,
-                    )
-                    eps_slider = gr.Slider(
-                        minimum=1,
-                        maximum=30,
-                        step=1,
-                        value=16,
-                        label="Accuracy (default = 16)",
-                        interactive=True,
-                    )
+                    with gr.Row():
+                        gr_diameter_min = gr.Slider(
+                            minimum=1, maximum=30, step=1, value=4,
+                            label="Diameter Min (default = 4)", interactive=True,
+                        )
+                        gr_diameter_max = gr.Slider(
+                            minimum=1, maximum=30, step=1, value=6,
+                            label="Diameter Max (default = 6)", interactive=True,
+                        )
+
+                    with gr.Row():
+                        gr_sigma_color_min = gr.Slider(
+                            minimum=1, maximum=30, step=1, value=6,
+                            label="SigmaColor Min (default = 6)", interactive=True,
+                        )
+                        gr_sigma_color_max = gr.Slider(
+                            minimum=1, maximum=30, step=1, value=10,
+                            label="SigmaColor Max (default = 10)", interactive=True,
+                        )
+
+                    with gr.Row():
+                        gr_sigma_space_min = gr.Slider(
+                            minimum=1, maximum=30, step=1, value=6,
+                            label="SigmaSpace Min (default = 6)", interactive=True,
+                        )
+                        gr_sigma_space_max = gr.Slider(
+                            minimum=1, maximum=30, step=1, value=10,
+                            label="SigmaSpace Max (default = 10)", interactive=True,
+                        )
+
+                    with gr.Row():
+                        gr_radius_min = gr.Slider(
+                            minimum=1, maximum=30, step=1, value=3,
+                            label="Radius Min (default = 3)", interactive=True,
+                        )
+                        gr_radius_max = gr.Slider(
+                            minimum=1, maximum=30, step=1, value=6,
+                            label="Radius Max (default = 6)", interactive=True,
+                        )
+
+                    with gr.Row():
+                        gr_eps_min = gr.Slider(
+                            minimum=1, maximum=30, step=1, value=16,
+                            label="Accuracy Min (default = 16)", interactive=True,
+                        )
+                        gr_eps_max = gr.Slider(
+                            minimum=1, maximum=30, step=1, value=24,
+                            label="Accuracy Max (default = 24)", interactive=True,
+                        )
+
+                    with gr.Row():
+                        gr_b_iters = gr.Slider(
+                            minimum=1, maximum=256, step=1, value=64,
+                            label="Bilateral Filter Iters (default = 64)", interactive=True,
+                        )
+                        gr_g_iters = gr.Slider(
+                            minimum=1, maximum=32, step=1, value=4,
+                            label="Guided Filter Iters (default = 4)", interactive=True,
+                        )
 
                 with gr.Accordion("Extra Restoration", open=False):
-                    gr_scunet = gr.Checkbox(label='Use SCUNET', value=False)
-                    gr_nafnet = gr.Checkbox(label='Use NAFNET', value=False)
+                    with gr.Row():
+                        gr_scunet = gr.Checkbox(label='Use SCUNET', value=False)
+                        gr_nafnet = gr.Checkbox(label='Use NAFNET', value=False)
 
             with gr.Column():
                 gr_output_image = gr.Image(label='Output Image', type="pil")
@@ -113,11 +160,18 @@ if __name__ == '__main__':
                 inputs=[
                     gr_input_image,
 
-                    diameter_slider,
-                    sigma_color_slider,
-                    sigma_space_slider,
-                    radius_slider,
-                    eps_slider,
+                    gr_diameter_min,
+                    gr_diameter_max,
+                    gr_sigma_color_min,
+                    gr_sigma_color_max,
+                    gr_sigma_space_min,
+                    gr_sigma_space_max,
+                    gr_radius_min,
+                    gr_radius_max,
+                    gr_eps_min,
+                    gr_eps_max,
+                    gr_b_iters,
+                    gr_g_iters,
 
                     gr_scunet,
                     gr_nafnet,