lucidrains/vit-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

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

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

Reference PyTorch implementation and models for DINOv3

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

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

Hackable and optimized Transformers building blocks, supporting a composable construction.

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

Datasets, Transforms and Models specific to Computer Vision