How to train on my custom data?

petercunha / Emotion

:smile: Recognizes human faces and their corresponding emotions from a video or webcam feed. Powered by OpenCV and Deep Learning.

MIT License

453 stars 169 forks source link

How to train on my custom data? #5

Closed akshatbhargava123 closed 6 years ago

akshatbhargava123 commented 6 years ago

Please tell something about how to train the model on my custom data? This project is awesome, I really love it but it's useless unless we don't have ability to train it ourselves.

petercunha commented 6 years ago

To train 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
Download train_emotion_classifier.py from orriaga's repo here
Run the train_emotion_classification.py file: python3 train_emotion_classifier.py