PyTorch implementation of Grad-CAM
Clone or download
jacobgil Merge pull request #5 from flyingpot/master
cam should be initialized to zero matrix. Modify example images
Latest commit e18e2c3 Jul 19, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
examples cam should be initialized to zero matrix. Modify example images Jan 7, 2018
README.md add guided backpropagation Jul 12, 2017
grad-cam.py cam should be initialized to zero matrix. Modify example images Jan 7, 2018

README.md

Grad-CAM implementation in Pytorch

What makes the network think the image label is 'pug, pug-dog' and 'tabby, tabby cat':

Dog Cat

Gradient class activation maps are a visualization technique for deep learning networks.

See the paper: https://arxiv.org/pdf/1610.02391v1.pdf

The paper authors torch implementation: https://github.com/ramprs/grad-cam

My Keras implementation: https://github.com/jacobgil/keras-grad-cam


This uses VGG19 from torchvision. It will be downloaded when used for the first time.

The code can be modified to work with any model. However the VGG models in torchvision have features/classifier methods for the convolutional part of the network, and the fully connected part. This code assumes that the model passed supports these two methods.


Usage: python grad-cam.py --image-path <path_to_image>

To use with CUDA: python grad-cam.py --image-path <path_to_image> --use-cuda