Very simple Emotion Recognition

In this article, I will try to provide the simplest TensorFlow Emotion Recognition implementation using TensorFlow.


First, the data

We will use FER2013 Dataset.

Then, the code

We use MobileNetV2 Model in order to keep the model light and usable on small devices.

Folder structure

  • /
  • /emotions/[test, train]/[angry, disgust, fear, happy, neutral, sad, surprise]/+.jpg

Fitting result

I obtained an accuracy of 0.4873 after 15 epochs, however progression is not over.

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store