search

face-analysis / emonet

Official implementation of the paper "Estimation of continuous valence and arousal levels from faces in naturalistic conditions", Antoine Toisoul, Jean Kossaifi, Adrian Bulat, Georgios Tzimiropoulos and Maja Pantic, Nature Machine Intelligence, 2021
https://www.nature.com/articles/s42256-020-00280-0
Other
267 stars 68 forks source link

8 classes are not clear #5

Closed GowthamGithubrepo closed 2 weeks ago

GowthamGithubrepo commented 3 years ago

can I know what each class number signifies. It is unclear from the repo code as what that expression is

antoinetlc commented 3 years ago

Hello,

Sorry I forgot to add it in the readme. The mapping from class number to expression is as follows.

For 8 emotions :

0 - Neutral 1 - Happy 2 - Sad 3 - Surprise 4 - Fear 5 - Disgust 6 - Anger 7 - Contempt

For 5 emotions :

0 - Neutral 1 - Happy 2 - Sad 3 - Surprise 4 - Fear

JeanKossaifi commented 3 years ago

It's also specified in the DataLoader: https://github.com/face-analysis/emonet/blob/47006243857718250907eed4c9b3c39e1578c955/emonet/data/affecnet.py#L10