chavdim / facial_expressions_cam

visualisation of areas used for classification.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Facial expressions class activation maps

visualisation of areas used for classification by a cnn. done by highlighting the class activation maps (cam).

facial expression classifier model trained on fer2013 dataset. validation error ~ 0.55

based on work: Learning Deep Features for Discriminative Localization Zhou et al. (2016) http://cnnlocalization.csail.mit.edu/

Required:

numpy tensorflow keras sklearn matplotlib scipy

** requires fer2013 dataset **

Running

run main.py to produce plots. loading dataset may take a few minutes.

Sample outputs

alt text alt text alt text alt text

About

visualisation of areas used for classification.

License:MIT License


Languages

Language:Python 100.0%