tdeboissiere / Vgg16cam Keras
Keras implementation of the VGG16-CAM model
Stars: ✭ 190
Programming Languages
python
139335 projects - #7 most used programming language
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.
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].