StyleGAN.pytorch

[ChineseGirl Dataset]

Thanks to dataset provider:Copyright(c) 2018, seeprettyface.com, BUPT_GWY contributes the dataset.

Features

• Progressive Growing Training
• Exponential Moving Average
• Equalized Learning Rate
• PixelNorm Layer
• Minibatch Standard Deviation Layer
• Style Mixing Regularization
• Truncation Trick
• Multi-GPU Training
• FP-16 Support
• Conditional GAN

How to use

Requirements

• yacs
• tqdm
• numpy
• torch
• torchvision

Running the training script:

Train on CelebA from scratch:

python train.py --config configs/sample.yaml

Using trained model:

Resume training from a checkpoint (start form 128x128):

python train.py --config config/sample.yaml --start_depth 5 --generator_file [] [--gen_shadow_file] --discriminator_file [] --gen_optim_file [] --dis_optim_file []

Thanks

Please feel free to open PRs / issues / suggestions here.

Reference

• stylegan[official]: https://github.com/NVlabs/stylegan
• pro_gan_pytorch: https://github.com/akanimax/pro_gan_pytorch
• pytorch_style_gan: https://github.com/lernapparat/lernapparat