ml.postprocessing.ultralytics — Ultralytics

De ml.postprocessing.ultralytics-module bevat klassen voor Ultralytics.

class YoloV5 – YOLO V5

Wordt gebruikt om de uitvoer van YOLO V5-modellen na te bewerken.

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

Maakt een YOLO V5-nabewerker aan.

threshold Score-drempelwaarde voor het filteren van detecties vóór NMS.

nms_threshold IoU-drempelwaarde die door NMS wordt gebruikt om overlappende begrenzingsvakken te onderdrukken.

nms_sigma Sigma-waarde die door Soft-NMS wordt gebruikt voor scoreverval.

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

Bewerkt de outputs-lijst met arrays na die door model is geproduceerd voor de opgegeven inputs-lijst. Retourneert een lijst met per-klasse lijsten; elke binnenlijst bevat ((x, y, w, h), score)-tuples in ROI-coördinaten. Lege klasselijsten worden behouden zodat elke lijstindex overeenkomt met de bijbehorende klasse-index in de model-uitvoer.

class YoloV8 – YOLO V8

Wordt gebruikt om de uitvoer van YOLO V8-modellen na te bewerken.

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

Maakt een YOLO V8-nabewerker aan.

threshold Score-drempelwaarde voor het filteren van detecties vóór NMS.

nms_threshold IoU-drempelwaarde die door NMS wordt gebruikt om overlappende begrenzingsvakken te onderdrukken.

nms_sigma Sigma-waarde die door Soft-NMS wordt gebruikt voor scoreverval.

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

Bewerkt de outputs-lijst met arrays na die door model is geproduceerd voor de opgegeven inputs-lijst. Retourneert een lijst met per-klasse lijsten; elke binnenlijst bevat ((x, y, w, h), score)-tuples in ROI-coördinaten. Lege klasselijsten worden behouden zodat elke lijstindex overeenkomt met de bijbehorende klasse-index in de model-uitvoer.