Stylegan on PyTorch

My implementation of StyleGAN on PyTorch, up to 256x256 generator.

• Original Paper
• Official implementation(TensorFlow)

Run

Docker image

There's a Dockerfile including all requirements. NGC Account is required for base image.

https://www.nvidia.com/en-us/gpu-cloud/

Settings

By default, the networks can generate up to 256x256 images.

1. Place images under ../images/{label}
2. Edit settings.json
3. Run python train_gay.py

Directory structure must be like below:

├─ images
|  ├─ ffhq
|  | ├─ image1.png
|  | └─ ...
|  ├─ your custom label1
|  | ├─ image1.png
|  | └─ ...
|  ├─ your custom label2
|  | ├─ image1.png
|  | └─ ...
|  └─ ...
└─ stylegan(this repository)

Result

Training time

It took about 3 days to train 256x256 image generator on single RTX2080 machine.