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.

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.