ml.postprocessing.edgeimpulse — Edge Impulse

A ml.postprocessing.edgeimpulse modul az Edge Impulse modellekhez tartalmaz utófeldolgozó osztályokat.

class Fomo – Fast Objects More Objects

Utófeldolgozó a FOMO (Fast Objects More Objects) modell kimenetéhez.

class ml.postprocessing.edgeimpulse.Fomo(threshold: float = 0.4, w_scale: float = 1.414214, h_scale: float = 1.414214, nms_threshold: float = 0.1, nms_sigma: float = 0.001)

Létrehoz egy FOMO utófeldolgozót.

threshold az észlelés megtartásához szükséges minimális pontszám.

w_scale a rácscella szélességére a nem maximális elnyomás előtt alkalmazott vízszintes skálázási tényező. A nagyobb értékek hatására a szomszédos cellák egyetlen észleléssé olvadnak össze.

h_scale a rácscella magasságára a nem maximális elnyomás előtt alkalmazott függőleges skálázási tényező. A nagyobb értékek hatására a szomszédos cellák egyetlen észleléssé olvadnak össze.

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

nms_sigma a nem maximális elnyomásnak (soft-NMS) átadott szigma érték.

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

A ml.Model.predict() hívja meg a modellel, annak bemeneteivel és nyers kimeneteivel. Visszaad egy osztályonkénti észlelési listákból álló listát. Minden észlelés egy ((x, y, w, h), score) tuple. Az üres osztálylisták is szerepelnek benne, így a kimenetben minden lista pozíciója megfelel a modell kimenetében szereplő osztályindexnek. Üres tuple-t ad vissza, ha semmi sem kerül észlelésre.