av-savchenko/face-emotion-recognition

Is it possible to predict action units?

tvaranka opened this issue · 4 comments

In the paper "Frame-level Prediction of Facial Expressions, Valence, Arousal and Action Units for Mobile Devices" predictions are made for AUs. I wonder if this is possible with the code provided here?

Thanks for your question. Yes, the abaw5_expr_va_au_train.ipynb contains all the required information to reproduce our paper

Thanks for the quick answer! For inference, which model would you recommend for the best AU performance?

Please, take a look at the above-mentioned ipynb file, there are several markdown comments after section "AU (Action Units)" that compare performance for several pre-trained models. I believe, enet_b0_8_best_vgaf works better

Thanks a lot!