Skip to content
No description, website, or topics provided.
Jupyter Notebook Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
__pycache__
assets
samples
.DS_Store
.gitignore
README.md
gradcam.py
gradcam_in_practice.ipynb

README.md

Grad-CAM in PyTorch

This is a Grad-CAM implementation in Pytorch inspired by @kazuto1011's work (https://github.com/kazuto1011/grad-cam-pytorch).

I modified his code in order to deal with following needs

  • to process multiple images based on their own class labels
  • initialize gradcam with target layer to avoid out-of-memory error
  • add bounding boxes using skimage

Dependencies

  • python 3.6
  • pytorch 0.4.0
  • torchvision 0.2.1
  • skimage 0.14.0

Example (resnet152, 'layer4.2')

result

You can’t perform that action at this time.