`ml.postprocessing.darknet` — Darknet¶

A ml.postprocessing.darknet modul a Darknet alapú objektumészlelő modellekhez tartalmaz utófeldolgozókat.

class YoloV2 – YOLO V2¶

Utófeldolgozó a YOLO V2 modell kimeneteihez.

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

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

threshold A nem maximális elnyomás előtt alkalmazott pontszám-küszöbérték.

anchors Egy (N, 2) alakú 2D numpy.ndarray, amely a modell betanításához használt (w, h) horgonydoboz-méreteket tartalmazza. Ha None, egy beépített, 5 horgonyból álló alapértelmezett halmaz kerül felhasználásra.

nms_threshold A nem maximális elnyomásnak átadott küszöbérték.

nms_sigma A nem maximális elnyomásnak átadott szigma érték.

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

A ml.Model.predict() hívja meg a modell kimeneteinek utófeldolgozásához.

model A ml.Model példány, amelyhez az utófeldolgozó csatolva van.

inputs A modell bemeneti objektumainak listája (a bemeneti ROI lekéréséhez használatos).

outputs A modell nyers kimeneti tenzorainak listája.

Visszaad egy osztályonkénti listákból álló listát, amelyek ((x, y, w, h), score) tuple-öket tartalmaznak. Pl. [[((x, y, w, h), score), ...], ...]. Az üres osztálylisták megmaradnak, így minden listaindex megfelel a modell osztályindexének. Üres tuple-t () ad vissza, ha egyetlen észlelés sem éri el a pontszám-küszöbértéket.

class YoloLC – YOLO LC¶

Utófeldolgozó a YOLO LC modell kimeneteihez. A YoloV2 alosztálya, amely egy eltérő, a YOLO LC modellhez hangolt alapértelmezett horgonyhalmazt biztosít.

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

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