Code used for TransGAN: Two Transformers Can Make One Strong GAN.
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
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}
}