liaorongfan / SmoothGradCAMplusplus

The re-implementation of Smooth Grad-CAM++ with pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Smooth Grad-CAM++ with pytorch

The re-implementation of Smooth Grad-CAM++ with pytorch. This repo also includes the code of CAM, Grad-CAM and Grad-CAM++.

Requirements

  • python 3.x
  • pytorch >= 0.4
  • pillow
  • numpy
  • opencv
  • matplotlib

How to use

You can use the CAM, GradCAM, GradCAM++ and Smooth Grad-CAM++ as a model wrapper described in cam.py. Please see demo.ipynb for the detail.

Results

image
CAM
Grad-CAM
Grad-CAM++
Smooth Grad-CAM++

References

  • Smooth Grad-CAM++: An Enhanced Inference Level Visualization Technique for Deep Convolutional Neural Network Models,
    Daniel Omeiza, Skyler Speakman, Celia Cintas, Komminist Weldermariam [paper]
  • Learning Deep Features for Discriminative Localization, Bolei Zhou, Aditya Khosla, Agata Lapedriza, Aude Oliva, Antonio Torralba [paper]
  • Grad-CAM: Visual explanations from deep networks via gradient-based localization, Ramprasaath R. Selvaraju, Michael Cogswell, Abhishek Das, Ramakrishna Vedantam, Devi Parikh, Dhruv Batra, [arXiv]
  • Grad-CAM++: Improved Visual Explanations for Deep Convolutional Networks, Aditya Chattopadhyay, Anirban Sarkar, Prantik Howlader and Vineeth N Balasubramanian[arXiv]

About

The re-implementation of Smooth Grad-CAM++ with pytorch


Languages

Language:Jupyter Notebook 90.7%Language:Python 9.3%