alwayslj/scorecam

the re-implementation of Score CAM with pytorch

Score CAM with pytorch

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

Requirements

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

How to use

You can use the CAM, GradCAM, GradCAM++, Smooth Grad-CAM++ and Score 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++
Score CAM

References

• Score-CAM: Improved Visual Explanations Via Score-Weighted Class Activation Mapping, Haofan Wang, Mengnan Du, Fan Yang, Zijian Zhang [paper]
• 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, [paper]
• Grad-CAM++: Improved Visual Explanations for Deep Convolutional Networks, Aditya Chattopadhyay, Anirban Sarkar, Prantik Howlader and Vineeth N Balasubramanian[paper]