CycleGAN and pix2pix in PyTorch
Linux下cuda9.1安装Pytorch和Torchvision
conda create -n pix2pix python=3.6.5
pip install torch==1.1.0
pip install pillow
pip install torchvision==0.3.0
pip install dominate
pip install visdom
- 测试
按照 git上的教程,下载对应数据集之后,运行 :
python test.py --dataroot datasets/horse2zebra/testA --name horse2zebra_pretrained --model test --no_dropout
发现:cycleGan-pix2pix/results 目录下生成对应的结果文件夹,查看效果即可