A new network that helps in extracting facial features and predict the emotion labels.
The emotion labels in this project are:
- Happiness 😀
- Surprise 😦
- Anger 😠
- Sadness
☹️ - Disgust 🤢
- Fear 😨
- Neutral 😐
- Create a Conda environment.
conda create --n "fer"
conda activate fer
- Install Python v3.8 using Conda.
conda install python=3.8
- Clone the repository.
git clone https://github.com/ArnabKumarRoy02/ResEmoteNet.git
- Install the required libraries.
pip install -r requirement.txt
Run the file.
cd train_files
python ResEmoteNet_train.py
All of the checkpoint models for FER2013, RAF-DB and AffectNet-7 can be found here.
- FER2013:
- Testing Accuracy: 79.79% (SoTA - 76.82%)
- CK+:
- Testing Accuracy: 100% (SoTA - 100%)
- RAF-DB:
- Testing Accuracy: 94.76% (SoTA - 92.57%)
- FERPlus:
- Testing Accuracy: 91.64% (SoTA - 95.55%)
- AffectNet (7 emotions):
- Testing Accuracy: 72.93% (SoTA - 69.4%)
- ExpW:
- Testing Accuracy: 75.67%
This repository is licensed under the MIT License. See the LICENSE file for more details.