Skip to content

manhminno/Emotion-Recognition

Repository files navigation

Emotion Recognition - Real-time detection

  • Emotion recognition using haar-like and CNN model.
  • Using haar-like feature to detect face in image
  • After detecting face, put it into CNN network to recognize emotions

1. Load data:

  python load_dataset.py

2. Train model:

  • Use gpu to train, if don't have gpu you don't use .cuda () in the lines
  python train.py

3. Enjoy result:

* Detect from webcam: python main.py --mode Webcam
* Detect from image: python main.py --mode Image --path (Img_file)

4. Demo:

Here I am using Vietnamese, you can replace emotions with your language in label2id

About

Emotion recognition using haar-like and CNN model.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published