1. PyTorch 的 hook 及其在 Grad-CAM 中的应用
https://bbs.cvmart.net/topics/1532?from=groupmessage&isappinstalled=0
grad-cam原理:利用logit输出值反向求取对应特征层的梯度,并把每层特征的梯度取均值作为该层的权重,最后将各层加权求和归一化到0~1之间作为heatmap
1. PyTorch 的 hook 及其在 Grad-CAM 中的应用
https://bbs.cvmart.net/topics/1532?from=groupmessage&isappinstalled=0
grad-cam原理:利用logit输出值反向求取对应特征层的梯度,并把每层特征的梯度取均值作为该层的权重,最后将各层加权求和归一化到0~1之间作为heatmap