jacobgil/pytorch-grad-cam

A game theoretic approach to explain the output of any machine learning model.

Reference PyTorch implementation and models for DINOv3

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Lime: Explaining the predictions of any machine learning classifier

Datasets, Transforms and Models specific to Computer Vision

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

PyTorch code and models for the DINOv2 self-supervised learning method.

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377