Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
updated default sample image values matching with the slider's default value 0.25 to demonstrate the effect visible clearly
b2997cc
import gradio as gr | |
import kornia as K | |
from kornia.core import concatenate, Tensor | |
def rescale_aa(file, height, width): | |
img: Tensor = K.io.load_image(file.name, K.io.ImageLoadType.RGB32) | |
img = img[None] | |
img_rescale: Tensor = K.geometry.rescale(img, (float(height),float(width)),antialias=False) | |
img_rescale_aa: Tensor = K.geometry.rescale(img, (float(height),float(width)),antialias=True) | |
img_out = concatenate([img_rescale, img_rescale_aa], -1) | |
# when antialiasing , some values are going greater than 1 i.e 1.00001 which is giving error while displaying the output image,so clipping the output values from 0 to 1 | |
return K.utils.tensor_to_image(img_out.clamp_(0, 1)) | |
examples = [ | |
["examples/a.png",0.25,0.25], | |
["examples/iron_man.jpeg",0.25,0.25], | |
] | |
kornia_resizing_demo = gr.Interface( | |
rescale_aa, | |
[ | |
gr.inputs.Image(type="file"), | |
gr.inputs.Slider(minimum=0.005, maximum=2, step=0.005, default=0.5, label="Height"), | |
gr.inputs.Slider(minimum=0.005, maximum=2, step=0.005, default=0.5, label="Width") | |
], | |
"image", | |
examples=examples, | |
live=False, | |
enable_queue = True, | |
allow_flagging = "never" | |
) | |
kornia_resizing_demo.launch() |