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
intopretrained
folder. - Open the notebook
StyleSpace_FFHQ.ipynb
Car
- Comming soon
LSUN
- Comming soon