smaranjitghose / DeepPixel

An open-source Python package for making computer vision and image processing simpler

MIT License

22 stars 37 forks source link

Added ScoreCAM #122

Closed nkpro2000sr closed 4 years ago

nkpro2000sr commented 4 years ago

116

[X] GradCAM
[ ] SmoothGrad
[X] GradCAM++
[X] ScoreCAM

reference : https://arxiv.org/pdf/1910.01279.pdf

demo : https://colab.research.google.com/drive/18YNBud4JcPKECyn7Eg6OwBXUV5RajVua?usp=sharing

Basic Demo

from deeppixel.cam import *
scorecam = ScoreCAM("model.h5", "last_conv_layer_name")
# heat map
scorecam.plot_heatmap("image.png")
# superimposed with image
scorecam.show("image.png")

SanjayMarreddi commented 4 years ago

@nkpro2000sr Well done! Good to see Improvements suggested by @smaranjitghose earlier being implemented here too !