Skip to content

Latest commit

 

History

History
20 lines (16 loc) · 1.21 KB

README.md

File metadata and controls

20 lines (16 loc) · 1.21 KB

StyleGAN

Concise implementation of StyleGAN and ProgressiveGAN in PyTorch.

Code organization:

  • stylegan: neural net classes, dataloaders, training script, config file
  • notebooks: sampling and inversion demo using a trained StyleGAN model

Training:

  1. Configure the training parameters in ./stylegan/config.py
  2. Run:
    python ./stylegan/train.py
    

References:

  1. Karras, Laine, Aila - A style-based generator architecture for generative adversarial networks. [arXiv 2018] [CVPR 2019]
  2. Karras, Aila, Laine, Lehtinen - Progressive growing of gans for improved quality, stability, and variation. [arXiv 2017] [ICLR 2018]
  3. Zhang - Making convolutional networks shift-invariant again. [ICML 2019]
  4. Mescheder, Geiger, Nowozin - Which training methods for GANs do actually converge?. [ICML 2018] [arXiv 2018]