AugSelf-GAN — Official PyTorch Implementation

Overview

AugSelf-BigGAN on CIFAR-10 and CIFAR-100

Please refer to the augself-biggan README to run AugSelf-BigGAN for conditional generation on CIFAR-10 and CIFAR-100.

AugSelf-StyleGAN2 on FFHQ, LSUN-Cat, and five low-shot datasets

Please refer to the augself-stylegan2 README to run AugSelf-StyleGAN2 for unconditional generation on FFHQ, LSUN-Cat, and five low-shot datasets.

Citation

If you find this code helpful, please cite our paper:

@article{hou2022augmentation,
  title={Augmentation-Aware Self-Supervision for Data-Efficient GAN Training},
  author={Hou, Liang and Cao, Qi and Yuan, Yige and Zhao, Songtao and Ma, Chongyang and Pan, Siyuan and Wan, Pengfei and Wang, Zhongyuan and Shen, Huawei and Cheng, Xueqi},
  journal={arXiv preprint arXiv:2205.15677},
  year={2022}
}

Acknowledgements

This repository is built upon DiffAugment.