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 use e4e and pkl files #64

Closed leeisack closed 2 years ago

leeisack commented 3 years ago

I have a pkl file learned through stylegan2-ada. But I'm not sure how to use it for e4e through this. Also, it would be helpful if you tell me how to make a pkl file into a pt file.

omertov commented 3 years ago

Hi @leeisack! Our repo assumes you already have a checkpoint compatible with rosinality's stylegan2 repo, here is a relevant issue. basically you will need to convert the checkpoint using the convert_weights.py script from his repository.

LMK if there are problems with that! Best, Omer

leeisack commented 3 years ago

thanks for you'r answer. I am very interested in your thesis. Hopefully the code will be fixed as soon as possible. thanks