rosinality/style-based-gan-pytorch

Health v2 · maintenance only

Flags: no_releases no_license

Adoption not part of the score

1110 stars · 229 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

A PyTorch implementation of the StyleGAN paper (A Style-Based Generator Architecture for Generative Adversarial Networks), including training scripts and pretrained checkpoints up to 1024px. It supports LMDB-prepared datasets, style mixing, and R1 regularization with progressive training schedules.

Use cases

• train a StyleGAN model on my own image dataset
• generate high-resolution faces with StyleGAN in PyTorch
• reproduce the StyleGAN paper results
• download pretrained StyleGAN checkpoints for FFHQ
• experiment with style mixing in generative adversarial networks

When to choose

• you want a straightforward, readable PyTorch implementation of the original StyleGAN
• you need pretrained checkpoints for FFHQ-style face generation
• you are learning how style-based generators and progressive training work

When to avoid

• you need StyleGAN2 or StyleGAN3 features and fixes
• you want a production-grade, actively maintained training framework
• you need multi-GPU distributed training out of the box

Facets

library · maturity maintenance

machine-learning deep-learning image-processing data-science deep-learning machine-learning image-processing artificial-intelligence python stylegan gan pytorch generative-models image-generation research-code gpu linux docker

Member repositories

For agents

markdown · JSON · MCP: product_card(name="rosinality/style-based-gan-pytorch")