annloya / pytorch-grad-cam Goto Github PK

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 Resnet50 from torchvision. It will be downloaded when used for the first time. The code can be modified to work with any model.

Usage: python gradcam.py --image-path <path_to_image>

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