ml.postprocessing.ultralytics — Ultralytics

Moduł ml.postprocessing.ultralytics zawiera klasy dla Ultralytics.

klasa YoloV5 – YOLO V5

Używany do przetwarzania końcowego wyjścia modelu YOLO V5.

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

Tworzy post-procesor YOLO V5.

threshold Próg wyniku do filtrowania wykryć przed NMS.

nms_threshold Próg IoU używany przez NMS do tłumienia nakładających się ramek ograniczających.

nms_sigma Wartość sigma używana przez Soft-NMS do zaniku wyniku.

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

Przetwarza końcowo listę tablic outputs wyprodukowaną przez model dla podanej listy inputs. Zwraca listę list dla poszczególnych klas; każda wewnętrzna lista zawiera krotki ((x, y, w, h), score) we współrzędnych ROI. Puste listy klas są zachowywane, aby indeks każdej listy odpowiadał odpowiedniemu indeksowi klasy w wyjściu modelu.

klasa YoloV8 – YOLO V8

Używany do przetwarzania końcowego wyjścia modelu YOLO V8.

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

Tworzy post-procesor YOLO V8.

threshold Próg wyniku do filtrowania wykryć przed NMS.

nms_threshold Próg IoU używany przez NMS do tłumienia nakładających się ramek ograniczających.

nms_sigma Wartość sigma używana przez Soft-NMS do zaniku wyniku.

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

Przetwarza końcowo listę tablic outputs wyprodukowaną przez model dla podanej listy inputs. Zwraca listę list dla poszczególnych klas; każda wewnętrzna lista zawiera krotki ((x, y, w, h), score) we współrzędnych ROI. Puste listy klas są zachowywane, aby indeks każdej listy odpowiadał odpowiedniemu indeksowi klasy w wyjściu modelu.