Обнаружение объектов

YOLO object detection with bounding boxes

Обнаружение объектов — это задача, включающая идентификацию местоположения и класса объектов на изображении или в потоке видео.

Результатом работы детектора объектов является набор ограничивающих рамок (bounding boxes), которые охватывают объекты на изображении, а также метки классов и показатели уверенности для каждой рамки. Обнаружение объектов — хороший выбор, если тебе нужно идентифицировать интересующие объекты в сцене, но не требуется знать точно, где находится объект или его точная форма.



Watch: Object Detection with Pretrained Ultralytics YOLO Model.
Совет

Модели YOLO26 Detect являются стандартными моделями YOLO26, например yolo26n.pt, и они предварительно обучены на COCO.

Модели

Здесь представлены предварительно обученные модели YOLO26 Detect. Модели Detect, Segment и Pose предварительно обучены на наборе данных COCO, в то время как модели Classify предварительно обучены на наборе данных ImageNet.

Модели при первом использовании автоматически загружаются из последнего релиза Ultralytics.

Модельразмер
(пиксели)
mAPval
50-95
mAPval
50-95(e2e)
Скорость
CPU ONNX
(мс)
Скорость
T4 TensorRT10
(мс)
параметры
(М)
FLOPs
(Б)
YOLO26n64040.940.138.9 ± 0.71.7 ± 0.02.45.4
YOLO26s64048.647.887.2 ± 0.92.5 ± 0.09.520.7
YOLO26m64053.152.5220.0 ± 1.44.7 ± 0.120.468.2
YOLO26l64055.054.4286.2 ± 2.06.2 ± 0.224.886.4
YOLO26x64057.556.9525.8 ± 4.011.8 ± 0.255.7193.9
  • Значения mAPval приведены для одиночной модели с одним масштабом на наборе данных COCO val2017.
    Воспроизведи их с помощью yolo val detect data=coco.yaml device=0
  • Скорость усреднена по изображениям COCO val с использованием инстанса Amazon EC2 P4d.
    Воспроизведи с помощью yolo val detect data=coco.yaml batch=1 device=0|cpu
  • Значения Params и FLOPs указаны для объединенной (fused) модели после model.fuse(), которая объединяет слои Conv и BatchNorm и, для моделей end2end, удаляет вспомогательную «один-ко-многим» голову детекции. Предварительно обученные чекпоинты сохраняют полную архитектуру обучения и могут показывать более высокие значения.

Обучение (Train)

Обучи YOLO26n на наборе данных COCO8 в течение 100 эпох при размере изображения 640. Полный список доступных аргументов смотри на странице Конфигурация.

Пример
from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n.yaml")  # build a new model from YAML
model = YOLO("yolo26n.pt")  # load a pretrained model (recommended for training)
model = YOLO("yolo26n.yaml").load("yolo26n.pt")  # build from YAML and transfer weights

# Train the model
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)

Полные детали режима train смотри на странице Обучение. Модели детекции также можно обучать на облачных GPU через Ultralytics Platform.

Формат набора данных

Подробности о формате набора данных для детекции YOLO можно найти в Руководстве по наборам данных. Чтобы конвертировать свой существующий набор данных из других форматов (например, COCO и т.д.) в формат YOLO, используй инструмент JSON2YOLO от Ultralytics. Ты также можешь аннотировать и управлять наборами данных для детекции прямо на Ultralytics Platform с помощью инструментов разметки с поддержкой ИИ.

Валидация (Val)

Проверь точность обученной модели YOLO26n на наборе данных COCO8. Аргументы не требуются, так как model сохраняет свои data обучения и аргументы в качестве атрибутов модели.

Пример
from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n.pt")  # load an official model
model = YOLO("path/to/best.pt")  # load a custom model

# Validate the model
metrics = model.val()  # no arguments needed, dataset and settings remembered
metrics.box.map  # map50-95
metrics.box.map50  # map50
metrics.box.map75  # map75
metrics.box.maps  # a list containing mAP50-95 for each category
metrics.box.image_metrics  # per-image metrics dictionary with precision, recall, F1, TP, FP, and FN

Прогнозирование

Используй обученную модель YOLO26n для выполнения предсказаний на изображениях.

Пример
from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n.pt")  # load an official model
model = YOLO("path/to/best.pt")  # load a custom model

# Predict with the model
results = model("https://ultralytics.com/images/bus.jpg")  # predict on an image

# Access the results
for result in results:
    xywh = result.boxes.xywh  # center-x, center-y, width, height
    xywhn = result.boxes.xywhn  # normalized
    xyxy = result.boxes.xyxy  # top-left-x, top-left-y, bottom-right-x, bottom-right-y
    xyxyn = result.boxes.xyxyn  # normalized
    names = [result.names[cls.item()] for cls in result.boxes.cls.int()]  # class name of each box
    confs = result.boxes.conf  # confidence score of each box

Полные детали режима predict смотри на странице Предсказание.

Экспорт (Export)

Экспортируй модель YOLO26n в другой формат, например ONNX, CoreML и т.д.

Пример
from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n.pt")  # load an official model
model = YOLO("path/to/best.pt")  # load a custom-trained model

# Export the model
model.export(format="onnx")

Доступные форматы экспорта YOLO26 приведены в таблице ниже. Ты можешь экспортировать в любой формат, используя аргумент format, например format='onnx' или format='engine'. Ты можешь выполнять предсказание или валидацию прямо на экспортированных моделях, например yolo predict model=yolo26n.onnx. Примеры использования показаны для твоей модели после завершения экспорта.

ФорматАргумент formatМодельМетаданныеАргументы
PyTorch-yolo26n.pt-
TorchScripttorchscriptyolo26n.torchscriptimgsz, half, dynamic, optimize, nms, batch, device
ONNXonnxyolo26n.onnximgsz, half, dynamic, simplify, opset, nms, batch, device
OpenVINOopenvinoyolo26n_openvino_model/imgsz, half, dynamic, int8, nms, batch, data, fraction, device
TensorRTengineyolo26n.engineimgsz, half, dynamic, simplify, workspace, int8, nms, batch, data, fraction, device
CoreMLcoremlyolo26n.mlpackageimgsz, dynamic, half, int8, nms, batch, device
TF SavedModelsaved_modelyolo26n_saved_model/imgsz, keras, int8, nms, batch, data, fraction, device
TF GraphDefpbyolo26n.pbimgsz, batch, device
TF Litetfliteyolo26n.tfliteimgsz, half, int8, nms, batch, data, fraction, device
TF Edge TPUedgetpuyolo26n_edgetpu.tfliteimgsz, int8, data, fraction, device
TF.jstfjsyolo26n_web_model/imgsz, half, int8, nms, batch, data, fraction, device
PaddlePaddlepaddleyolo26n_paddle_model/imgsz, batch, device
MNNmnnyolo26n.mnnimgsz, batch, int8, half, device
NCNNncnnyolo26n_ncnn_model/imgsz, half, batch, device
IMX500imxyolo26n_imx_model/imgsz, int8, data, fraction, nms, device
RKNNrknnyolo26n_rknn_model/imgsz, batch, name, device
ExecuTorchexecutorchyolo26n_executorch_model/imgsz, batch, device
Axeleraaxelerayolo26n_axelera_model/imgsz, batch, int8, data, fraction, device
DeepXdeepxyolo26n_deepx_model/imgsz, int8, data, optimize, device

Смотри полные сведения об export на странице Экспорт.

Часто задаваемые вопросы (FAQ)

Могу ли я обучать и развертывать модели детекции без программирования?

Да. Ultralytics Platform предоставляет браузерный рабочий процесс для аннотирования наборов данных, обучения моделей детекции на облачных GPU и их развертывания на эндпоинтах вывода. Чтобы начать, смотри Краткое руководство по платформе.

Как мне обучить модель YOLO26 на моем собственном наборе данных?

Обучение модели YOLO26 на собственном наборе данных включает в себя несколько шагов:

  1. Подготовь набор данных: Убедись, что твой набор данных в формате YOLO. За рекомендациями обращайся к нашему Руководству по наборам данных.
  2. Загрузи модель: Используй библиотеку Ultralytics YOLO для загрузки предварительно обученной модели или создания новой модели из файла YAML.
  3. Обучи модель: Выполни метод train в Python или команду yolo detect train в CLI.
Пример
from ultralytics import YOLO

# Load a pretrained model
model = YOLO("yolo26n.pt")

# Train the model on your custom dataset
model.train(data="my_custom_dataset.yaml", epochs=100, imgsz=640)

Для получения подробных опций конфигурации посети страницу Конфигурация.

Какие предварительно обученные модели доступны в YOLO26?

Ultralytics YOLO26 предлагает различные предварительно обученные модели для обнаружения объектов, сегментации и оценки позы. Эти модели предварительно обучены на наборе данных COCO или ImageNet для задач классификации. Вот некоторые из доступных моделей:

Для получения подробного списка и метрик производительности перейди в раздел Модели.

Как я могу проверить точность моей обученной модели YOLO?

Чтобы проверить точность твоей обученной модели YOLO26, ты можешь использовать метод .val() в Python или команду yolo detect val в CLI. Это предоставит метрики, такие как mAP50-95, mAP50 и другие.

Пример
from ultralytics import YOLO

# Load the model
model = YOLO("path/to/best.pt")

# Validate the model
metrics = model.val()
print(metrics.box.map)  # mAP50-95

Для получения дополнительных сведений о валидации посети страницу Валидация.

В какие форматы я могу экспортировать модель YOLO26?

Ultralytics YOLO26 позволяет экспортировать модели в различные форматы, такие как ONNX, TensorRT, CoreML и другие, чтобы обеспечить совместимость с различными платформами и устройствами.

Пример
from ultralytics import YOLO

# Load the model
model = YOLO("yolo26n.pt")

# Export the model to ONNX format
model.export(format="onnx")

Проверь полный список поддерживаемых форматов и инструкции на странице Экспорт.

Почему мне стоит использовать Ultralytics YOLO26 для обнаружения объектов?

Ultralytics YOLO26 разработан для обеспечения передовой производительности в обнаружении объектов, сегментации и оценке позы. Вот некоторые ключевые преимущества:

  1. Предварительно обученные модели: Используй модели, предварительно обученные на популярных наборах данных, таких как COCO и ImageNet, для более быстрой разработки.
  2. Высокая точность: Достигает впечатляющих показателей mAP, обеспечивая надежное обнаружение объектов.
  3. Скорость: Оптимизирована для вывода в реальном времени, что делает ее идеальной для приложений, требующих быстрой обработки.
  4. Гибкость: Экспортируй модели в различные форматы, такие как ONNX и TensorRT, для развертывания на нескольких платформах.

Изучи наш Блог, чтобы ознакомиться с вариантами использования и историями успеха, демонстрирующими работу YOLO26.

Комментарии