|
|
|
|
|
from __future__ import annotations |
|
|
|
import json |
|
import shlex |
|
import subprocess |
|
|
|
import gradio as gr |
|
|
|
|
|
def run(image_path: str, class_index: int, scale: str, sigma_y: float) -> str: |
|
out_name = image_path.split("/")[-1].split(".")[0] |
|
subprocess.run( |
|
shlex.split( |
|
f"python main.py --config confs/inet256.yml --resize_y --deg sr_averagepooling --scale {scale} --class {class_index} --path_y {image_path} --save_path {out_name} --sigma_y {sigma_y}" |
|
), |
|
cwd="DDNM/hq_demo", |
|
) |
|
return f"DDNM/hq_demo/results/{out_name}/final/00000.png" |
|
|
|
|
|
def create_demo(): |
|
examples = [ |
|
[ |
|
"DDNM/hq_demo/data/datasets/gts/inet256/323.png", |
|
"monarch, monarch butterfly, milkweed butterfly, Danaus plexippus", |
|
"4", |
|
0, |
|
], |
|
[ |
|
"DDNM/hq_demo/data/datasets/gts/inet256/orange.png", |
|
"orange", |
|
"4", |
|
0, |
|
], |
|
[ |
|
"DDNM/hq_demo/data/datasets/gts/inet256/monarch.png", |
|
"monarch, monarch butterfly, milkweed butterfly, Danaus plexippus", |
|
"4", |
|
0.5, |
|
], |
|
[ |
|
"DDNM/hq_demo/data/datasets/gts/inet256/bear.png", |
|
"brown bear, bruin, Ursus arctos", |
|
"4", |
|
0, |
|
], |
|
[ |
|
"DDNM/hq_demo/data/datasets/gts/inet256/flamingo.png", |
|
"flamingo", |
|
"2", |
|
0, |
|
], |
|
[ |
|
"DDNM/hq_demo/data/datasets/gts/inet256/kimono.png", |
|
"kimono", |
|
"2", |
|
0, |
|
], |
|
[ |
|
"DDNM/hq_demo/data/datasets/gts/inet256/zebra.png", |
|
"zebra", |
|
"4", |
|
0, |
|
], |
|
] |
|
|
|
with open("imagenet_classes.json") as f: |
|
imagenet_class_names = json.load(f) |
|
|
|
with gr.Blocks() as demo: |
|
with gr.Row(): |
|
with gr.Column(): |
|
image = gr.Image(label="Input image", type="filepath") |
|
class_index = gr.Dropdown(label="Class name", choices=imagenet_class_names, type="index", value=950) |
|
scale = gr.Dropdown(label="Scale", choices=["2", "4", "8"], value="4") |
|
sigma_y = gr.Number(label="sigma_y", value=0, precision=2) |
|
run_button = gr.Button("Run") |
|
with gr.Column(): |
|
result = gr.Image(label="Result", type="filepath") |
|
|
|
gr.Examples( |
|
examples=examples, |
|
inputs=[ |
|
image, |
|
class_index, |
|
scale, |
|
sigma_y, |
|
], |
|
) |
|
|
|
run_button.click( |
|
fn=run, |
|
inputs=[ |
|
image, |
|
class_index, |
|
scale, |
|
sigma_y, |
|
], |
|
outputs=result, |
|
) |
|
return demo |
|
|
