File size: 3,121 Bytes
c1c16cb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
"""
实现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(
# theme=gr.themes.Soft(),
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()
# 如果不以`demo`命名，`gradio app.py`会报错`Error loading ASGI app. Attribute "demo.app" not found in module "app".`
# 注意gradio库的reload.py的头信息 $ gradio app.py my_demo, to use variable names other than "demo"
# my_demo 是定义的变量。离谱o(╥﹏╥)o
|