
pytorch-grad-cam
jacobgil
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
AI 简介
这是一个面向PyTorch的先进计算机视觉可解释性(XAI)工具包,提供多种类激活图(CAM)算法实现,用于可视化深度学习模型对图像预测的关键依据区域。核心支持Grad-CAM、Score-CAM、HiResCAM等十余种像素级归因方法,兼容CNN与Vision Transformer架构,并覆盖图像分类、目标检测、语义分割及图像相似度等任务;具备批量处理、平滑优化、可信度评估与量化指标等功能。适用于模型调试、结果诊断、科研对比及AI系统可信性验证等场景。
Python
MIT License12.9k
Stars
1.7k
Forks
45
Watchers
154
Issues
Star 增长
今日0
近 7 天0
近 30 天+22
综合评分71.9
默认分支master