【风格迁移】CycleGAN的训练和实时视频风格迁移

CycleGAN简单来说，功能就是：自动将某一类图片转换成另外一类图片。

效果：

需要源码和训练好的模型的同学请见文章末联系方式~

大纲：

1. 环境配置
2. Github地址
3. 数据地址
4. 准备数据
5. 训练模型
6. 调用模型

论文传送门：https://arxiv.org/abs/1703.10593

环境：

tensorflow1.0.0
python3.6.0

Github地址：

https://github.com/vanhuyz/CycleGAN-TensorFlow

数据地址：

https://people.eecs.berkeley.edu/~taesung_park/CycleGAN/datasets/apple2orange.zip

数据集如图：

准备数据：

运行build_data.py，如图：

其中需要把参数修改一下：

① 改成数据集trainA的路径；
② 改成数据集trainB的路径；
③ 改成想要存放trainA数据的tfrecords文件路径；
④ 改成想要存放trainB数据的tfrecords文件路径；


运行结束后可以看到，这里生成了相应的tfrecords文件；

训练模型：

运行train.py：

其中需要把圈出的两个地方改成刚才③和④的路径；

然后就可以开心的训练啦~

调用模型：

运行inference.py程序：

其中圈出来的地方改成输入的图片路径即可；

联系我们：

权重文件需要的请私戳作者~

联系我时请备注所需模型权重，我会拉你进交流群~

该群会定时分享各种源码和模型，之前分享过的请从群文件中下载~