pytorch-grad-cam

pytorch-grad-cam 是一个用于计算机视觉的高级可解释性AI工具包，支持多种深度学习模型和任务。它提供了包括GradCAM、HiResCAM在内的多种像素归因方法，适用于CNNs、Vision Transformers等常见网络结构，并能在分类、目标检测、语义分割及图像相似度等多种场景下工作。项目通过平滑处理使可视化结果更加清晰，同时支持批量图像处理以提高性能，并包含一系列评估指标来优化解释效果。此工具非常适合需要理解和调试模型预测的研究者与开发者使用。