ml.postprocessing.ultralytics — Ultralytics

A ml.postprocessing.ultralytics modul az Ultralytics-hez tartalmaz osztályokat.

class YoloV5 – YOLO V5

A YOLO V5 modell kimenetének utófeldolgozásához használatos.

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

Létrehoz egy YOLO V5 utófeldolgozót.

threshold Pontszám-küszöbérték az észlelések NMS előtti szűréséhez.

nms_threshold Az NMS által az átfedő határoló dobozok elnyomásához használt IoU küszöbérték.

nms_sigma A Soft-NMS által a pontszámcsökkenéshez használt szigma érték.

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

Utófeldolgozza a model által az adott inputs listához előállított outputs tömblistát. Visszaad egy osztályonkénti listákból álló listát; minden belső lista ((x, y, w, h), score) tuple-öket tartalmaz ROI-koordinátákban. Az üres osztálylisták megmaradnak, így minden listaindex megfelel a modell kimenetében szereplő megfelelő osztályindexnek.

class YoloV8 – YOLO V8

A YOLO V8 modell kimenetének utófeldolgozásához használatos.

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

Létrehoz egy YOLO V8 utófeldolgozót.

threshold Pontszám-küszöbérték az észlelések NMS előtti szűréséhez.

nms_threshold Az NMS által az átfedő határoló dobozok elnyomásához használt IoU küszöbérték.

nms_sigma A Soft-NMS által a pontszámcsökkenéshez használt szigma érték.

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

Utófeldolgozza a model által az adott inputs listához előállított outputs tömblistát. Visszaad egy osztályonkénti listákból álló listát; minden belső lista ((x, y, w, h), score) tuple-öket tartalmaz ROI-koordinátákban. Az üres osztálylisták megmaradnak, így minden listaindex megfelel a modell kimenetében szereplő megfelelő osztályindexnek.