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.
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.1.0.tar.gz
(7.3 kB
view hashes)
Built Distribution
Close
Hashes for pytorch_gradcam-0.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e58a9eb187f11c321b9948910b8075f024869b0f05481c88842745a9a571a1b2 |
|
MD5 | 188ae72ba8e7ac8d2e1dec34929c33bb |
|
BLAKE2b-256 | c8dd358e293eb97a515815380e3b3e6d8a959a9afafc69683b96253b8248f104 |