ml.postprocessing.edgeimpulse — Edge Impulse

Modulul ml.postprocessing.edgeimpulse conține clase de post-procesare pentru modelele Edge Impulse.

class Fomo – Fast Objects More Objects

Post-procesor pentru ieșirea modelului 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)

Creează un post-procesor FOMO.

threshold scorul minim necesar pentru ca o detectare să fie păstrată.

w_scale factorul de scalare orizontal aplicat lățimii celulei de grilă înainte de suprimarea non-maximelor. Valorile mai mari determină fuzionarea celulelor vecine într-o singură detectare.

h_scale factorul de scalare vertical aplicat înălțimii celulei de grilă înainte de suprimarea non-maximelor. Valorile mai mari determină fuzionarea celulelor vecine într-o singură detectare.

nms_threshold pragul IoU transmis suprimării non-maximelor.

nms_sigma valoarea sigma transmisă suprimării non-maximelor (soft-NMS).

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

Apelat de ml.Model.predict() cu modelul, intrările sale și ieșirile sale brute. Returnează o listă de liste de detectare per clasă. Fiecare detectare este un tuplu ((x, y, w, h), score). Listele de clasă goale sunt incluse astfel încât poziția fiecărei liste în ieșire să corespundă indicelui de clasă din ieșirea modelului. Returnează un tuplu gol când nu este detectat nimic.