pytorch-grad-cam
A PyTorch library providing state-of-the-art methods for generating visual explanations (Class Activation Maps) for computer vision models.
What is pytorch-grad-cam?
PyTorch Grad-CAM is a Python library that implements advanced Explainable AI (XAI) techniques for computer vision models. It generates visual explanations, known as Class Activation Maps (CAMs), to highlight which regions of an image influenced a model's prediction. This helps developers and researchers understand, debug, and trust their deep learning models by making their decision-making processes more transparent.
Machine learning researchers, computer vision engineers, and data scientists who are developing, deploying, or researching convolutional neural networks, Vision Transformers, or other vision models and need to interpret model predictions.
It offers the most comprehensive collection of modern CAM algorithms in a single, well-tested PyTorch library, with unique support for advanced tasks like object detection, segmentation, and embeddings, along with quantitative evaluation metrics to validate explanation quality.
Overview
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
Use Cases
Best For
- Generating visual explanations for CNN and Vision Transformer predictions
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.