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
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
OpenMMLab Text Detection, Recognition and Understanding Toolbox
Updated
OpenMMLab Model Deployment Framework
Updated
Torchreid: Deep learning person re-identification in PyTorch.
Updated
Scaled-YOLOv4: Scaling Cross Stage Partial Network
Updated
Open source deep learning based unsupervised image retrieval toolbox built on PyTorch🔥
Updated
Flappy Bird hack using Deep Reinforcement Learning (Deep Q-learning).
Updated