GAN模型测试
1. 下载项目代码
git clone https://github.com/vanhuyz/CycleGAN-TensorFlow.git
2. 将图片转换成tfrecords格式
这个项目中提供了一个build_data脚本,用于将图片转换成tfrecords形式。假设我们的图片存放在data/apple2orange/trainA/和 data/apple2orange/trainB/目录下,对应的命令就是:
python build_data.py \
--X_input_dir data/apple2orange/trainA/ \
--Y_input_dir data/apple2orange/trainB/ \
--X_output_file data/apple2orange/apple.tfrecords \
--Y_output_file data/apple2orange/orange.tfrecords
3. 训练
训练的命令为:
python train