huggingface/pytorch-image-models

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

Datasets, Transforms and Models specific to Computer Vision

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

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

Reference PyTorch implementation and models for DINOv3

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

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

A comprehensive list of pytorch related content on github,such as different models,implementations,helper libraries,tutorials etc.

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