مرجع ل ultralytics/trackers/byte_tracker.py
ملاحظه
هذا الملف متاح في https://github.com/ultralytics/ultralytics/ نقطة / الرئيسية /ultralytics/بتتبع/byte_tracker.py. إذا اكتشفت مشكلة ، فيرجى المساعدة في إصلاحها من خلال المساهمة في طلب 🛠️ سحب. شكرا لك 🙏!
ultralytics.trackers.byte_tracker.STrack
قواعد: BaseTrack
تمثيل تتبع كائن واحد يستخدم تصفية Kalman لتقدير الحالة.
هذه الفئة مسؤولة عن تخزين جميع المعلومات المتعلقة بالمسارات الفردية وإجراء تحديثات الحالة والتنبؤات القائمة على مرشح كالمان.
سمات:
اسم | نوع | وصف |
---|---|---|
shared_kalman |
KalmanFilterXYAH
|
عامل تصفية Kalman المشترك المستخدم عبر جميع مثيلات STrack للتنبؤ. |
_tlwh |
ndarray
|
سمة خاصة لتخزين إحداثيات الزاوية العلوية اليسرى وعرض المربع المحيط وارتفاعه. |
kalman_filter |
KalmanFilterXYAH
|
مثيل مرشح Kalman المستخدم لمسار الكائن هذا. |
mean |
ndarray
|
متوسط متجه تقدير الحالة. |
covariance |
ndarray
|
التباين المشترك لتقدير الدولة. |
is_activated |
bool
|
علامة منطقية تشير إلى ما إذا كان قد تم تنشيط المسار. |
score |
float
|
درجة الثقة في المسار. |
tracklet_len |
int
|
طول المسار. |
cls |
any
|
تسمية الفئة للكائن. |
idx |
int
|
فهرس أو معرف للكائن. |
frame_id |
int
|
معرف الإطار الحالي. |
start_frame |
int
|
الإطار حيث تم اكتشاف الكائن لأول مرة. |
أساليب:
اسم | وصف |
---|---|
predict |
توقع الحالة التالية للكائن باستخدام مرشح كالمان. |
multi_predict |
توقع الحالات التالية لمسارات متعددة. |
multi_gmc |
قم بتحديث حالات المسار المتعددة باستخدام مصفوفة التماثل. |
activate |
تنشيط مسار جديد. |
re_activate |
أعد تنشيط مسار مفقود مسبقا. |
update |
تحديث حالة المسار المطابق. |
convert_coords |
تحويل المربع المحيط إلى تنسيق x-y-aspect-height. |
tlwh_to_xyah |
تحويل المربع المحيط tlwh إلى تنسيق xyah. |
شفرة المصدر في ultralytics/trackers/byte_tracker.py
12 13 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 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 |
|
result
property
احصل على نتائج التتبع الحالية.
tlwh
property
احصل على الموضع الحالي بتنسيق المربع المحيط (أعلى اليسار x ، أعلى اليسار y ، العرض ، الارتفاع).
xywh
property
احصل على الموضع الحالي بتنسيق المربع المحيط (المركز x ، المركز y ، العرض ، الارتفاع).
xywha
property
احصل على الموضع الحالي بتنسيق المربع المحيط (المركز x ، المركز y ، العرض ، الارتفاع ، الزاوية).
xyxy
property
تحويل المربع المحيط إلى تنسيق (min x ، min y ، max x ، max y) ، أي (أعلى اليسار ، أسفل اليمين).
__init__(xywh, score, cls)
تهيئة مثيل STrack جديد.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
__repr__()
إرجاع تمثيل سلسلة لكائن BYTETracker مع إطارات البداية والنهاية ومعرف المسار.
activate(kalman_filter, frame_id)
ابدأ مسارا جديدا.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
convert_coords(tlwh)
قم بتحويل تنسيق ارتفاع العرض العلوي الأيسر للمربع المحيط إلى مرادفه x-y-aaspect height.
multi_gmc(stracks, H=np.eye(2, 3))
staticmethod
تحديث حالة تتبع المواضع والتباينات باستخدام مصفوفة التماثل.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
multi_predict(stracks)
staticmethod
قم بإجراء تتبع تنبؤي متعدد الكائنات باستخدام مرشح Kalman لمسارات معينة.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
predict()
يتنبأ بالمتوسط والتغاير باستخدام مرشح كالمان.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
re_activate(new_track, frame_id, new_id=False)
يعيد تنشيط مسار مفقود مسبقا باكتشاف جديد.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
tlwh_to_xyah(tlwh)
staticmethod
تحويل المربع المحيط إلى تنسيق (المركز x ، الوسط y ، نسبة العرض إلى الارتفاع ، الارتفاع) ، حيث تكون نسبة العرض إلى الارتفاع هي العرض / ارتفاع.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
update(new_track, frame_id)
تحديث حالة المسار المطابق.
البارامترات:
اسم | نوع | وصف | افتراضي |
---|---|---|---|
new_track |
STrack
|
المسار الجديد الذي يحتوي على معلومات محدثة. |
مطلوب |
frame_id |
int
|
معرف الإطار الحالي. |
مطلوب |
شفرة المصدر في ultralytics/trackers/byte_tracker.py
ultralytics.trackers.byte_tracker.BYTETracker
BYTETracker: خوارزمية تتبع مبنية على رأس YOLOv8 للكشف عن الكائنات وتتبعها.
الفصل مسؤول عن تهيئة وتحديث وإدارة المسارات للكائنات المكتشفة في الفيديو تسلسل. يحافظ على حالة المسارات المتعقبة والمفقودة وإزالتها عبر الإطارات ، ويستخدم تصفية Kalman ل التنبؤ بمواقع الكائنات الجديدة ، وتنفيذ اقتران البيانات.
سمات:
اسم | نوع | وصف |
---|---|---|
tracked_stracks |
list[STrack]
|
قائمة المسارات التي تم تنشيطها بنجاح. |
lost_stracks |
list[STrack]
|
قائمة المسارات المفقودة. |
removed_stracks |
list[STrack]
|
قائمة المسارات التي تمت إزالتها. |
frame_id |
int
|
معرف الإطار الحالي. |
args |
namespace
|
وسيطات سطر الأوامر. |
max_time_lost |
int
|
الحد الأقصى للإطارات ليتم اعتبار المسار "مفقودا". |
kalman_filter |
object
|
كائن مرشح كالمان. |
أساليب:
اسم | وصف |
---|---|
update |
تحديث تعقب الكائنات مع الاكتشافات الجديدة. |
get_kalmanfilter |
ترجع كائن مرشح Kalman لتتبع المربعات المحيطة. |
init_track |
تهيئة تتبع الكائن مع الاكتشافات. |
get_dists |
يحسب المسافة بين المسارات والاكتشافات. |
multi_predict |
يتنبأ بموقع المسارات. |
reset_id |
إعادة تعيين عداد معرف STrack. |
joint_stracks |
يجمع بين قائمتين من stracks. |
sub_stracks |
يقوم بتصفية المسارات الموجودة في القائمة الثانية من القائمة الأولى. |
remove_duplicate_stracks |
يزيل المسارات المكررة بناء على IoU. |
شفرة المصدر في ultralytics/trackers/byte_tracker.py
212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 |
|
__init__(args, frame_rate=30)
تهيئة ملف YOLOv8 كائن لتعقب الكائنات مع وسيطات معينة ومعدل الإطارات.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
get_dists(tracks, detections)
يحسب المسافة بين المسارات والاكتشافات باستخدام IoU ودرجات الصمامات.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
get_kalmanfilter()
init_track(dets, scores, cls, img=None)
تهيئة تتبع الكائن مع الاكتشافات والنتائج باستخدام خوارزمية STrack.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
joint_stracks(tlista, tlistb)
staticmethod
اجمع بين قائمتين من المسارات في قائمة واحدة.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
multi_predict(tracks)
remove_duplicate_stracks(stracksa, stracksb)
staticmethod
قم بإزالة المسارات المكررة بمسافة غير قصوى ل IoU.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
reset()
إعادة تعيين تعقب.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
reset_id()
staticmethod
sub_stracks(tlista, tlistb)
staticmethod
DEPRECATED CODE in https://github.com/ultralytics/ultralytics/pull/1890/ stracks = {t.track_id: t for t in tlista} for t in tlistb: tid = t.track_id if stracks.get(tid, 0): del stracks[tid] return list(stracks.values())
شفرة المصدر في ultralytics/trackers/byte_tracker.py
update(results, img=None)
يقوم بتحديث تعقب الكائنات باكتشافات جديدة وإرجاع المربعات المحيطة بالكائنات المتعقبة.
شفرة المصدر في ultralytics/trackers/byte_tracker.py
253254255256257258259 260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295 296297298299300301 302303304305306307308309310311312313314315316317318319320321322323324325 326327328329330331332333334335336337338339340341342343344345346347348349350351352353 354355356357358359360361362363364365 |
|