ml.postprocessing.ultralytics — Ultralytics

Модуль ml.postprocessing.ultralytics містить класи для Ultralytics.

class YoloV5 – YOLO V5

Використовується для постобробки вихідних даних моделі YOLO V5.

class ml.postprocessing.ultralytics.YoloV5(threshold: float = 0.6, nms_threshold: float = 0.1, nms_sigma: float = 0.1)

Створює постпроцесор YOLO V5.

threshold Поріг оцінки для фільтрації виявлень перед NMS.

nms_threshold Поріг IoU, що використовується NMS для придушення перекриваючих обмежувальних прямокутників.

nms_sigma Значення сигма, що використовується Soft-NMS для зменшення оцінки.

__call__(model: ml.Model, inputs: list, outputs: list) → list

Постобробляє список масивів outputs, виготовлених model для даного списку inputs. Повертає список посписних списків; кожен внутрішній список містить кортежі ((x, y, w, h), score) у координатах ROI. Порожні списки класів зберігаються, щоб індекс кожного списку відповідав відповідному індексу класу у виході моделі.

class YoloV8 – YOLO V8

Використовується для постобробки вихідних даних моделі YOLO V8.

class ml.postprocessing.ultralytics.YoloV8(threshold: float = 0.6, nms_threshold: float = 0.1, nms_sigma: float = 0.1)

Створює постпроцесор YOLO V8.

threshold Поріг оцінки для фільтрації виявлень перед NMS.

nms_threshold Поріг IoU, що використовується NMS для придушення перекриваючих обмежувальних прямокутників.

nms_sigma Значення сигма, що використовується Soft-NMS для зменшення оцінки.

__call__(model: ml.Model, inputs: list, outputs: list) → list

Постобробляє список масивів outputs, виготовлених model для даного списку inputs. Повертає список посписних списків; кожен внутрішній список містить кортежі ((x, y, w, h), score) у координатах ROI. Порожні списки класів зберігаються, щоб індекс кожного списку відповідав відповідному індексу класу у виході моделі.