مرجع ل ultralytics/trackers/bot_sort.py
ملاحظه
هذا الملف متاح في https://github.com/ultralytics/ultralytics/ نقطة / الرئيسية /ultralytics/بتتبع/bot_sort.py. إذا اكتشفت مشكلة ، فيرجى المساعدة في إصلاحها من خلال المساهمة في طلب 🛠️ سحب. شكرا لك 🙏!
ultralytics.trackers.bot_sort.BOTrack
قواعد: STrack
نسخة موسعة من فئة STrack ل YOLOv8، مضيفا ميزات تتبع الكائنات.
سمات:
اسم | نوع | وصف |
---|---|---|
shared_kalman |
KalmanFilterXYWH
|
مرشح Kalman مشترك لجميع مثيلات BOTrack. |
smooth_feat |
ndarray
|
ناقلات ميزة سلسة. |
curr_feat |
ndarray
|
متجه الميزة الحالية. |
features |
deque
|
deque لتخزين متجهات المعالم بطول أقصى محدد بواسطة |
alpha |
float
|
عامل تجانس للمتوسط المتحرك الأسي للميزات. |
mean |
ndarray
|
متوسط حالة مرشح كالمان. |
covariance |
ndarray
|
مصفوفة التغاير لمرشح كالمان. |
أساليب:
اسم | وصف |
---|---|
update_features |
تحديث متجه الميزات وتنعيمها باستخدام المتوسط المتحرك الأسي. |
predict |
يتنبأ بالمتوسط والتغاير باستخدام مرشح كالمان. |
re_activate |
يعيد تنشيط مسار بميزات محدثة ومعرف جديد اختياريا. |
update |
قم بتحديث YOLOv8 مثيل مع مسار جديد ومعرف الإطار. |
tlwh |
الخاصية التي تحصل على الموضع الحالي بتنسيق tlwh |
multi_predict |
يتنبأ بالمتوسط والتباين المشترك لمسارات كائنات متعددة باستخدام مرشح كالمان المشترك. |
convert_coords |
يحول إحداثيات المربع المحيط tlwh إلى تنسيق xywh. |
tlwh_to_xywh |
تحويل المربع المحيط إلى تنسيق xywh |
استخدام
bo_track = BOTrack (tlwh ، النتيجة ، cls ، الفذ) bo_track.توقع () bo_track.تحديث (new_track ، frame_id)
شفرة المصدر في ultralytics/trackers/bot_sort.py
14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 |
|
tlwh
property
الحصول على الموضع الحالي بتنسيق المربع المحيط (top left x, top left y, width, height)
.
__init__(tlwh, score, cls, feat=None, feat_history=50)
تهيئه YOLOv8 كائن مع المعلمات الزمنية ، مثل محفوظات المعالم وألفا والميزات الحالية.
شفرة المصدر في ultralytics/trackers/bot_sort.py
convert_coords(tlwh)
multi_predict(stracks)
staticmethod
يتنبأ بالمتوسط والتباين المشترك لمسارات كائنات متعددة باستخدام مرشح كالمان المشترك.
شفرة المصدر في ultralytics/trackers/bot_sort.py
predict()
يتنبأ بالمتوسط والتغاير باستخدام مرشح كالمان.
شفرة المصدر في ultralytics/trackers/bot_sort.py
re_activate(new_track, frame_id, new_id=False)
يعيد تنشيط مسار بميزات محدثة ويعين معرفا جديدا اختياريا.
شفرة المصدر في ultralytics/trackers/bot_sort.py
tlwh_to_xywh(tlwh)
staticmethod
تحويل المربع المحيط إلى تنسيق (center x, center y, width, height)
.
update(new_track, frame_id)
قم بتحديث YOLOv8 مثيل مع مسار جديد ومعرف الإطار.
update_features(feat)
تحديث متجه الميزات وتنعيمها باستخدام المتوسط المتحرك الأسي.
شفرة المصدر في ultralytics/trackers/bot_sort.py
ultralytics.trackers.bot_sort.BOTSORT
قواعد: BYTETracker
نسخة موسعة من فئة BYTETracker ل YOLOv8، مصممة لتتبع الكائنات باستخدام خوارزمية ReID وGMC.
سمات:
اسم | نوع | وصف |
---|---|---|
proximity_thresh |
float
|
عتبة القرب المكاني (IoU) بين المسارات والاكتشافات. |
appearance_thresh |
float
|
عتبة تشابه المظهر (تضمينات ReID) بين المسارات والاكتشافات. |
encoder |
object
|
كائن لمعالجة تضمينات ReID، واضبط على لا شيء إذا لم يتم تمكين ReID. |
gmc |
GMC
|
مثيل لخوارزمية GMC لاقتران البيانات. |
args |
object
|
وسيطات سطر الأوامر التي تم تحليلها والتي تحتوي على معلمات التعقب. |
أساليب:
اسم | وصف |
---|---|
get_kalmanfilter |
إرجاع مثيل KalmanFilterXYWH لتعقب الكائن. |
init_track |
تهيئة المسار مع الاكتشافات والنتائج والفئات. |
get_dists |
احصل على مسافات بين المسارات والاكتشافات باستخدام IoU و (اختياريا) ReID. |
multi_predict |
توقع وتتبع كائنات متعددة باستخدام YOLOv8 نموذج. |
استخدام
bot_sort = بوتسورت (args ، frame_rate) bot_sort.init_track (ديتس ، عشرات ، CLS ، IMG) bot_sort.multi_predict (المسارات)
ملاحظه
تم تصميم الفصل للعمل مع YOLOv8 نموذج الكشف عن الكائن ويدعم ReID فقط إذا تم تمكينه عبر args.
شفرة المصدر في ultralytics/trackers/bot_sort.py
__init__(args, frame_rate=30)
تهيئه YOLOv8 كائن مع وحدة ReID وخوارزمية GMC.
شفرة المصدر في ultralytics/trackers/bot_sort.py
get_dists(tracks, detections)
احصل على مسافات بين المسارات والاكتشافات باستخدام IoU و (اختياريا) تضمينات ReID.
شفرة المصدر في ultralytics/trackers/bot_sort.py
get_kalmanfilter()
init_track(dets, scores, cls, img=None)
تهيئة المسار مع الاكتشافات والنتائج والفئات.