Skip to content

Performing facial emotion recognition with seven facial expressions as classes by training a custom CNN based model and also transfer learning

Notifications You must be signed in to change notification settings

Sravan4465/Facial_emotion_recognition_using_Transfer_learning

Repository files navigation

Facial_emotion_recognition_using_transfer_learning

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

image

MobileNet model

image

Fine Tunning

image

About

Performing facial emotion recognition with seven facial expressions as classes by training a custom CNN based model and also transfer learning

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published