代码复现:CycleGAN and pix2pix

安装visdom

1
conda install visdom

开启visdom服务

1
python -m visdom.server

训练数据集(数据集名“xxx”更改自己的数据集名称)

cyclegan训练命令

1
python train.py --dataroot ./datasets/xxx --name xxx_cyclegan --model cycle_gan

pix2pix训练命令(AtoB表示由A数据集训练生成B数据集)

1
python train.py --dataroot ./datasets/xxx --name xxx_pix2pix --model pix2pix --direction AtoB

测试训练好的模型

1
python test.py --dataroot ./datasets/xxx --name xxx_pix2pix --model pix2pix

代码复现:CycleGAN and pix2pix
http://lian-shuo.github.io/2023/01/14/代码复现:CycleGAN-and-pix2pix/
作者
LianShuo
发布于
2023年1月14日
许可协议