A state-of-the-art PyTorch-based computer vision model for object detection, segmentation, and classification.
Ultralytics YOLOv5 is a state-of-the-art computer vision model built on PyTorch, designed for object detection, instance segmentation, and image classification. It solves the problem of performing real-time, accurate visual analysis by providing a suite of pre-trained models that are easy to train, validate, and deploy across various hardware and software platforms.
Computer vision researchers, developers, and engineers working on real-time object detection, segmentation, or classification tasks, particularly those who value ease of use, speed, and a wide range of deployment options.
Developers choose YOLOv5 for its balance of speed and accuracy, its extensive model zoo ranging from nano to extra-large sizes, and its straightforward workflow for training on custom data and exporting to multiple formats like ONNX, TensorRT, and TFLite for production deployment.
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
Offers a range from YOLOv5n to YOLOv5x, allowing developers to balance speed and accuracy based on their needs, with pre-trained models on COCO for quick deployment.
Supports export to multiple formats like ONNX, TensorRT, and TFLite, enabling easy deployment across cloud, edge, and mobile platforms as detailed in the export tutorials.
Includes tools for hyperparameter evolution, multi-GPU training, and integration with MLOps platforms like Weights & Biases, enhancing workflow efficiency.
Simple inference via PyTorch Hub or detect.py script, handling diverse sources from images to video streams with minimal code, as shown in the quickstart examples.
The README actively promotes YOLO11, indicating YOLOv5 is no longer the primary focus and may lack future updates, breaking changes, and new features.
Training from scratch requires significant GPU time, e.g., up to 8 days for YOLOv5x on a single V100, which can be prohibitive for teams with limited computational budgets.
Lacks support for tasks like pose estimation and oriented object detection, which are highlighted in YOLO11, limiting its utility for cutting-edge applications.