ml.postprocessing.edgeimpulse — Edge Impulse

Le module ml.postprocessing.edgeimpulse contient des classes de post-traitement pour les modèles Edge Impulse.

class Fomo – Fast Objects More Objects

Post-traitement pour la sortie du modèle FOMO (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)

Crée un post-traitement FOMO.

threshold score minimum requis pour qu’une détection soit conservée.

w_scale facteur d’échelle horizontal appliqué à la largeur de la cellule de grille avant la suppression des non-maxima. Des valeurs plus grandes provoquent la fusion des cellules voisines en une seule détection.

h_scale facteur d’échelle vertical appliqué à la hauteur de la cellule de grille avant la suppression des non-maxima. Des valeurs plus grandes provoquent la fusion des cellules voisines en une seule détection.

nms_threshold seuil d’IoU transmis à la suppression des non-maxima.

nms_sigma valeur sigma transmise à la suppression des non-maxima (soft-NMS).

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

Invoqué par ml.Model.predict() avec le modèle, ses entrées et ses sorties brutes. Renvoie une liste de listes de détections par classe. Chaque détection est un tuple ((x, y, w, h), score). Les listes de classes vides sont incluses afin que la position de chaque liste dans la sortie corresponde à l’index de classe dans la sortie du modèle. Renvoie un tuple vide lorsque rien n’est détecté.