windows10下面用tensorflow实现Neural Style图像风格转移教程
新手一枚,最近实验室要求跑一跑图像风格转移的实验,通过查阅资料,跟随大神们的步伐,成功完成了实验。
下面记录了我的实验过程~
参考博客:https://blog.csdn.net/aaronjny/article/details/79681080
1、实验环境
python 3.5.2
tensorflow 1.2.1
2、下载预训练的vgg网络,并放入到项目根目录中
下载地址:http://www.vlfeat.org/matconvnet/models/beta16/imagenet-vgg-verydeep-19.mat
3、选定风格图片和内容图片,并放入根目录的images文件夹
如果要跑自己的数据,将自己的内容图片和风格图片替换图中所示的两张图,路径不变。
4、修改settings.py里面的代码,换成自己保存的路径
5、开始训练
运行train.py文件开始训练,在训练过程中,程序会定期提示进度,并保存过程图片。
所有的结果保存在根目录的output文件夹当中。
6、结果展示
内容图片
风格图片
所得结果