Live Display

.gitignore

@@ -1,3 +1,2 @@
-/kornia/*
-kornia/*
-kornia
+kornia
+flagged

app.py

@@ -6,12 +6,9 @@ from kornia.core import Tensor
 
 def load_img(file):
     # load the image using the rust backend
-    img_bgr: Tensor = K.io.load_image(file.name, K.io.ImageLoadType.RGB32)
-    img_bgr = img_bgr[None, ...].float() / 255.0
-
-    img_rgb: Tensor = K.color.bgr_to_rgb(img_bgr)
+    img_rgb: Tensor = K.io.load_image(file.name, K.io.ImageLoadType.RGB32)
+    img_rgb = img_rgb[None]
     img_gray = K.color.rgb_to_grayscale(img_rgb)
-
     return img_gray
 
 
@@ -29,98 +26,124 @@ def sobel_edge_detector(file):
     return K.utils.tensor_to_image(img_out)
 
 
-def simple_edge_detector(file, order=1, dir="x"):
+def simple_edge_detector(file, order, direction):
     x_gray = load_img(file)
-    grads: Tensor = K.filters.spatial_gradient(x_gray, order=2)  # BxCx2xHxW
+    grads: Tensor = K.filters.spatial_gradient(
+        x_gray, order=order
+    )  # BxCx2xHxW
     grads_x = grads[:, :, 0]
     grads_y = grads[:, :, 1]
-    if dir == "x":
+    if direction == "x":
         img_out = 1.0 - grads_x.clamp(0.0, 1.0)
     else:
         img_out = 1.0 - grads_y.clamp(0.0, 1.0)
     return K.utils.tensor_to_image(img_out)
 
 
-def laplacian_edge_detector(file, kernel_size=5):
+def laplacian_edge_detector(file, kernel):
     x_gray = load_img(file)
-    x_canny: Tensor = K.filters.canny(x_gray)[0]
-    img_out = 1.0 - x_canny.clamp(0.0, 1.0)
+    x_laplacian: Tensor = K.filters.laplacian(x_gray, kernel_size=kernel)
+    img_out = 1.0 - x_laplacian.clamp(0.0, 1.0)
     return K.utils.tensor_to_image(img_out)
 
 
-examples = [
-    ["examples/doraemon.png"],
-]
+examples = [["examples/doraemon.png"], ["examples/kornia.png"]]
 
 title = "Kornia Edge Detector"
 description = "<p style='text-align: center'>This is a Gradio demo for Kornia's Edge Detector.</p><p style='text-align: center'>To use it, simply upload your image, or click one of the examples to load them, and use the sliders to enhance! Read more at the links at the bottom.</p>"
 article = "<p style='text-align: center'><a href='https://kornia.readthedocs.io/en/latest/' target='_blank'>Kornia Docs</a> | <a href='https://github.com/kornia/kornia' target='_blank'>Kornia Github Repo</a> | <a href='https://kornia-tutorials.readthedocs.io/en/latest/image_enhancement.html' target='_blank'>Kornia Enhancements Tutorial</a></p>"
 
 
-def change_kernel(choice):
+def change_layout(choice):
+    kernel = gr.update(visible=False)
+    order = gr.update(visible=False)
+    direction = gr.update(visible=False)
     if choice == "Laplacian":
-        return gr.update(visible=True)
+        return [gr.update(value=3, visible=True), order, direction]
+    elif choice == "Simple":
+        return [
+            kernel,
+            gr.update(value=2, visible=True),
+            gr.update(value="x", visible=True),
+        ]
+    return [kernel, order, direction]
+
+
+def Detect(file, choice):
+    layout = change_layout(choice)
+    if choice == "Canny":
+        img = canny_edge_detector(file)
+    elif choice == "Sobel":
+        img = sobel_edge_detector(file)
+    elif choice == "Laplacian":
+        img = laplacian_edge_detector(file, 5)
     else:
-        return gr.update(visible=False)
-
-
-def change_order(choice):
-    if choice == "simple":
-        return gr.update(visible=True)
-    else:
-        return gr.update(visible=False)
-
-
-def change_button(choice):
-    if choice == "simple":
-        return gr.update("Detect").click(
-            canny_edge_detector, inputs=image_input, outputs=image_output
-        )
-    elif choice == "sobel":
-        return gr.update("Detect").click(
-            sobel_edge_detector, inputs=image_input, outputs=image_output
-        )
-    elif choice == "laplacian":
-        return gr.update("Detect").click(
-            laplacian_edge_detector, inputs=image_input, outputs=image_output
-        )
+        img = simple_edge_detector(file, 1, "x")
+    layout.extend([img])
+    return layout
+
+
+def Detect_wo_layout(file, choice, kernel, order, direction):
+    if choice == "Canny":
+        img = canny_edge_detector(file)
+    elif choice == "Sobel":
+        img = sobel_edge_detector(file)
+    elif choice == "Laplacian":
+        img = laplacian_edge_detector(file, kernel)
     else:
-        return gr.update("Detect").click(
-            simple_edge_detector, inputs=image_input, outputs=image_output
-        )
+        img = simple_edge_detector(file, order, direction)
+    return img
 
 
 with gr.Blocks() as demo:
     with gr.Row():
-        image_input = gr.Image()
-        image_output = gr.Image()
-        radio = gr.Radio(
-            ["canny", "simple", "sobel", "laplacian"],
-            label="Type of Edge Detector",
+        with gr.Column():
+            image_input = gr.Image(type="file")
+            kernel = gr.Slider(
+                minimum=1,
+                maximum=7,
+                step=2,
+                value=3,
+                label="kernel_size",
+                visible=False,
+            )
+            order = gr.Radio(
+                [1, 2], value=1, label="Derivative Order", visible=False
+            )
+            direction = gr.Radio(
+                ["x", "y"],
+                value="x",
+                label="Derivative Direction",
+                visible=False,
+            )
+
+            radio = gr.Radio(
+                ["Canny", "Simple", "Sobel", "Laplacian"],
+                label="Type of Edge Detector",
+            )
+            gr.Examples(examples, inputs=[image_input])
+        image_output = gr.Image(shape=(256, 256))
+
+        radio.change(
+            fn=Detect,
+            inputs=[image_input, radio],
+            outputs=[kernel, order, direction, image_output],
         )
-    with gr.Row():
-        kernel = gr.Slider(
-            minimum=1,
-            maximum=6,
-            step=1,
-            default=5,
-            label="kernel_size",
-            visible=False,
+        kernel.change(
+            fn=Detect_wo_layout,
+            inputs=[image_input, radio, kernel, order, direction],
+            outputs=[image_output],
         )
-        order = gr.Slider(
-            minimum=1,
-            maximum=2,
-            step=1,
-            default=1,
-            label="Derivative order",
-            visible=False,
+        order.change(
+            fn=Detect_wo_layout,
+            inputs=[image_input, radio, kernel, order, direction],
+            outputs=[image_output],
+        )
+        direction.change(
+            fn=Detect_wo_layout,
+            inputs=[image_input, radio, kernel, order, direction],
+            outputs=[image_output],
         )
-        Button = gr.Button("Detect")
-
-        radio.change(fn=change_kernel, inputs=radio, outputs=kernel)
-        radio.change(fn=change_order, inputs=radio, outputs=order)
-        radio.change(fn=change_button, inputs=radio, outputs=Button)
-    gr.Examples(examples, inputs=[image_input])
 
 
 demo.launch()