ml.postprocessing.ultralytics — Ultralytics

Das Modul ml.postprocessing.ultralytics enthält Klassen für Ultralytics.

class YoloV5 – YOLO V5

Wird zur Nachbearbeitung der Ausgabe von YOLO-V5-Modellen verwendet.

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

Erstellt einen YOLO-V5-Nachbearbeiter.

threshold Schwellenwert für den Score zum Filtern von Erkennungen vor der NMS.

nms_threshold IoU-Schwellenwert, der von der NMS verwendet wird, um sich überlappende Begrenzungsrahmen zu unterdrücken.

nms_sigma Sigma-Wert, der von Soft-NMS für den Score-Abfall verwendet wird.

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

Bearbeitet die von model für die gegebene inputs-Liste erzeugte outputs-Liste von Arrays nach. Gibt eine Liste von Listen pro Klasse zurück; jede innere Liste enthält ((x, y, w, h), score)-Tupel in ROI-Koordinaten. Leere Klassenlisten werden beibehalten, sodass jeder Listenindex dem entsprechenden Klassenindex in der Modellausgabe entspricht.

class YoloV8 – YOLO V8

Wird zur Nachbearbeitung der Ausgabe von YOLO-V8-Modellen verwendet.

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

Erstellt einen YOLO-V8-Nachbearbeiter.

threshold Schwellenwert für den Score zum Filtern von Erkennungen vor der NMS.

nms_threshold IoU-Schwellenwert, der von der NMS verwendet wird, um sich überlappende Begrenzungsrahmen zu unterdrücken.

nms_sigma Sigma-Wert, der von Soft-NMS für den Score-Abfall verwendet wird.

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

Bearbeitet die von model für die gegebene inputs-Liste erzeugte outputs-Liste von Arrays nach. Gibt eine Liste von Listen pro Klasse zurück; jede innere Liste enthält ((x, y, w, h), score)-Tupel in ROI-Koordinaten. Leere Klassenlisten werden beibehalten, sodass jeder Listenindex dem entsprechenden Klassenindex in der Modellausgabe entspricht.