#340 · 主分类: 计算机视觉

grad-cam

convolutional-neural-networks deep-learning grad-cam heatmap iccv17 interpretability visual-explanation

[ICCV 2017] 用于 Grad-CAM 的 Torch 代码

项目最后更新:09/17/22

GitHub Stars

1.7K

Forks数量

237

贡献者数量

5

许可证

Other

收录理由

这是 Grad-CAM 方法的官方参考实现,能把卷积神经网络的判断转化成热力图,直观显示图像中哪些区域影响了模型的预测结果。代码用 Torch/Lua 编写,依赖 Caffe 模型,放在今天的 PyTorch 或 TensorFlow 环境里可能不太顺手,但作为理解和学习这个经典方法的历史资料,价值依然明显。仓库覆盖了图像分类、视觉问答和图像描述等任务,通过简单的命令行参数就能指定网络层、输出热力图或原始梯度,操作起来很直接。如果你在研究模型可解释性,或者想复现 ICCV 2017 那篇论文的实验,这份作者亲写的代码提供了清晰可读的参考,能帮你一步步理清 Grad-CAM 的内部机制,再动手迁移到其他框架。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目