一、下载源码
源码下载链接: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: