Emotion classification of human facial images by a neural network
DOI:
https://doi.org/10.55632/pwvas.v94i1.867Keywords:
Machine learning, Convolutional neural network, Image classificationAbstract
Facial recognition using artificial neural networks is a biometric technology currently being used in fields such as cybersecurity and criminal investigation. We sought to automatically distinguish between an image of a happy human face and an image of a sad human face with predictions that are better than random guesses. We trained a machine learning model (VGG16, a type of convolutional neural network) on a public image dataset of 12,000 human faces. The resulting model predicted the emotion label 93% of time when shown a test set of 200 images that the model had never seen during training. The results show that the VGG16 convolutional neural network learned features from our data that produced an output which was sufficient to train the additional layers of the model to perform our task at 93% accuracy. We suspect this is because VGG16 was already familiar with features that it learned from ImageNet (a separate dataset of over 2 million images). It is currently unknown whether our model can predict labels for new images outside of the FER13 dataset, but preliminary tests show promising results. Our model was trained using a cloud computing service and a relatively small amount of data, indicating that these kinds of results are easily obtainable by all.
Downloads
Published
How to Cite
Issue
Section
License
![Creative Commons License](http://i.creativecommons.org/l/by-nc/4.0/88x31.png)
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
Proceedings of the West Virginia Academy of Science applies the Creative Commons Attribution-NonCommercial (CC BY-NC) license to works we publish. By virtue of their appearance in this open access journal, articles are free to use, with proper attribution, in educational and other non-commercial settings.