#99 · Primary category: Image Generation

PyTorch-StudioGAN

biggan clean-fid data-efficient-gan-training deep-learning generative-adversarial-network machine-learning pytorch stylegan2 stylegan2-ada stylegan3

StudioGAN is a Pytorch library providing implementations of representative Generative Adversarial Networks (GANs) for conditional/unconditional image generation.

Project last updated:08/09/24

GitHub Stars

3.5K

Forks

344

Contributors

13

License

Other

Why we included this project

Most GAN research starts with pulling together code from a stack of different repositories, each with its own quirks. StudioGAN bundles seven GAN architectures, including BigGAN, StyleGAN2 and StyleGAN3, along with pluggable conditioning methods, adversarial losses, and regularization modules that you wire together through YAML config files. That makes it handy for researchers who want to test a new idea against established baselines without rebuilding the infrastructure, and the built-in evaluation metrics (IS, FID, PRDC, IFID) plus benchmark results on CIFAR10, ImageNet, AFHQv2, and FFHQ give everyone the same measuring stick. The work is published at TPAMI and the authors provide pre-trained checkpoints on Hugging Face, so verifying that a run reproduces is fairly painless.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category