Skip to content

Facial emotion recognition by using convolutional neural network (CNN)

License

Notifications You must be signed in to change notification settings

ErfanMomeniii/face-emotion-recognition

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Face Emotion Recognition

This project is an implementation of facial emotion recognition using convolutional neural networks. It is capable of detecting and categorizing facial emotions into the following categories:

  • Angry
  • Disgusted
  • Fearful
  • Happy
  • Neutral
  • Sad
  • Surprised

Training Models

For training models, I utilized Kaggle data. The prediction accuracy of the models is above 87 percent, as shown below:

You can improve the accuracy by adjusting hyperparameters or implementing data augmentation techniques.

Installation

  1. Clone this project:
git clone https://github.com/ErfanMomeniii/face-emotion-recognition.git
  1. Install the required libraries:
pip install -r requirements.txt
  1. Download the Kaggle data and store it in the /dataset/ folder.
  2. Train the model and run it to capture and recognize facial emotions:
python face-emotion-recognition.py

Data Reference

Data source: Emotion Detection FER Dataset

Related Papers

License

face-emotion-recognition is released under MIT License.