Puzer / stylegan-encoder

StyleGAN Encoder - converts real images to latent space
Other
1.07k stars 166 forks source link

shape (1, 12, 512) to (1, 18, 512)? #35

Open ahmedshingaly opened 4 years ago

ahmedshingaly commented 4 years ago

how can I create stylegan model with (1, 18, 512) my stylegan model is creating shape (1, 12, 512) and I cannot find the latent space developed by Puzer because of shape difference

in more details: my model produce shape with (1, 12, 512) using (https://github.com/NVlabs/stylegan) but when I use stylegan encoder (https://github.com/Puzer/stylegan-encoder) to find latent space it requires (1, 18, 512), do you have any idea how can I produce (1, 18, 512) model shapes instead of (1, 12, 512)?

eps696 commented 4 years ago

number of layers depends on the model resolution. (1,18,512) is for 1024x1024, (1,12,512) is for 128x128. you should either take bigger model, or manually change that hard-coded shape.