Performing facial emotion recognition by training with custom CNN based model, transfer learning with Mobilenet pretrained model and fine tunning
Facial Emotion recognition dataset contains 7 classes (angry, happy, sad, surprise, disgust, fear, neutral)
Performed training on the data using simple convolutional neural networks based model, mobilenet pretrained model, and fine tunning it.
Models are trained for 40 epochs
Model | training accuracy | test accuracy |
---|---|---|
simple CNN model | 65% | 61.5% |
MobileNet | 59% | 53.3% |
Fine Tunning | 85% | 63.8% |
Confusion Matrices for all three models is as follows
Simple CNN based model
MobileNet model
Fine Tunning