Implementation of the VGG-CAM model with keras
Source
Original matlab implementation and paper here.
Requirements
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.
The network is focusing on the cell phone the driver is holding, regardless of driver ID or hand position.