最近感觉类激活图可视化是一件很有趣的事情。
CAM(传送门:CAM实现的流程(pytorch))由于对网络结构有定性要求,所以在可视化一些有多个全连接层的网络时,表现不太友好,于是出现了Grad-CAM。
文章目录
算法思路
Grad-CAM根据输出向量,进行backward,求取特征图的梯度,得到每个特征图上每个像素点对应的梯度,也就是特征图对应的梯度图,然后再对每个梯度图求平均,这个平均值就对应于每个特征图的权重,然后再将权重与特征图进行加权求和,最后经过relu激活函数就可以得到最终的类激活图
实现过程
先准备图片、标签以及模型
类别标签下载方法: