cnncam: an open source library to produce attention heatmaps in a single line of code

Hello hello!

I'm here to share a library I built for interpretability of keras computer vision models that contain convolutional layers. It produces GradCAM heatmaps in a single function call:

from cnncam import display_heatmap

display_heatmap(model=model, # your keras model
img=img, # your image
predicted_class=pred, # your models prediction for the image
layer_name='block5_conv3', # the layer you would like to see
alpha=0.6 # opacity of heatmap overlayed on image
)

The output represents a localization of features that were found to …

attention call code computer computer vision computervision function hello image import interpretability keras library line open source vision vision models