zhanghang1989/ResNeSt

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

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

[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model

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

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

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

Official PyTorch implementation of SegFormer

Reference PyTorch implementation and models for DINOv3

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