美文网首页
Grad-CAM caffe具体实现中的bug

Grad-CAM caffe具体实现中的bug

作者: 涂山容容 | 来源:发表于2018-05-08 13:31 被阅读0次

参考http://sandaw89.blogspot.com/2017/08/gradcam-implementation-in-pycaffe.html的实现,对自己的model实现可视化。代码调通了之后发现,输出的heatmap是这样的


相当于全都是一个值,肯定哪里有错了。输出梯度的值看了一下,全都是0,说明梯度根本没有回传,说明问题出现在了反向传播上。
查了一下,https://github.com/BVLC/caffe/issues/583
是因为caffe 默认没有损失就不算梯度。
所以要在deploy中添加force_backward: true

现在打印梯度的值,就不是全为零了。
输出的heatmap如下

相关文章

网友评论

      本文标题:Grad-CAM caffe具体实现中的bug

      本文链接:https://www.haomeiwen.com/subject/giyirftx.html