ml.postprocessing.ultralytics — Ultralytics

Modul ml.postprocessing.ultralytics obsahuje třídy pro Ultralytics.

class YoloV5 – YOLO V5

Slouží k následnému zpracování výstupu modelu YOLO V5.

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

Vytvoří následný procesor YOLO V5.

threshold Práh skóre pro filtrování detekcí před NMS.

nms_threshold Práh IoU používaný NMS k potlačení překrývajících se ohraničujících rámečků.

nms_sigma Hodnota sigma používaná Soft-NMS pro pokles skóre.

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

Provede následné zpracování seznamu polí outputs vytvořeného modelem model pro daný seznam inputs. Vrací seznam seznamů po jednotlivých třídách; každý vnitřní seznam obsahuje n-tice ((x, y, w, h), score) v souřadnicích ROI. Prázdné seznamy tříd jsou zachovány, takže index každého seznamu odpovídá příslušnému indexu třídy ve výstupu modelu.

class YoloV8 – YOLO V8

Slouží k následnému zpracování výstupu modelu YOLO V8.

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

Vytvoří následný procesor YOLO V8.

threshold Práh skóre pro filtrování detekcí před NMS.

nms_threshold Práh IoU používaný NMS k potlačení překrývajících se ohraničujících rámečků.

nms_sigma Hodnota sigma používaná Soft-NMS pro pokles skóre.

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

Provede následné zpracování seznamu polí outputs vytvořeného modelem model pro daný seznam inputs. Vrací seznam seznamů po jednotlivých třídách; každý vnitřní seznam obsahuje n-tice ((x, y, w, h), score) v souřadnicích ROI. Prázdné seznamy tříd jsou zachovány, takže index každého seznamu odpovídá příslušnému indexu třídy ve výstupu modelu.