Installation
Packages
sudo pip install -r requirements.txt
Get the training data
Go to https://www.dropbox.com/s/opuvvdv3uligypx/fer2013.tar and click Download.
Move the file into the root directory of this project and then run
wget https://www.dropbox.com/s/opuvvdv3uligypx/fer2013.tar
tar xf fer2013.tar
Training
wandb run python train_emotion_classifier.py
Testing
python run_classifier.py test.jpg
Things to try:
Look at the data!
- How is it distributed? How well would a random model do?
- Can the model learn on a tiny subset of the data?
Better models
- Better architecture
Data cleanup
- Normalize