/StyleSpace-pytorch

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

Primary LanguageJupyter Notebook

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

An image

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