get latent code of image

mit-han-lab / data-efficient-gans

[NeurIPS 2020] Differentiable Augmentation for Data-Efficient GAN Training

https://arxiv.org/abs/2006.10738

BSD 2-Clause "Simplified" License

1.28k stars 175 forks source link

get latent code of image #37

Closed molo32 closed 3 years ago

molo32 commented 3 years ago

I trained a model in diffargument-stylegan2 I implement a function to get latent code from an image? can you please show me an example to invert an image.

zsyzzsoft commented 3 years ago

This is not directly supported, but you may try stylegan2's projection code though it requires some modifications.