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

Codebase from AutoGAN, pytorch-image-models

Citation

if you find this repo is helpful, please cite

@misc{jiang2021transgan,
      title={TransGAN: Two Transformers Can Make One Strong GAN}, 
      author={Yifan Jiang and Shiyu Chang and Zhangyang Wang},
      year={2021},
      eprint={2102.07074},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}