#340 · Primary category: Computer Vision

grad-cam

convolutional-neural-networks deep-learning grad-cam heatmap iccv17 interpretability visual-explanation

[ICCV 2017] Torch code for Grad-CAM

Project last updated:09/17/22

GitHub Stars

1.7K

Forks

237

Contributors

5

License

Other

Why we included this project

This is the reference implementation of Grad-CAM, the method that turns a CNN's decision into a heatmap showing which image regions drove the prediction. Written in Torch/Lua and built around Caffe models, it is best treated as a historical and pedagogical resource rather than something to drop into a modern PyTorch or TensorFlow stack. The repo covers classification, visual question answering, and captioning, with simple command-line flags for selecting a layer and saving heatmaps, which makes it easy to experiment with the technique. For anyone studying interpretability or reproducing the ICCV 2017 results, having the authors' own code is a convenient, readable way to see how the method works before porting it.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category