Style-Based GAN in PyTorch

Implementation of A Style-Based Generator Architecture for Generative Adversarial Networks (https://arxiv.org/abs/1812.04948) in PyTorch

Dependencies

• Python 3.6.8
• PyTorch 1.3

Dataset

FFHQ

Usage

Data Pre-processing

Extract training images:

$ python pre_process.py

Train

$ python train.py

If you want to visualize during training, run in your terminal:

$ tensorboard --logdir runs

Acknowledge

Most codes are borrowed from https://github.com/rosinality/style-based-gan-pytorch.