genforce / idinvert_pytorch

[ECCV 2020] In-Domain GAN Inversion for Real Image Editing (PyTorch code)
https://genforce.github.io/idinvert/
MIT License
407 stars 54 forks source link

Invert the image into the W space #20

Open falloncandra opened 3 years ago

falloncandra commented 3 years ago

Hi, I would like to invert the image into the W space of size (1, 512). Could you please tell me how to do it?

I have taken a look at the code, but I think the StyleGANEncoder.encode always encode the image into W+ space of size (1, 14, 512)?

Thank you very much for your help!

ShenYujun commented 3 years ago

W space is not representative enough (since all layers will be using the same w) to fully reconstruct the input.