This is the repository containing Keras code for emotion detection of a group. This repository has Jupyter notebooks for creating various heatmaps, the two convolutional neural networks used in the paper. Data is assumed to be in the Data directory. The data couldn't be shared since we don't have permission to freely distribute the dataset.
The notebook Demo.ipynb has been obtained from the webpage for Emotion Recognition in the Wild via Convolutional Neural Networks and Mapped Binary Patterns
Weights for the models can be downloaded from here.
- Python
- Jupyter
- scikit-learn
- PIL
- tqdm
- Keras
- Numpy
- Scipy
- scikit-image
- Matplotlib
- guppy (for tracking memory footprint)
- Pycaffe
If you use this code, please cite the following:
@INPROCEEDINGS{8347115,
author={S. N. Shamsi and B. P. Singh and M. Wadhwa},
booktitle={2018 IEEE Winter Applications of Computer Vision Workshops (WACVW)},
title={Group Affect Prediction Using Multimodal Distributions},
year={2018},
pages={77-83},
doi={10.1109/WACVW.2018.00015},
month={March},}