Skip to content

Emotion detection is topic of research now-a-days. This project builds a model that can detect emotions from face images using CNN., YoloV8 Architecture and then AutoML. The model is trained, validated, and tested on a preprocessed and augmented dataset. The best-performing model is used for emotion detection on unseen face images.

Notifications You must be signed in to change notification settings

prathmesh444/Emotion-Detection-using-Face-Recognition

Repository files navigation

Emotion Detection using Face Recognition

This project is created using facial expression dataset of 35000 images from kaggle->https://www.kaggle.com/datasets/jonathanoheix/face-expression-recognition-dataset.

It uses haar-cascade frontal-face model for face recognition task.


Model 1:-

In this model we fine Tuned this model MobileNet v2 through Transfer Learning concept and used keras for facial feature extraction and creating certain feature map.

For model-1 we have achieved 99% accuracy on train and 49% on test data which was a result of OVERFITTING.

You can study about this model in detail here:- https://github.com/prathmesh444/Emotion-Detection-using-Face-Recognition/blob/main/Detail%20Study%20on%20Model%201.pdf


Model 2:-

Furthur we improved the model's accuracy using Validation data. Additionally We used Data Augmentation and BatchImageGenerator to increase model's performance. This time we created out own CNN model and used OpenCV to deploy our model for realtime Emotion Detection.

For model-2 we have achieved 69% accuracy over validation data.

You can study about this model in detail here:- https://github.com/prathmesh444/Emotion-Detection-using-Face-Recognition/blob/main/Detail%20Study%20on%20Model%202.pdf


Model 3:- With the release of YOLO-v8 on 2023 we decided to train out our dataset using yolo-v8 extra classification model, we used ultranlytics Library for that.

For model-3 we managed to achieve 72.3% accuracy over validation data, which is the GLOBAL BEST ACCURACY on the current dataset.


Model 4:- We are working on furthur improvement by incorporating AutoML to generate more optimal CNN layers with auto hyper-parameter tuning.


1 1

About

Emotion detection is topic of research now-a-days. This project builds a model that can detect emotions from face images using CNN., YoloV8 Architecture and then AutoML. The model is trained, validated, and tested on a preprocessed and augmented dataset. The best-performing model is used for emotion detection on unseen face images.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published