akanimax / pro_gan_pytorch

Unofficial PyTorch implementation of the paper titled "Progressive growing of GANs for improved Quality, Stability, and Variation"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Can i use well trained progan generator do gan inversion?

MHX1203 opened this issue · comments

I have used my own customal dataset train a progan generator, and now, i 'd like to use the progan do gan inversion, so i can get a latent code of related image through gan inversion.

can anyone help me, what should i do this.

thanks very much.

Hi there.
I believe yes you can very well use a trained Generator from this repo to do it. Easiest thing to check out could be by doing the following:

  1. Download the Metfaces model displayed on repo.
  2. Write a small snippet to load the model. Refer to this script for an example on how to load a trained model.
  3. Turn off model gradients and optimize for a latent code taking your own image. (The inversion process).

I unfortunately don't have time to work on this feature, but please feel free to post here if you need any help doing it code-wise.
Cheers 🍻!
@akanimax

Hi there ...
I am closing this because of inactivity. Please feel free to open again if you are still facing some problems.
Cheers 🍻!
@akanimax