Модули плагина acmenra-yolo¶
Плагин содержит один основной модуль, реализующий контракт acmenra_cv.inference.Backend.
YOLOBackend¶
Конкретная реализация интерфейса Backend для моделей Ultralytics YOLO.
Свойства (со строгой валидацией)¶
model(YOLO): Экземпляр модели Ultralytics. Смена модели сбрасывает внутреннее состояние трекинга.iou(float): Порог IoU для Non-Maximum Suppression (NMS) в диапазоне[0.0, 1.0].imgsz(int): Размер входного изображения в пикселях (должен быть > 0, рекомендуется кратный 32).half(bool): Флаг использования FP16-инференса (работает только на CUDA).refined(bool): Флаг генерации высококачественных масок (соответствует параметруretina_masksв Ultralytics). Значительно улучшает гладкость границ сегментации ценой снижения FPS на 20-30%.
Методы¶
predict(frame, max_det=640, track=False, verbose=False)→Result: Выполняет инференс. Автоматически маршрутизирует вывод в_process_bbox_results,_process_seg_resultsили_process_obb_resultsв зависимости от типа задачи.track(frame, ...)→Result: Делегирует выполнение методуpredictс включенным нативным трекингом YOLO (persist=True).
Внутренняя обработка¶
_process_bbox_results: Конвертирует YOLOboxesв нормализованныеacmenra_cv.instance.Box._process_seg_results: Конвертирует YOLOmasks.xynвacmenra_cv.instance.Polygonи вычисляет ограничивающийBox._process_obb_results: Конвертирует YOLOobb.xywhrвacmenra_cv.instance.Obbс помощью нормализации углов и координат.