PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
Updated
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
Updated
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
Updated
The official code for “DocTr: Document Image Transformer for Geometric Unwarping and Illumination Correction”, ACM MM, Oral Paper, 2021.
Updated
Updated
Ultralytics YOLO11 🚀
Updated
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Updated
OpenMMLab Foundational Library for Training Deep Learning Models
Updated
OpenMMLab Image Classification Toolbox and Benchmark
Updated
OpenMMLab Pre-training Toolbox and Benchmark
Updated
Real-time multi-camera multi-object tracker using YOLOv5 and StrongSORT with OSNet
Updated
An all-in-one toolkit for computer vision
Updated
SOTA Re-identification Methods and Toolbox
Updated
OpenMMLab Detection Toolbox and Benchmark
Updated
MOT using deepsort and yolov3 with pytorch
Updated
OpenMMLab Text Detection, Recognition and Understanding Toolbox
Updated
OpenMMLab Semantic Segmentation Toolbox and Benchmark.
Updated
The state-of-the-art image restoration model without nonlinear activation functions.
Updated
PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations
Updated
Official PyTorch implementation of "Multi-modal Queried Object Detection in the Wild" (accepted by NeurIPS 2023)
Updated