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]