Code used for TransGAN: Two Transformers Can Make One Strong GAN.

checkpoint for generating images on celeba64 dataset is released!

mkdir fid_stat

Download the pre-calculated statistics (Google Drive) to ./fid_stat.

pip install -r requirements.txt

Notice: Pytorch version has to be <=1.3.0 !

Coming soon

Firstly download the checkpoint from (Google Drive) to ./pretrained_weight

# cifar-10
sh exps/cifar10_test.sh

# stl-10
sh exps/stl10_test.sh

# celeba64
sh exps/celeba64_test.sh

Codebase from AutoGAN, pytorch-image-models

if you find this repo is helpful, please cite

@article{jiang2021transgan,
  title={TransGAN: Two Transformers Can Make One Strong GAN},
  author={Jiang, Yifan and Chang, Shiyu and Wang, Zhangyang},
  journal={arXiv preprint arXiv:2102.07074},
  year={2021}
}