genforce / interfacegan

[CVPR 2020] Interpreting the Latent Space of GANs for Semantic Face Editing
https://genforce.github.io/interfacegan/
MIT License
1.51k stars 281 forks source link

Pretrained Model on pggan #88

Open RamboRick opened 3 years ago

RamboRick commented 3 years ago

First, thank you for any suggestions.

Based on the guide, first we need to download the pre-trained pggan model. I found one that is in the here and rename it to pggan_celebahq.pth and put it in the models/pretrain folder.

And then I run the code below:

LATENT_CODE_NUM=10
python edit.py \
    -m pggan_celebahq \
    -b boundaries/pggan_celebahq_smile_boundary.npy \
    -n "$LATENT_CODE_NUM" \
    -o results/pggan_celebahq_smile_editing

But this will post a warning saying that 'No pre-trained model will be loaded'.

Coudl you please give me some suggestions about this or I missed something here?