PyTorch Grad-CAM
先进的计算机视觉可解释性工具库,支持 CNN、Vision Transformer 等模型的 Grad-CAM 可视化,覆盖分类、目标检测、图像分割和图像相似度等多种任务。
🎯适用场景:可视化 CNN 和 Transformer 模型的注意力区域,解释图像分类和检测结果,适合模型调试、学术研究和医疗影像分析
#类激活图#计算机视觉#可解释 AI#可解释 ML#Grad-CAM
📥 收录于 2026/5/25
📊 仓库数据
Stars12,958
Forks1,705
语言Python
更新2026/8/23
📈 Stars 变化 ↑3 天 +5· 统计区间 8/21 13:15 → 8/24 03:32(3 天)
✅ 优点
- •支持多种 CV 任务和模型架构,覆盖范围广
- •API 简单易用,几行代码即可生成可视化热力图
⚠️ 限制
- •Grad-CAM 的解释粒度较粗,无法精确到像素级别
- •对非 CNN 架构(如纯 MLP)的解释效果有限
