使用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

测试命令行:

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

 五、训练参数的修改

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