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
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.