vision-analytics / keras-heatmap-class-activation-maps

Project Details

Visualize class activation maps with TensorFlow(Keras)

Create a heatmap over the input image and debug your deep learning model!

Installation

Use the package manager pip to install requirements.

pip3 install -r requirements.txt

Usage

jupyter notebook
keras-heatmap-CAM-vgg16.ipynb #VGG16
keras-heatmap-CAM-xception.ipynb #Xception
keras-heatmap-CAM-custom-model.ipynb #custom model (Xception TL)

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.

License

MIT