本项目核心代码基于fast-neural-style-tensorflow,在此基础上,添加了简单的flask框架,实现了页面上传图片,选择转换风格,生成转换图片并展示的流程。
更多说明,移步CSDN博客图片风格快速转换的简单web实现
- Python 2.7.x
- Now support Tensorflow >= 1.0
- pyyaml
- flask
- gevent
在原项目带有的7中模型(wave,cubist,denoised_starry,mosaic,scream,feathers,udnie)基础上,添加了铅笔画的模型pencil.
模型下载链接:https://pan.baidu.com/s/1LV46jqAe-M0CV8dOekoUYA 密码:ezvv
web服务端代码位于predict_flask.py文件中,修改代码中host、port,执行
python predict_flask.py
即可启动服务。
在浏览器中输入 http://host:port/index,即可用。
configuration | 原图 | 转换风格后图片 |
---|---|---|
pencil.yml |