Skip to content

Real-time face detection and emotion/gender classification using fer2013/imdb datasets with a keras CNN model and openCV.

License

Notifications You must be signed in to change notification settings

xxmiprai/face_classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Face classification and detection from the B-IT-BOTS robotics team.

Real-time face detection and emotion/gender classification using fer2013/IMDB datasets with a keras CNN model and openCV.

  • IMDB gender classification test accuracy: 96%.
  • fer2013 emotion classification test accuracy: 66%.

Emotion/gender examples alt tag

Live demo

Guided back-prop alt tag

Emotion/gender classification of the B-IT-BOTS robotics team :) alt tag

Instructions

To train previous/new models for emotion classification:

  • Download the fer2013.tar.gz file from here

  • Move the downloaded file to the datasets directory inside this repository.

  • Untar the file:

tar -xzf fer2013.tar

  • Run the train_emotion_classification.py file

python3 train_emotion_classifier.py

To train previous/new models for gender classification:

  • Download the imdb_crop.tar file from here (It's the 7GB button with the tittle Download faces only).

  • Move the downloaded file to the datasets directory inside this repository.

  • Untar the file:

tar -xfv imdb_crop.tar

  • Run the train_emotion_classification.py file

python3 train_emotion_classifier.py

About

Real-time face detection and emotion/gender classification using fer2013/imdb datasets with a keras CNN model and openCV.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.3%
  • Dockerfile 0.7%