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.txtRun the file.
cd train_files
python ResEmoteNet_train.pyAll 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%)
This repository is licensed under the MIT License. See the LICENSE file for more details.