使用CycleGAN训练自己的数据集

一、下载源码

源码下载链接：https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix

 二、调整自己的数据集格式

按照如下格式放 

三、安装好所需要的python包

 使用命令：pip install 包名称 在anaconda里面安装，

 四、训练和测试

 在开始训练前，需要打开visdom（训练可视化界面），在命令行直接输入以下命令：

python -m visdom.server

 如果没有提前打开，可以去anaconda黑窗口打开，然后复制网址http://localhost:8097到浏览器

 训练命令行：

python train.py --dataroot ./datasets/GPRData --name gpr_cyclegan --model cycle_gan

测试命令行:

（1）生成的文件夹里没图片，index.html的图片也没正常显示

python test.py --dataroot ./datasets/GPRData --name gpr_cyclegan --model cycle_gan --no_dropout

（2） 可以生成由A风格转换得来的假B风格的图片

python test.py --dataroot ./datasets/GPRData/testA --name gpr_cyclegan_test_results --model test --no_dropout

 五、训练参数的修改

（1）训练轮数epoch，改动两个参数，参数之和为epoch：

 （2）如果册灰姑娘需出现问题，自动结束或者停滞不前，可以重新运行，改动--continue_train、--epoch_count这两个参数，比如我的在运行到第五轮的时候，就一直不动了（前几轮的权重都是保存下来了的），于是我结束了运行，首先将--epoch_count参数的默认值改为5，然后训练的时候，在训练命令后面加上--continue_train这个参数，如下图所示：

python train.py --dataroot ./datasets/GPRData --name gpr_cyclegan --model cycle_gan --continue_train

六、输出结果