search

pbaylies / stylegan-encoder

StyleGAN Encoder - converts real images to latent space
Other
740 stars 182 forks source link

StyleGAN2 support #47

Closed eladrich closed 4 years ago

eladrich commented 4 years ago

Hi, Really like what you did here, I was wondering if you're planning to train a new ResNet model for StyleGAN2, or could guide me through the process, it would be interesting to see how it compares :)

pbaylies commented 4 years ago

I do also have a repo that supports StyleGAN 2 now, based on skyflynil's repo and a few others. The ResNet code is basically the same, though; you just need to run the train_resnet.py script for a few days :)

eladrich commented 4 years ago

Nice! Do you also happen to have a pretrained model there to save these few days? 🙃

pbaylies commented 4 years ago

I have one in my repo for the WikiArt model I trained, but not for the default NVIDIA models.