smaranjitghose / DeepPixel

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

MIT License

22 stars 37 forks source link

Added GradCAM #119

Closed nkpro2000sr closed 4 years ago

nkpro2000sr commented 4 years ago

116

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

reference : https://arxiv.org/abs/1610.02391

demo : https://colab.research.google.com/drive/1Or1rM73KPBSd1SZwLoFYPechAxPoopuh?usp=sharing

Basic Demo

from deeppixel.cam import *
gradcam = GradCAM(model, "last_conv_layer_name")
# heat map
gradcam.plot_heatmap(image)
# superimposed with image
gradcam.show(image)

smaranjitghose commented 4 years ago

Nice work @nkpro2000sr.