#41 · 主分类: 计算机视觉
pytorch-grad-cam
面向计算机视觉的高级AI可解释性。支持CNN、视觉Transformer、分类、目标检测、分割、图像相似度等。
项目最后更新:08/13/26
GitHub Stars
13.0K
Forks数量
1.7K
贡献者数量
54
许可证
MIT
收录理由
在PyTorch视觉模型的训练或部署过程中,几乎总要面对一个问题:网络究竟在关注什么?这个库把从原始GradCAM到ScoreCAM、LayerCAM、HiResCAM等一大批像素归因方法整合到统一的接口里,改几行代码就能切换不同技术,不必重写整个流程。它不限于图像分类,目标检测、语义分割、嵌入相似度乃至CLIP风格的文本提示都能处理,因此既可用于调试实际业务中的模型,也适合研究阶段的实验验证。内置的平滑处理、完整批量推理支持,以及用于评估解释可信度的度量工具,让生成的注意力图更可靠、更易用。做应用计算机视觉的人,遇到模型行为异常时,这个库往往是第一个值得尝试的工具。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。