ml.postprocessing.edgeimpulse — Edge Impulse

De ml.postprocessing.edgeimpulse-module bevat nabewerkingsklassen voor Edge Impulse-modellen.

class Fomo – Fast Objects More Objects

Nabewerker voor de uitvoer van FOMO-modellen (Fast Objects More Objects).

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)

Maakt een FOMO-nabewerker aan.

threshold minimale score die vereist is om een detectie te behouden.

w_scale horizontale schaalfactor die wordt toegepast op de breedte van de rastercel vóór non-max-suppression. Grotere waarden zorgen ervoor dat naburige cellen worden samengevoegd tot één detectie.

h_scale verticale schaalfactor die wordt toegepast op de hoogte van de rastercel vóór non-max-suppression. Grotere waarden zorgen ervoor dat naburige cellen worden samengevoegd tot één detectie.

nms_threshold IoU-drempelwaarde die wordt doorgegeven aan non-max-suppression.

nms_sigma sigma-waarde die wordt doorgegeven aan non-max-suppression (soft-NMS).

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

Aangeroepen door ml.Model.predict() met het model, de invoer ervan en de ruwe uitvoer ervan. Retourneert een lijst met per-klasse detectielijsten. Elke detectie is een ((x, y, w, h), score)-tuple. Lege klasselijsten worden opgenomen zodat de positie van elke lijst in de uitvoer overeenkomt met de klasse-index in de model-uitvoer. Retourneert een lege tuple wanneer er niets wordt gedetecteerd.