Skip to content

tjGThub/Emotion-Detector

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 

Repository files navigation

Emotion Detector

This program takes a jpg image as an input and determines the emotion of the person inside the image. It is currently able to detect only anger, happy and neutral emotions. This model is trained using only anger, happy and neutral data of FER-2013 dataset from Kaggle. Further improvements will be done to improve the models capability.

Motivation

To improve interaction between human and machine.

Network Architecture

input -> conv(relu) -> maxpool -> batchNorm -> conv(relu) -> conv(relu) -> maxpool -> fc(tanh) -> fc(tanh) -> softmax(mxcent) -> output

Steps

  1. Download the whole folder.
  2. Download the FER-2013 dataset.
  3. Import pom.xml as project in IntelliJ.
  4. Once the dependencies are resolved, open the test1.java file.
  5. Copy and paste angry, neutral and happy files for test and train into the resource file and rename the file path for training purposes. (rename the modelFilename to create a new model as the default model has been created.)
  6. Change image path to own image path under testImage() function.
  7. Run the program.
  8. The output stating the emotion of the person will be shown in the console.

Future Development

  1. Ability to detect more emotions.
  2. Ability to perform localization using bounding boxes.
  3. Apply transfer learning.
  4. Ability to detect emotions using webcam video as input.

Contact Information

  1. Tang Jie Email: [email protected]
  2. Muhammad Khairul Asyraf bin Suaimi Email: [email protected]
  3. Muhamad Noorazizi Bin Abd Ghani Email: [email protected]

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages