Implementation of the VGG-CAM model with keras
Source
Original matlab implementation and paper here.
Requirements
- keras with theano backend (keras version 1.0.4 or later)
- h5py
- numpy
- matplotlib
- opencv3
External data
Download the keras vgg16 weights.
Usage
- Use the
train_VGGCAM
function to fine tune the VGG16 model on your data. You should write your own code to feed the data into the network. - Use the
plot_classmap
function to then plot the class activation map on an image specified by its path.
Example plot
On the Kaggle StateFarm dataset:
The network is focusing on the cell phone the driver is holding, regardless of driver ID or hand position.