StyleGAN2 Tensorflow 2.0 Save

StyleGAN 2 in Tensorflow 2.0

Project README

StyleGAN2 Tensorflow 2.0

Unofficial implementation of StyleGAN 2 using TensorFlow 2.0. (Compatible with up to 2.5)

Original paper: Analyzing and Improving the Image Quality of StyleGAN


This implementation includes all improvements from StyleGAN to StyleGAN2, including:

Modulated/Demodulated Convolution, Skip block Generator, ResNet Discriminator, No Growth,

Lazy Regularization, Path Length Regularization, and can include larger networks (by adjusting the cha variable).

Image Samples

Trained on Landscapes for 3.48 million images (290k steps, batch size 12, channel coefficient 24): To clarify, 3.48 million images were shown to the Discriminator, but the dataset consists of only ~14k images. Thus, of those 3.48 million images, most are repeats of already seen images.

Teaser image

Mixing Styles:

Teaser image

Before Running

Please ensure you have created the following folders:

  1. /Models/
  2. /Results/
  3. /data/

Additionally, please ensure that your folder with images is in /data/ and changed at the top of

For pretrained models, download the pretrained models from here

Then, adjust the main code at the bottom to use model.load(model_num), where model_num = the number from the downloaded model.

Open Source Agenda is not affiliated with "StyleGAN2 Tensorflow 2.0" Project. README Source: manicman1999/StyleGAN2-Tensorflow-2.0
Open Issues
Last Commit
2 years ago

Open Source Agenda Badge

Open Source Agenda Rating