← all repositories

jacobgil/pytorch-grad-cam

A PyTorch library for visualizing and evaluating explainability in CNN and Vision Transformer models.

12.9k stars Python LLMOps · Eval
pytorch-grad-cam
Velocity · 7d
+3.9
★ / day
Trend
steady
star history

This repository provides state-of-the-art explainability methods for computer vision models in PyTorch. It implements various pixel attribution techniques like Grad-CAM and Score-CAM for diagnosing model predictions. The library supports CNNs and Vision Transformers across tasks including classification, object detection, semantic segmentation, and embedding similarity. It also includes smoothing methods and trust metrics for evaluating explanation quality.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.