You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

2.0 KiB

comments description
true Learn about the supported models and architectures, such as YOLOv3, YOLOv5, and YOLOv8, and how to contribute your own model to Ultralytics.

Models

Ultralytics supports many models and architectures with more to come in the future. Want to add your model architecture? Here's how you can contribute.

In this documentation, we provide information on four major models:

  1. YOLOv3: The third iteration of the YOLO model family, known for its efficient real-time object detection capabilities.
  2. YOLOv5: An improved version of the YOLO architecture, offering better performance and speed tradeoffs compared to previous versions.
  3. YOLOv6: Released by Meituan in 2022 and is in use in many of the company's autonomous delivery robots.
  4. YOLOv8: The latest version of the YOLO family, featuring enhanced capabilities such as instance segmentation, pose/keypoints estimation, and classification.
  5. Segment Anything Model (SAM): Meta's Segment Anything Model (SAM).
  6. YOLO-NAS: YOLO Neural Architecture Search (NAS) Models.
  7. Realtime Detection Transformers (RT-DETR): Baidu's PaddlePaddle Realtime Detection Transformer (RT-DETR) models.

You can use these models directly in the Command Line Interface (CLI) or in a Python environment. Below are examples of how to use the models with CLI and Python:

CLI Example

yolo task=detect mode=train model=yolov8n.yaml data=coco128.yaml epochs=100

Python Example

from ultralytics import YOLO

model = YOLO("model.yaml")  # build a YOLOv8n model from scratch
# YOLO("model.pt")  use pre-trained model if available
model.info()  # display model information
model.train(data="coco128.yaml", epochs=100)  # train the model

For more details on each model, their supported tasks, modes, and performance, please visit their respective documentation pages linked above.