frgfm/torch-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

Model interpretability and understanding for PyTorch

Collection of common code that's shared among different research projects in FAIR computer vision team.

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

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

A modern model graph visualizer and debugger

Reference PyTorch implementation and models for DINOv3

Datasets, Transforms and Models specific to Computer Vision

Karras et al. (2022) diffusion models for PyTorch