MarekKowalski / DeepAlignmentNetwork

A deep neural network for face alignment
MIT License
510 stars 136 forks source link

How to train a model of 5 landmarks #48

Open JuiceLemonLemon opened 5 years ago

JuiceLemonLemon commented 5 years ago

Hi ,

Thanks for your codes.

Now I want to train model of 5 landmarks, then how can I do it? Could you please give me some suggestion?

Thank you very much!

MarekKowalski commented 5 years ago

Hi,

Are those 5 landmarks among the ones in the 300-W set? If so, you can simply train a model with 68 and only use 5, there will be no performance impact.

If not, you will need to create your own dataset. The easiest way would be to follow the format used by the 300-W dataset. You will also need the "average" positions of those 5 landmarks, similarly to the meanFaceShape.npz present in the /data/ directory of the project.

Thanks,

Marek