in extract_edit_directions.py

mit-han-lab / anycost-gan

[CVPR 2021] Anycost GANs for Interactive Image Synthesis and Editing

https://hanlab.mit.edu/projects/anycost-gan/

MIT License

776 stars 97 forks source link

in extract_edit_directions.py #21

Closed Winnie202 closed 2 years ago

Winnie202 commented 2 years ago

Hi，I am very confused，Attribute-predictor is poorly explained，How do I get an Attribute-predictor .pt file on different datasets？

tonylins commented 2 years ago

Hi, for the face-related dataset (e.g., FFHQ), we have a supervised dataset CelebA (https://mmlab.ie.cuhk.edu.hk/projects/CelebA.html), which contains face images and associated labels (e.g., is smiling, is wearing sunglasses, etc.). We used the dataset to train an attribute predictor.

For other datasets, you can find similar labeled data to train the predictor.