Skip to content

yimiwawa/fast-neural-style-flask

Repository files navigation

图片风格快速转换的简单web实现

本项目核心代码基于fast-neural-style-tensorflow,在此基础上,添加了简单的flask框架,实现了页面上传图片,选择转换风格,生成转换图片并展示的流程。

更多说明,移步CSDN博客图片风格快速转换的简单web实现

Requirements:

  • 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

About

fast-neural-style-tensorflow的web实现

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published