StyleSpace-pytorch

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

GitHub

https://github.com/xrenaa/StyleSpace-pytorch