Grad-CAM caffe具体实现中的bug

参考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如下

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 发现 关注 消息 iOS 第三方库、插件、知名博客总结 作者大灰狼的小绵羊哥哥关注 2017.06.26 09:4...
    肇东周阅读 14,222评论 4 61
  • 尚俊平,焦点网络中级,坚持分享552天,2017年10月2日周一 今天看到一个美国人的演讲:如何提高自信?讲的内容...
    32598db751bb阅读 988评论 0 0
  • 感觉手感不错 界面也很清晰 非常友好 排版也容易 爪机写东西也很方便 还有分割线! 话说这东西怎么居中? 爪机没法...
    阿尔法驴阅读 1,631评论 0 1
  • 今天是智慧树寒假七天乐第二期结业的日子。今天的主要活动是拼豆豆和结业式。拼豆豆主要培养孩子的眼手协调能力,对色彩的...
    别闹了魂淡阅读 2,187评论 0 0
  • 对于刚一开始就拖延了两天的写作,我是对自己无话可说了。想问问自己,说好的坚持哪里去了? 今天我想说说自己的工作。鉴...
    嘉55阅读 1,235评论 0 0