smaranjitghose / DeepPixel

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

MIT License

22 stars 37 forks source link

Added GramCAM++ #120

Closed nkpro2000sr closed 4 years ago

nkpro2000sr commented 4 years ago

116

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

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

demo : https://colab.research.google.com/drive/15LuhriKDernYecTP_kueQpR_8h3LgvJE?usp=sharing

Basic Demo

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

SanjayMarreddi commented 4 years ago

@nkpro2000sr Good Work !

smaranjitghose commented 4 years ago

@nkpro2000sr

Some suggestions for improvements:

Compare GRADCAM++ and GRADCAM over the same images
After training the model, save it as h5
Now read that, give the layer and your image
For the images, (if you are testing CAMs), it's suggested to do on X-Ray Images. Try on Pneumonia Dataset

nkpro2000sr commented 4 years ago

@smaranjitghose

your suggestions are implemented. please check the demo now.