|
""" |
|
实现web界面 |
|
""" |
|
|
|
from pathlib import Path |
|
|
|
import gradio as gr |
|
from detect import detect,opt |
|
from util import get_all_weights |
|
|
|
|
|
def main(): |
|
app_introduce = """ |
|
# CycleGAN |
|
功能：上传本地文件、选择转换风格 |
|
""" |
|
|
|
css = """ |
|
# :root{ |
|
# --block-background-fill: #f3f3f3; |
|
# } |
|
# footer{ |
|
# display:none!important; |
|
# } |
|
""" |
|
|
|
demo = gr.Blocks( |
|
|
|
css=css |
|
) |
|
|
|
|
|
def add_img(img): |
|
imgs = [] |
|
for style in get_all_weights(): |
|
fake_img = detect(img, style=style) |
|
imgs.append(fake_img) |
|
return imgs |
|
|
|
|
|
def tab1(label="上传单张图片"): |
|
default_img_paths = [ |
|
[Path.cwd().joinpath("./imgs/horse.jpg"), "horse2zebra"], |
|
[Path.cwd().joinpath("./imgs/monet.jpg"), "monet2photo"], |
|
] |
|
|
|
with gr.Tab(label): |
|
with gr.Row(): |
|
with gr.Column(): |
|
img = gr.Image(type="pil", label="选择需要进行风格转换的图片") |
|
style = gr.Dropdown(choices=get_all_weights(), label="转换的风格") |
|
detect_btn = gr.Button("♻️风格转换") |
|
with gr.Column(): |
|
out_img = gr.Image(label="风格图") |
|
detect_btn.click(fn=detect, inputs=[img, style], outputs=[out_img]) |
|
gr.Examples(default_img_paths, inputs=[img, style]) |
|
|
|
|
|
def tab2(label="单图多风格试转换"): |
|
with gr.Tab(label): |
|
with gr.Row(): |
|
with gr.Column(scale=1): |
|
img = gr.Image(type="pil", label="选择需要进行风格转换的图片") |
|
gr.Markdown("上传一张图片，会将所有风格推理一遍。") |
|
btn = gr.Button("♻️风格转换") |
|
with gr.Column(scale=2): |
|
gallery = gr.Gallery( |
|
label="风格图", |
|
elem_id="gallery", |
|
).style(grid=[3], height="auto") |
|
btn.click(fn=add_img, inputs=[img], outputs=gallery) |
|
|
|
|
|
def tab3(label="参数设置"): |
|
with gr.Tab(label): |
|
with gr.Column(): |
|
for k, v in sorted(vars(opt).items()): |
|
if type(v) == bool: |
|
gr.Checkbox(label=k, value=v) |
|
elif type(v) == (int or float): |
|
gr.Number(label=k, value=v) |
|
elif type(v) == list: |
|
gr.CheckboxGroup(label=k, value=v) |
|
else: |
|
gr.Textbox(label=k, value=v) |
|
|
|
|
|
with demo: |
|
gr.Markdown(app_introduce) |
|
tab1() |
|
tab2() |
|
tab3() |
|
demo.launch(share=True) |
|
|
|
if __name__ == "__main__": |
|
main() |
|
|
|
|
|
|
|
|
