Skip to content

Latest commit

 

History

History

fbp

README.md

Facial Beauty Prediction

Introduction

Developing machine learning models to recognize a person's beauty score from a given image. I have reimplemented several SOTA methods in this module.

If you are looking for a SOTA algorithm in FBP, please refer our new algorithm named ComboLoss!

demo

Performance

Results on SCUT-FBP5500

Model MAE RMSE PC
ShuffleNetV2 0.4389 0.3348 0.7639
ResNet18 0.2887 0.3814 0.8276
CRNet 0.2835 0.3677 0.8558
TransFBP 0.2805 0.3559 0.8519
HMTNet 0.2439 0.3226 0.8801
AaNet 0.2236 0.2954 0.9055
R^2 ResNeXt 0.2416 0.3046 0.8957
R^3CNN 0.2120 0.2800 0.9142
ComboLoss 0.2170 0.2742 0.9177

References

  1. Liang L, Lin L, Jin L, et al. SCUT-FBP5500: A diverse benchmark dataset for multi-paradigm facial beauty prediction[C]//2018 24th International Conference on Pattern Recognition (ICPR). IEEE, 2018: 1598-1603.
  2. Xu L, Xiang J, Yuan X. CRNet: Classification and Regression Neural Network for Facial Beauty Prediction[C]//Pacific Rim Conference on Multimedia. Springer, Cham, 2018: 661-671.
  3. Lin L, Liang L, Jin L, et al. Attribute-aware convolutional neural networks for facial beauty prediction[C]//Proceedings of the 28th International Joint Conference on Artificial Intelligence. AAAI Press, 2019: 847-853.
  4. Xu L, Fan H, Xiang J. Hierarchical Multi-Task Network For Race, Gender and Facial Attractiveness Recognition[C]//2019 IEEE International Conference on Image Processing (ICIP). IEEE, 2019: 3861-3865.
  5. Liu X, Li T, Peng H, et al. Understanding beauty via deep facial features[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition Workshops. 2019: 0-0.
  6. Liang L, Lin L, Jin L, et al. SCUT-FBP5500: A diverse benchmark dataset for multi-paradigm facial beauty prediction[C]//2018 24th International Conference on Pattern Recognition (ICPR). IEEE, 2018: 1598-1603.
  7. Lin L, Liang L, Jin L. Regression Guided by Relative Ranking Using Convolutional Neural Network (R3CNN) for Facial Beauty Prediction[J]. IEEE Transactions on Affective Computing, 2019.
  8. Lin L, Liang L, Jin L. R 2-ResNeXt: A ResNeXt-Based Regression Model with Relative Ranking for Facial Beauty Prediction[C]//2018 24th International Conference on Pattern Recognition (ICPR). IEEE, 2018: 85-90.