Code used for TransGAN: Two Transformers Can Make One Strong GAN.
checkpoint for generating images on celeba64 dataset is released!
mkdir fid_statDownload the pre-calculated statistics
(Google Drive) to ./fid_stat.
pip install -r requirements.txtNotice: 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.shCodebase 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}
}