- PyTorch implementation
- Linux or macOS
- Python 2 or 3
- NVIDIA GPU + CUDA cuDNN
- Install PyTorch and dependencies from http://pytorch.org
- Install python libraries dominate.
pip install dominate
- Please download the pre-trained model from here (google drive link), and put it under
./checkpoints/IC_gll_v36/
- Test the model :
python test.py --dataroot datasets/ --name IC_gll_v36 --phase test
The test results will be saved to here: ./results/
.
This code borrows heavily from pytorch-CycleGAN-and-pix2pix.
If you have any questions, please contact with me. (1007642157@qq.com)