TransGAN: Two Transformers Can Make One Strong GAN
Code used for TransGAN: Two Transformers Can Make One Strong GAN.
Main Pipeline
Visual Results
prepare fid statistic file
mkdir fid_stat
Download the pre-calculated statistics (Google Drive) to ./fid_stat
.
Environment
pip install -r requirements.txt
Notice: Pytorch version has to be <=1.3.0 !
Training
Coming soon
Testing
Firstly download the checkpoint from (Google Drive) to ./pretrained_weight
# cifar-10
sh exps/cifar10_test.sh
# stl-10
sh exps/stl10_test.sh
Acknowledgement
FID code and CIFAR-10 statistics file from https://github.com/bioinf-jku/TTUR (official). Codebase from AutoGAN