#99 · Primary category: Image Generation
PyTorch-StudioGAN
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
stable-diffusion-webui
Stable Diffusion web UI
ComfyUI
The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface.
diffusers
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
upscayl
🆙 Upscayl - #1 Free and Open Source AI Image Upscaler for Linux, MacOS and Windows.
InvokeAI
InvokeAI is a leading open-source creative engine for Stable Diffusion, offering an industry-leading web UI for generating and editing visual media.