pytorch/vision

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

We write your reusable computer vision tools. 💜

Reference PyTorch implementation and models for DINOv3

🐍 Geometric Computer Vision Library for Spatial AI

Refine high-quality datasets and visual AI models

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

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

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

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