A Simple pytorch implementation of GradCAM, and GradCAM++
Project description
A Simple pytorch implementation of GradCAM[1], and GradCAM++[2]
Installation
pip install pytorch-gradcam
Supported torchvision models
- alexnet
- vgg
- resnet
- densenet
- squeezenet
Usage
please refer to example.ipynb
for general usage and refer to documentations of each layer-finding functions in utils.py
if you want to know how to set target_layer_name
properly.
Use your own model and layer:
model = MyModel()
target_layer = model.my_submodule
gradcam = GradCAM(model, target_layer)
References:
[1] Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localization, Selvaraju et al, ICCV, 2017
[2] Grad-CAM++: Generalized Gradient-based Visual Explanations for Deep Convolutional Networks, Chattopadhyay et al, WACV, 2018
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
pytorch-gradcam-0.2.1.tar.gz
(6.0 MB
view hashes)