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 目錄下生成對應的結果文件夾,查看效果即可