search

yunjey / stargan

StarGAN - Official PyTorch Implementation (CVPR 2018)
MIT License
5.23k stars 970 forks source link

About attributes during inference #97

Open dagongji10 opened 5 years ago

dagongji10 commented 5 years ago

I have download the pre-trained model "celeba-256x256-5attrs", and I can use it for my own images. But the results seems only change in fixed 5 attributes: 'Black_Hair', 'Blond_Hair', 'Brown_Hair', 'Male', 'Young'. Can I do some change? Maybe the num of attr couldn't be change, I wonder id I can use other attr such as 'Bald'? Or, I must train it with new attrs?

yunjey commented 5 years ago

@dagongji10 You should re-train the StarGAN for different attributes. The provided model checkpoints are trained on five attributes (hair colors, gender, age).