← all repositories

NVlabs/stylegan2-ada-pytorch

A PyTorch implementation of StyleGAN2-ADA, a generative adversarial network that uses adaptive discriminator augmentation to train high-quality image synthesis models with limited data.

4.5k stars Python Image · Video · Audio
stylegan2-ada-pytorch
Velocity · 7d
+2.3
★ / day
Trend
steady
star history

This repository is a faithful PyTorch reimplementation of StyleGAN2-ADA, a method for training generative adversarial networks with limited data. The approach uses adaptive discriminator augmentation to prevent overfitting of the discriminator, allowing the model to train effectively with fewer images. It supports training from scratch and fine-tuning existing models on new datasets, achieving strong results with sometimes an order of magnitude fewer training images than typically required.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.