Implementation of StyleSpace Analysis: Disentangled Controls for StyleGAN Image Generation in PyTorch

StyleSpace Analysis: Disentangled Controls for StyleGAN Image Generation

Implementation of StyleSpace Analysis: Disentangled Controls for StyleGAN Image Generation (https://arxiv.org/pdf/2011.12799.pdf) in PyTorch

This implementation is mostly relied on rosinality's stylegan2-pytorch

Requirements

I have tested on:

• PyTorch 1.3.1
• CUDA 10.1

Usage

For the index and channel, please check the paper (https://arxiv.org/pdf/2011.12799.pdf), e.g., (11_286), channel 286 of generator level 11.

FFHQ

• Firstly, you should download pretrained model from here and place the stylegan2-ffhq-config-f.pkl into pretrained folder.
• Open the notebook StyleSpace_FFHQ.ipynb

Car

• Comming soon

LSUN

• Comming soon

Credit

The pretrained model weights are from https://github.com/NVlabs/stylegan2 and converted with https://github.com/rosinality/stylegan2-pytorch