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.2.0.tar.gz
(6.4 MB
view hashes)
Built Distribution
Close
Hashes for pytorch_gradcam-0.2.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 97050fb474b10272cc63885a55d2b180d472b23eacdc7b507d5f2d7fc6eb167a |
|
MD5 | ead83f2ce640c6f1ef0d6e81d82ac275 |
|
BLAKE2b-256 | 6aaaa87cd1619418cdfca1f7db1edbe4b9ef1b20f90f1fd66e78201ba5d5e6f8 |