ml.postprocessing.ultralytics — Ultralytics

המודול ml.postprocessing.ultralytics מכיל מחלקות עבור Ultralytics.

class YoloV5 – YOLO V5

משמש לעיבוד מאוחר של פלט מודל YOLO V5.

class ml.postprocessing.ultralytics.YoloV5(threshold: float = 0.6, nms_threshold: float = 0.1, nms_sigma: float = 0.1)

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

threshold סף הציון לסינון זיהויים לפני NMS.

nms_threshold סף IoU שבו משתמש NMS לדיכוי תיבות תוחמות חופפות.

nms_sigma ערך סיגמא שבו משתמש Soft-NMS לדעיכת ציון.

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

מבצע עיבוד מאוחר של רשימת המערכים outputs שהופקה על ידי model עבור רשימת ה-inputs הנתונה. מחזיר רשימה של רשימות לכל מחלקה; כל רשימה פנימית מכילה tuples מסוג ((x, y, w, h), score) בקואורדינטות ROI. רשימות מחלקה ריקות נשמרות כך שכל אינדקס רשימה תואם לאינדקס המחלקה המתאים בפלט המודל.

class YoloV8 – YOLO V8

משמש לעיבוד מאוחר של פלט מודל YOLO V8.

class ml.postprocessing.ultralytics.YoloV8(threshold: float = 0.6, nms_threshold: float = 0.1, nms_sigma: float = 0.1)

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

threshold סף הציון לסינון זיהויים לפני NMS.

nms_threshold סף IoU שבו משתמש NMS לדיכוי תיבות תוחמות חופפות.

nms_sigma ערך סיגמא שבו משתמש Soft-NMS לדעיכת ציון.

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

מבצע עיבוד מאוחר של רשימת המערכים outputs שהופקה על ידי model עבור רשימת ה-inputs הנתונה. מחזיר רשימה של רשימות לכל מחלקה; כל רשימה פנימית מכילה tuples מסוג ((x, y, w, h), score) בקואורדינטות ROI. רשימות מחלקה ריקות נשמרות כך שכל אינדקס רשימה תואם לאינדקס המחלקה המתאים בפלט המודל.