GitHub地址:https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
先看一下论文给出的效果:
首先准备自己的数据集。
在datasets文件夹下新建try文件夹,文件夹命名如下:
其中trainA、B中放入需要进行风格迁移的图片。
分别是没有水的地面,和淋了水的地面。
其次进行pip install安装需要的模块
pip install visdom
pip install dominate
运行train.py脚本
python train.py --dataroot ./datasets/try/ --name try_cyclegan