Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
lappemic
commited on
Commit
·
e3c3f88
1
Parent(s):
c06cbee
Change to tab app
Browse files
app.py
CHANGED
|
@@ -6,14 +6,14 @@ import kornia as K
|
|
| 6 |
from kornia.core import Tensor
|
| 7 |
|
| 8 |
|
| 9 |
-
def filters(file,
|
| 10 |
# load the image using the rust backend
|
| 11 |
img: Tensor = K.io.load_image(file.name, K.io.ImageLoadType.RGB32)
|
| 12 |
img = img[None] # 1xCxHxW / fp32 / [0, 1]
|
| 13 |
|
| 14 |
# apply tensor image enhancement
|
| 15 |
-
x_out: Tensor = K.filters.box_blur(img, (int(box_blur), int(box_blur)))
|
| 16 |
x_out = K.filters.blur_pool2d(x_out, int(blur_pool2d))
|
|
|
|
| 17 |
x_out = K.filters.gaussian_blur2d(x_out,
|
| 18 |
(int(gaussian_blur2d), int(gaussian_blur2d)),
|
| 19 |
(float(gaussian_blur2d), float(gaussian_blur2d)))
|
|
@@ -28,16 +28,29 @@ examples = [
|
|
| 28 |
["examples/pikachu.jpg", 1, 1, 1, 1, 1],
|
| 29 |
]
|
| 30 |
|
| 31 |
-
|
| 32 |
-
description = "<p style='text-align: center'>This is a Gradio demo for Kornia's Image Filters.</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>"
|
| 33 |
-
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>"
|
| 34 |
-
|
| 35 |
-
iface = gr.Interface(
|
| 36 |
filters,
|
| 37 |
[
|
| 38 |
gr.inputs.Image(type="file"),
|
| 39 |
gr.inputs.Slider(minimum=1, maximum=10, step=1, default=1, label="Box Blur"),
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
gr.inputs.Slider(minimum=1, maximum=10, step=1, default=1, label="Blur Pool"),
|
|
|
|
| 41 |
gr.inputs.Slider(minimum=1, maximum=21, step=2, default=1, label="Gaussian Blur"),
|
| 42 |
gr.inputs.Slider(minimum=1, maximum=20, step=1, default=1, label="Max Pool"),
|
| 43 |
gr.inputs.Slider(minimum=1, maximum=5, step=2, default=1, label="Median Blur"),
|
|
@@ -45,9 +58,20 @@ iface = gr.Interface(
|
|
| 45 |
"image",
|
| 46 |
examples=examples,
|
| 47 |
# title=title,
|
| 48 |
-
|
| 49 |
# article=article,
|
| 50 |
live=True
|
| 51 |
)
|
| 52 |
|
| 53 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
from kornia.core import Tensor
|
| 7 |
|
| 8 |
|
| 9 |
+
def filters(file, blur_pool2d, box_blur, gaussian_blur2d, max_blur_pool2d, median_blur):
|
| 10 |
# load the image using the rust backend
|
| 11 |
img: Tensor = K.io.load_image(file.name, K.io.ImageLoadType.RGB32)
|
| 12 |
img = img[None] # 1xCxHxW / fp32 / [0, 1]
|
| 13 |
|
| 14 |
# apply tensor image enhancement
|
|
|
|
| 15 |
x_out = K.filters.blur_pool2d(x_out, int(blur_pool2d))
|
| 16 |
+
x_out: Tensor = K.filters.box_blur(img, (int(box_blur), int(box_blur)))
|
| 17 |
x_out = K.filters.gaussian_blur2d(x_out,
|
| 18 |
(int(gaussian_blur2d), int(gaussian_blur2d)),
|
| 19 |
(float(gaussian_blur2d), float(gaussian_blur2d)))
|
|
|
|
| 28 |
["examples/pikachu.jpg", 1, 1, 1, 1, 1],
|
| 29 |
]
|
| 30 |
|
| 31 |
+
without_downsampling_demo = gr.Interface(
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
filters,
|
| 33 |
[
|
| 34 |
gr.inputs.Image(type="file"),
|
| 35 |
gr.inputs.Slider(minimum=1, maximum=10, step=1, default=1, label="Box Blur"),
|
| 36 |
+
gr.inputs.Slider(minimum=1, maximum=21, step=2, default=1, label="Gaussian Blur"),
|
| 37 |
+
gr.inputs.Slider(minimum=1, maximum=20, step=1, default=1, label="Max Pool"),
|
| 38 |
+
gr.inputs.Slider(minimum=1, maximum=5, step=2, default=1, label="Median Blur"),
|
| 39 |
+
],
|
| 40 |
+
"image",
|
| 41 |
+
examples=examples,
|
| 42 |
+
# title=title,
|
| 43 |
+
description= 'If you want to use the filters with downsampled image, use tab "With image downsampling"',
|
| 44 |
+
# article=article,
|
| 45 |
+
live=True
|
| 46 |
+
)
|
| 47 |
+
|
| 48 |
+
with_downsampling_demo = gr.Interface(
|
| 49 |
+
filters,
|
| 50 |
+
[
|
| 51 |
+
gr.inputs.Image(type="file"),
|
| 52 |
gr.inputs.Slider(minimum=1, maximum=10, step=1, default=1, label="Blur Pool"),
|
| 53 |
+
gr.inputs.Slider(minimum=1, maximum=10, step=1, default=1, label="Box Blur"),
|
| 54 |
gr.inputs.Slider(minimum=1, maximum=21, step=2, default=1, label="Gaussian Blur"),
|
| 55 |
gr.inputs.Slider(minimum=1, maximum=20, step=1, default=1, label="Max Pool"),
|
| 56 |
gr.inputs.Slider(minimum=1, maximum=5, step=2, default=1, label="Median Blur"),
|
|
|
|
| 58 |
"image",
|
| 59 |
examples=examples,
|
| 60 |
# title=title,
|
| 61 |
+
description = 'Blur Pooling downsamples the image in the default setting!',
|
| 62 |
# article=article,
|
| 63 |
live=True
|
| 64 |
)
|
| 65 |
|
| 66 |
+
demo = gr.TabbedInterface(
|
| 67 |
+
[
|
| 68 |
+
without_downsampling_demo,
|
| 69 |
+
with_downsampling_demo
|
| 70 |
+
],
|
| 71 |
+
[
|
| 72 |
+
"Without image downsampling",
|
| 73 |
+
"With image downsampling"
|
| 74 |
+
]
|
| 75 |
+
)
|
| 76 |
+
|
| 77 |
+
demo.launch()
|