对视觉类论文详解(免费)感兴趣的同学,可以关注微信公众号 李卓璐随手记,将会不定期发布,注意查收哦~
续可视化笔记2-pytorch 可视化卷积网络中间特征层的基础上使用CAM方法可视化网络对待测对象关注的位置。
1.定义GradCAM类
注意:代码中需要改的3个地方已经用注释标清,大家使用时注意修改。
class GradCAM(nn.Module):
def __init__(self):
super(GradCAM, self).__init__(
续可视化笔记2-pytorch 可视化卷积网络中间特征层的基础上使用CAM方法可视化网络对待测对象关注的位置。
注意:代码中需要改的3个地方已经用注释标清,大家使用时注意修改。
class GradCAM(nn.Module):
def __init__(self):
super(GradCAM, self).__init__(