search

sajjjadayobi / FaceLib

Face Analysis: Detection, Age Gender Estimation & Recognition
MIT License
294 stars 52 forks source link

How to train Face Expression Recognition? #4

Closed DWCTOD closed 3 years ago

DWCTOD commented 4 years ago

Hi, hangsome boy, thanks for your nice work , could you share me how to train Facial Expression Recognition. I didn't find something about these( Dataset and train code). Hope your reply , thanks.

sajjjadayobi commented 3 years ago

it's UTKFace DataSet and trained with mobile net and resnet34 as the backbone and you can train with other backbones in FacialExpression/models/resnet.py it's a classification and the loss is cross-entropy

DWCTOD commented 3 years ago

it's UTKFace DataSet and trained with mobile net and resnet34 as the backbone and you can train with other backbones in FacialExpression/models/resnet.py it's a classification and the loss is cross-entropy

Thank you very much