图片风格快速转换的简单web实现
图片风格转换,是指利用深度学习算法学习某种风格图片的特征,将其应用到另一张图片中,合成新风格的图片,目前技术较为成熟,github上有很多有趣的项目与应用。
本项目核心代码基于fast-neural-style-tensorflow,在此基础上,添加了简单的flask框架,实现了页面上传图片,选择转换风格,生成转换图片并展示的流程。
项目的介绍、配置与部署见Github,地址:fast-neural-style-flask
一、核心代码分解
由于我对图像的深度学习算法和web都是刚入门学习,所以接下来只是简单介绍一下,不涉及核心算法。如果文中有说的不对的地方,还请大家多多指正。
我将这个项目分为三个部分:训练风格模型,实现图片转换,flask框架。
1. 训练风格模型
原项目中,作者提供了重新训练模型的方法,以训练铅笔画风格 pencil 为例。
首先,需要下载 VGG16 model 预训练模型和