1

app.py

@@ -1,99 +1,87 @@
+# 2023年2月23日
 """
 实现web界面
+
+>>> streamlit run app.py
 """
 
+from io import BytesIO
 from pathlib import Path
 
-import gradio as gr
-from detect import detect,opt
+import streamlit as st
+from detect import detect, opt
+from PIL import Image
 from util import get_all_weights
 
+"""
+# CycleGAN
+功能：上传本地文件、选择转换风格
+"""
 
-def main():
-    app_introduce = """
-    # CycleGAN
-    功能：上传本地文件、选择转换风格
-    """
 
-    css = """
-    # :root{
-    # --block-background-fill: #f3f3f3;
-    # }
-    # footer{
-    #     display:none!important;
-    # }
+def load_css(css_path="./util/streamlit/css.css"):
     """
-
-    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
+    加载CSS文件
+    :param css_path: CSS文件路径
+    """
+    if Path(css_path).exists():
+        with open(css_path) as f:
+            # 将CSS文件内容插入到HTML中
+            st.markdown(
+                f"""<style>{f.read()}</style>""",
+                unsafe_allow_html=True,
+            )
+
+
+def load_img_file(file):
+    """读取图片文件"""
+    img = Image.open(BytesIO(file.read()))
+    st.image(img, use_column_width=True)  # 显示图片
+    return img
+
+
+def set_style_options(label: str, frame=st):
+    """风格选项"""
+    style_options = get_all_weights()
+    options = [None] + style_options  # 默认空
+    style_param = frame.selectbox(label=label, options=options)
+    return style_param
+
+
+# load_css()
+tab_mul2mul, tab_mul2one, tab_set = st.tabs(["多图多风格转换", "多图同风格转换", "参数"])
+
+with tab_mul2mul:
+    uploaded_files = st.file_uploader(label="选择本地图片", accept_multiple_files=True, key=1)
+    if uploaded_files:
+        for idx, uploaded_file in enumerate(uploaded_files):
+            colL, colR = st.columns(2)
+            with colL:
+                img = load_img_file(uploaded_file)
+                style = set_style_options(label=str(uploaded_file), frame=st)
+            with colR:
+                if style:
+                    fake_img = detect(img=img, style=style)
+                    st.image(fake_img, caption="", use_column_width=True)
+
+with tab_set:
+    colL, colR = st.columns([1, 3])
+    for k, v in sorted(vars(opt).items()):
+        st.text_input(label=k, value=v, disabled=True)
+    # st.selectbox("ss", options=opt.parse_args())
+    confidence_threshold = st.slider("Confidence threshold", 0.0, 1.0, 0.5, 0.01)
+    opt.no_dropout = st.radio("no_droput", [True, False])
+
+with tab_mul2one:
+    uploaded_files = st.file_uploader(label="选择本地图片", accept_multiple_files=True, key=2)
+    if uploaded_files:
+        colL, colR = st.columns(2)
+        with colL:
+            imgs = [load_img_file(ii) for ii in uploaded_files]
+        with colR:
+            style = set_style_options(label="选择风格", frame=st)
+            if style:
+                if st.button("♻️风格转换", use_container_width=True):
+                    for img in imgs:
+                        fake_img = detect(img, style)
+                        st.image(fake_img, caption="", use_column_width=True)

app1.py

@@ -1,87 +0,0 @@
-# 2023年2月23日
-"""
-实现web界面
-
->>> streamlit run app.py
-"""
-
-from io import BytesIO
-from pathlib import Path
-
-import streamlit as st
-from detect import detect, opt
-from PIL import Image
-from util import get_all_weights
-
-"""
-# CycleGAN
-功能：上传本地文件、选择转换风格
-"""
-
-
-def load_css(css_path="./util/streamlit/css.css"):
-    """
-    加载CSS文件
-    :param css_path: CSS文件路径
-    """
-    if Path(css_path).exists():
-        with open(css_path) as f:
-            # 将CSS文件内容插入到HTML中
-            st.markdown(
-                f"""<style>{f.read()}</style>""",
-                unsafe_allow_html=True,
-            )
-
-
-def load_img_file(file):
-    """读取图片文件"""
-    img = Image.open(BytesIO(file.read()))
-    st.image(img, use_column_width=True)  # 显示图片
-    return img
-
-
-def set_style_options(label: str, frame=st):
-    """风格选项"""
-    style_options = get_all_weights()
-    options = [None] + style_options  # 默认空
-    style_param = frame.selectbox(label=label, options=options)
-    return style_param
-
-
-# load_css()
-tab_mul2mul, tab_mul2one, tab_set = st.tabs(["多图多风格转换", "多图同风格转换", "参数"])
-
-with tab_mul2mul:
-    uploaded_files = st.file_uploader(label="选择本地图片", accept_multiple_files=True, key=1)
-    if uploaded_files:
-        for idx, uploaded_file in enumerate(uploaded_files):
-            colL, colR = st.columns(2)
-            with colL:
-                img = load_img_file(uploaded_file)
-                style = set_style_options(label=str(uploaded_file), frame=st)
-            with colR:
-                if style:
-                    fake_img = detect(img=img, style=style)
-                    st.image(fake_img, caption="", use_column_width=True)
-
-with tab_set:
-    colL, colR = st.columns([1, 3])
-    for k, v in sorted(vars(opt).items()):
-        st.text_input(label=k, value=v, disabled=True)
-    # st.selectbox("ss", options=opt.parse_args())
-    confidence_threshold = st.slider("Confidence threshold", 0.0, 1.0, 0.5, 0.01)
-    opt.no_dropout = st.radio("no_droput", [True, False])
-
-with tab_mul2one:
-    uploaded_files = st.file_uploader(label="选择本地图片", accept_multiple_files=True, key=2)
-    if uploaded_files:
-        colL, colR = st.columns(2)
-        with colL:
-            imgs = [load_img_file(ii) for ii in uploaded_files]
-        with colR:
-            style = set_style_options(label="选择风格", frame=st)
-            if style:
-                if st.button("♻️风格转换", use_container_width=True):
-                    for img in imgs:
-                        fake_img = detect(img, style)
-                        st.image(fake_img, caption="", use_column_width=True)

app_gradio.py

@@ -0,0 +1,99 @@
+"""
+实现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

requirements.txt

@@ -1,4 +1,4 @@
-gradio
+streamlit
 numpy
 Pillow
 torch