`ml.postprocessing.ultralytics` — Ultralytics¶

Modulen ml.postprocessing.ultralytics innehåller klasser för Ultralytics.

class YoloV5 – YOLO V5¶

Används för att efterbehandla utdata från YOLO V5-modeller.

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

Skapa en YOLO V5-efterbehandlare.

threshold Poängtröskelvärde för att filtrera detekteringar före NMS.

nms_threshold IoU-tröskelvärde som används av NMS för att undertrycka överlappande begränsningsrutor.

nms_sigma Sigma-värde som används av Soft-NMS för poängavtagning.

__call__(model: ml.Model, inputs: list, outputs: list) → list¶: Efterbehandla listan outputs med arrayer som produceras av model för den angivna listan inputs. Returnerar en lista med listor per klass; varje inre lista innehåller ((x, y, w, h), score)-tupler i ROI-koordinater. Tomma klasslistor bevaras så att varje listindex matchar motsvarande klassindex i modellens utdata.

Används för att efterbehandla utdata från YOLO V8-modeller.

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

Skapa en YOLO V8-efterbehandlare.

threshold Poängtröskelvärde för att filtrera detekteringar före NMS.

nms_threshold IoU-tröskelvärde som används av NMS för att undertrycka överlappande begränsningsrutor.

nms_sigma Sigma-värde som används av Soft-NMS för poängavtagning.

__call__(model: ml.Model, inputs: list, outputs: list) → list¶: Efterbehandla listan outputs med arrayer som produceras av model för den angivna listan inputs. Returnerar en lista med listor per klass; varje inre lista innehåller ((x, y, w, h), score)-tupler i ROI-koordinater. Tomma klasslistor bevaras så att varje listindex matchar motsvarande klassindex i modellens utdata.