ml.postprocessing.edgeimpulse — Edge Impulse

המודול ml.postprocessing.edgeimpulse מכיל מחלקות לעיבוד מאוחר עבור מודלים של Edge Impulse.

class Fomo – Fast Objects More Objects

מבצע עיבוד מאוחר עבור פלט מודל 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)

יוצר מבצע עיבוד מאוחר של FOMO.

threshold הציון המינימלי הנדרש כדי שזיהוי יישמר.

w_scale גורם סקאלה אופקי המוחל על רוחב תא הרשת לפני דיכוי הלא-מקסימום. ערכים גדולים יותר גורמים למיזוג תאים שכנים לכדי זיהוי יחיד.

h_scale גורם סקאלה אנכי המוחל על גובה תא הרשת לפני דיכוי הלא-מקסימום. ערכים גדולים יותר גורמים למיזוג תאים שכנים לכדי זיהוי יחיד.

nms_threshold סף IoU המועבר לדיכוי הלא-מקסימום.

nms_sigma ערך סיגמא המועבר לדיכוי הלא-מקסימום (soft-NMS).

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

נקרא על ידי ml.Model.predict() עם המודל, הקלטים שלו והפלטים הגולמיים שלו. מחזיר רשימה של רשימות זיהוי לכל מחלקה. כל זיהוי הוא tuple מסוג ((x, y, w, h), score). רשימות מחלקה ריקות נכללות כך שמיקום כל רשימה בפלט תואם לאינדקס המחלקה בפלט המודל. מחזיר tuple ריק כאשר דבר אינו מזוהה.