1Konny / gradcam_plus_plus-pytorch

A Simple pytorch implementation of GradCAM and GradCAM++
359 stars 95 forks source link

Fixed wrong implementation of denominator in Eq.19 #3

Open pherrusa7 opened 5 years ago

pherrusa7 commented 5 years ago

changes in gradcam.py: fixed wrong denominator of Eq.19 (second term: Sum_{a,b} A^{k}_{a,b}) in the paper.

changes in example.ipynb: added a flag to automatically detect if Cuda can be used, otherwise, use CPU.