tree: 7f4a2bbe8bcadb608eaa6cd4fcbb8ae952eff2cd [path history] [tgz]
  1. gradcam.py
  2. gradcam_demo.py
  3. README.md
  4. vgg.py
example/cnn_visualization/README.md

Visualzing CNN decisions

This folder contains an MXNet Gluon implementation of Grad-CAM that helps visualize CNN decisions.

A tutorial on how to use this from Jupyter notebook is available here.

You can also do the visualization from terminal:

$ python gradcam_demo.py hummingbird.jpg
Predicted category  : hummingbird (94)
Original Image      : hummingbird_orig.jpg
Grad-CAM            : hummingbird_gradcam.jpg
Guided Grad-CAM     : hummingbird_guided_gradcam.jpg
Saliency Map        : hummingbird_saliency.jpg

Output of gradcam_demo.py