search

omertov / encoder4editing

Official implementation of "Designing an Encoder for StyleGAN Image Manipulation" (SIGGRAPH 2021) https://arxiv.org/abs/2102.02766
MIT License
945 stars 154 forks source link

How to train on other animal dataset #26

Closed UniBester closed 3 years ago

UniBester commented 3 years ago

If I want to train this model on other dataset like animal faces, but I don't have the correspond pretrained stylegan model. How can I train the stylegan model by myself?

omertov commented 3 years ago

Hi @UniBester!

As this repository is intended for training an encoder for a pre-trained StyleGAN2 model, I would suggest using either rosinality's stylegan2-pytorch repository or one of the official NVIDIA repositories to train the StyleGAN based on your images.

After you obtain a trained model you can use the repository to train the encoder :)

UniBester commented 3 years ago

I‘ve got it. Thanks a lot

omertov commented 3 years ago

Closing the issue, feel free to open a new issue in case someting else is needed :)