`ml.postprocessing.darknet` — Darknet¶

המודול ml.postprocessing.darknet מכיל מבצעי עיבוד מאוחר עבור מודלים לזיהוי עצמים מבוססי Darknet.

class YoloV2 – YOLO V2¶

מבצע עיבוד מאוחר עבור פלטי מודל YOLO V2.

class ml.postprocessing.darknet.YoloV2(threshold: float = 0.6, anchors: numpy.ndarray = None, nms_threshold: float = 0.1, nms_sigma: float = 0.1)¶

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

threshold סף הציון המוחל לפני דיכוי הלא-מקסימום (non-maximum suppression).

anchors numpy.ndarray דו-ממדי בצורה (N, 2) המכיל את מידות תיבות העוגן (w, h) שעליהן אומן המודל. אם None, נעשה שימוש בקבוצת ברירת מחדל מובנית של 5 עוגנים.

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

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

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

נקרא על ידי ml.Model.predict() כדי לבצע עיבוד מאוחר של פלטי המודל.

model מופע ה-ml.Model שאליו מצורף מבצע העיבוד המאוחר.

inputs רשימה של אובייקטי קלט של המודל (משמשת להשגת אזור העניין (ROI) של הקלט).

outputs רשימה של טנזורי פלט גולמיים של המודל.

מחזיר רשימה של רשימות לכל מחלקה של tuples מסוג ((x, y, w, h), score). למשל [[((x, y, w, h), score), ...], ...]. רשימות מחלקה ריקות נשמרות כך שכל אינדקס רשימה תואם לאינדקס המחלקה של המודל. מחזיר tuple ריק () כאשר אף זיהוי אינו עובר את סף הציון.

class YoloLC – YOLO LC¶

מבצע עיבוד מאוחר עבור פלטי מודל YOLO LC. תת-מחלקה של YoloV2 המספקת קבוצת עוגנים שונה כברירת מחדל, מכווננת עבור מודל YOLO LC.

class ml.postprocessing.darknet.YoloLC(threshold: float = 0.6, anchors: numpy.ndarray = None, nms_threshold: float = 0.1, nms_sigma: float = 0.1)¶

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