【风格迁移】CycleGAN-tensorflow的部署和训练

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程序：

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