高代码应用-快速开始
更新时间:2024-03-24
高代码应用快速开始
使用样例-Streamlit
以下是基于Streamlit应用创建工具使用PaddleHub中stylepro_artistic模型实现图片风格迁移的应用代码和交互效果。需要注意的是代码中迁移风格图片(StarryNight.png)需要您手动下载并上传至应用文件的同级目录。 示例代码:
import streamlit as st
import paddlehub as hub
import cv2
from PIL import Image
import numpy as np
import os
import paddlehub as hub
import cv2
stylepro_artistic = hub.Module(name="stylepro_artistic")
st.markdown('### 梵高星空夜图像风格迁移')
st.write('点击 Browse files 上传张图片,生成梵高星空夜风格的图片,赶快试试吧~')
with st.form(key="图像风格迁移"):
per_image = st.file_uploader("上传图片", type=['png', 'jpg','jpeg'], label_visibility='hidden')
col1, col2 = st.columns(2)
with col1:
st.markdown('#### 原始图片')
if per_image:
st.image(per_image)
else:
st.image("https://codelab-public.bj.bcebos.com/base.jpeg")
submit = st.form_submit_button("开始生成")
with col2:
st.markdown('#### 预测结果')
if per_image:
img = Image.open(per_image)
img = cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)
result =stylepro_artistic.style_transfer(images=
[{
'content': img,
'styles': [cv2.imread(os.path.join(os.path.dirname(os.path.realpath(__file__)),'StarryNight.png'))]
}])
st.image(result[0]['data'])
else:
st.image("https://codelab-public.bj.bcebos.com/test.jpeg")
st.markdown('#### 梵高星空夜作品展示')
st.image("https://bkimg.cdn.bcebos.com/pic/d01373f082025aafcce937a8f7eda
交互效果如下:
完整项目实践代码可点击此处查看。
使用样例-Gradio
以下是基于Gradio应用创建工具使用PaddleHub中stylepro_artistic模型实现图片风格迁移的应用代码和交互效果。需要注意的是代码中迁移风格图片(StarryNight.png)需要您手动下载并上传至应用文件的同级目录。 示例代码:
import gradio as gr
import paddlehub as hub
import cv2
import numpy as np
import os
from PIL import Image
stylepro_artistic = hub.Module(name="stylepro_artistic")
def quickstart(image):
result =stylepro_artistic.style_transfer(images=
[{
'content': image,
'styles': [cv2.imread(os.path.join(os.path.dirname(os.path.realpath(__file__)),'StarryNight.png'))]
}])
return result[0]['data']
demo = gr.Interface(fn=quickstart,title="梵高星空夜图像风格迁移",inputs="image", outputs="image")
demo.launch()
交互效果如下:
完整项目代码教程,可点击此处查看。