`ml.postprocessing.darknet` — Darknet¶

De ml.postprocessing.darknet-module bevat nabewerkers voor op Darknet gebaseerde objectdetectiemodellen.

class YoloV2 – YOLO V2¶

Nabewerker voor de uitvoer van YOLO V2-modellen.

class ml.postprocessing.darknet.YoloV2(threshold: float = 0.6, anchors: numpy.ndarray = None, nms_threshold: float = 0.1, nms_sigma: float = 0.1)¶

Maakt een YOLO V2-nabewerker aan.

threshold Score-drempelwaarde die wordt toegepast vóór non-maximum suppression.

anchors 2D numpy.ndarray met vorm (N, 2) die de (w, h)-afmetingen bevat van de ankervakken waarop het model is getraind. Indien None wordt een ingebouwde standaardset van 5 ankers gebruikt.

nms_threshold Drempelwaarde die wordt doorgegeven aan non-maximum suppression.

nms_sigma Sigma-waarde die wordt doorgegeven aan non-maximum suppression.

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

Aangeroepen door ml.Model.predict() om de model-uitvoer na te bewerken.

model De ml.Model-instantie waaraan de nabewerker is gekoppeld.

inputs Lijst met model-invoerobjecten (gebruikt om de invoer-ROI te verkrijgen).

outputs Lijst met ruwe model-uitvoertensoren.

Retourneert een lijst met per-klasse lijsten van ((x, y, w, h), score)-tuples. Bijv. [[((x, y, w, h), score), ...], ...]. Lege klasselijsten worden behouden zodat elke lijstindex overeenkomt met de klasse-index van het model. Retourneert een lege tuple () wanneer geen enkele detectie de score-drempelwaarde overschrijdt.

class YoloLC – YOLO LC¶

Nabewerker voor de uitvoer van YOLO LC-modellen. Subklasse van YoloV2 die een andere standaard-ankerset biedt die is afgestemd op het YOLO LC-model.

class ml.postprocessing.darknet.YoloLC(threshold: float = 0.6, anchors: numpy.ndarray = None, nms_threshold: float = 0.1, nms_sigma: float = 0.1)¶

Maakt een YOLO LC-nabewerker aan.