Loading repository data…
Loading repository data…
StraysWonderland / repository
Jupyter Notebook containing a Convolutional Neural Network to classify emotions from facial expressions trained on the FER2013 dataset
This repository is a simple jupyter notebook that is trained on the Facial Expressions Recognition (FER2013) Dataset Currently it does not include any additional data manipluation to avoid overfitting, but this might be added in the future
If you want to start a pull request to contribute to hacktoberfest, either:
Simply load the FER2013 dataset from kaggle and place the csv in the data folder
Dataset can be downloaded from this link
the required virtual environment can be created from either the requirements.txt or the environment.yaml in the requirements folder
run
jupyter notebook
and go through all the codeblocks if you want to train the network from scratch.
the model also saves its architecture and weights to json and hdf file so they can be later on loaded for testing without re-training
to do so ; only run the blocks below the loading the model codeblock
Our network achieves around 65% test accuracy
Here is the plot for the loss function: