search

eclique / pytorch-gradcam

PyTorch reimplementation of GradCAM saliency method.
25 stars 3 forks source link

GradCAM using densenet #1

Closed akhanss closed 5 years ago

akhanss commented 5 years ago

Thanks a lot for nice implementation. I am just wondering if you can add code snipppet for densenet as well. I tried it myself using your example but failed to manage that. Thanks in advance for your kind consideration.

eclique commented 5 years ago

Done. https://github.com/eclique/pytorch-gradcam/commit/7422b531ccc675adb92444dd0b460a1c226f3a1a