Skip to content

junkwhinger/pytorch_gradcam

master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published