how can i get the train code

1adrianb / face-alignment

:fire: 2D and 3D Face alignment library build using pytorch

https://www.adrianbulat.com

BSD 3-Clause "New" or "Revised" License

6.99k stars 1.34k forks source link

Open zhouyufei0001 opened 5 years ago

zhouyufei0001 commented 5 years ago

great jobs! but i can't see the train code,how can i get it. the speed of 2d face align is too slow,i want to train a simpler model to speed

fzyzcjy commented 5 years ago

rakadambi commented 4 years ago

andreasmarxer commented 4 years ago

rohitdavas commented 4 years ago

I don't know the purpose you want the model to train for but,

I was reading the code today, and I think if you want to train it for your purpose.

here is what I understood:

In api.pythe class FaceAlignment use the FAN model for inference loading the weights.
you can define a your method in it for training or whatever way you want it. train it. save the weights.