在执行Grad-CAM所遇问题
1)
修改后解决
2)
修改后解决,因为numpy需要在cpu上进行,所有需要加上.cpu()
3)plt.matshow(heatmap)出错
原因是get_heatmap()中的mean_gradients = torch.mean(gradients, dim=[0, 2, 3])出错,尺寸不对。
参考上述公式进行修改,成功!
详细的执行步骤见
Grad-CAM对任意CNN网络进行可视化处理教学,普适+好用!!!
在执行Grad-CAM所遇问题
1)
修改后解决
2)
修改后解决,因为numpy需要在cpu上进行,所有需要加上.cpu()
3)plt.matshow(heatmap)出错
原因是get_heatmap()中的mean_gradients = torch.mean(gradients, dim=[0, 2, 3])出错,尺寸不对。
参考上述公式进行修改,成功!
详细的执行步骤见
Grad-CAM对任意CNN网络进行可视化处理教学,普适+好用!!!
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:/a/582518.html
如若内容造成侵权/违法违规/事实不符,请联系我们进行投诉反馈qq邮箱809451989@qq.com,一经查实,立即删除!