مرجع ل ultralytics/data/loaders.py
ملاحظه
هذا الملف متاح في https://github.com/ultralytics/ultralytics/ نقطة / الرئيسية /ultralytics/البيانات/loaders.py. إذا اكتشفت مشكلة ، فيرجى المساعدة في إصلاحها من خلال المساهمة في طلب 🛠️ سحب. شكرا لك 🙏!
ultralytics.data.loaders.SourceTypes
dataclass
فئة لتمثيل أنواع مختلفة من مصادر الإدخال للتنبؤات.
شفرة المصدر في ultralytics/data/loaders.py
ultralytics.data.loaders.LoadStreams
محمل الدفق لأنواع مختلفة من تدفقات الفيديو ، يدعم تدفقات RTSP و RTMP و HTTP و TCP.
سمات:
اسم | نوع | وصف |
---|---|---|
sources |
str
|
مسارات إدخال المصدر أو عناوين URL لتدفقات الفيديو. |
vid_stride |
int
|
خطوة معدل إطارات الفيديو ، الإعدادات الافتراضية هي 1. |
buffer |
bool
|
ما إذا كنت تريد تخزين تدفقات الإدخال مؤقتا ، يتم الإعدادات الافتراضية إلى False. |
running |
bool
|
ضع علامة للإشارة إلى ما إذا كان مؤشر ترابط الدفق قيد التشغيل. |
mode |
str
|
اضبط على "بث" للإشارة إلى الالتقاط في الوقت الفعلي. |
imgs |
list
|
قائمة إطارات الصور لكل دفق. |
fps |
list
|
قائمة FPS لكل دفق. |
frames |
list
|
قائمة الإطارات الإجمالية لكل دفق. |
threads |
list
|
قائمة سلاسل الرسائل لكل دفق. |
shape |
list
|
قائمة الأشكال لكل دفق. |
caps |
list
|
قائمة السيرة الذاتية 2. فيديوالتقاط الكائنات لكل دفق. |
bs |
int
|
حجم الدفعة للمعالجة. |
أساليب:
اسم | وصف |
---|---|
__init__ |
تهيئة محمل الدفق. |
update |
قراءة إطارات الدفق في موضوع الخفي. |
close |
أغلق محمل الدفق وحرر الموارد. |
__iter__ |
إرجاع كائن مكرر للفئة. |
__next__ |
إرجاع مسارات المصدر والصور المحولة والأصلية للمعالجة. |
__len__ |
إرجاع طول كائن المصادر. |
شفرة المصدر في ultralytics/data/loaders.py
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 |
|
__init__(sources='file.streams', vid_stride=1, buffer=False)
تهيئة متغيرات المثيل والتحقق من وجود أشكال تدفق إدخال متسقة.
شفرة المصدر في ultralytics/data/loaders.py
__iter__()
__len__()
__next__()
إرجاع مسارات المصدر والصور المحولة والأصلية للمعالجة.
شفرة المصدر في ultralytics/data/loaders.py
close()
أغلق محمل الدفق وحرر الموارد.
شفرة المصدر في ultralytics/data/loaders.py
update(i, cap, stream)
قراءة البث i
إطارات في الموضوع الخفي.
شفرة المصدر في ultralytics/data/loaders.py
ultralytics.data.loaders.LoadScreenshots
YOLOv8 أداة تحميل بيانات لقطة الشاشة.
يدير هذا الفصل تحميل صور لقطة الشاشة للمعالجة باستخدام YOLOv8. مناسب للاستخدام مع yolo predict source=screen
.
سمات:
اسم | نوع | وصف |
---|---|---|
source |
str
|
يشير إدخال المصدر إلى الشاشة التي يجب التقاطها. |
screen |
int
|
رقم الشاشة المراد التقاطه. |
left |
int
|
الإحداثيات اليسرى لمنطقة التقاط الشاشة. |
top |
int
|
الإحداثيات العلوية لمنطقة التقاط الشاشة. |
width |
int
|
عرض منطقة التقاط الشاشة. |
height |
int
|
ارتفاع منطقة التقاط الشاشة. |
mode |
str
|
اضبط على "بث" للإشارة إلى الالتقاط في الوقت الفعلي. |
frame |
int
|
عداد للإطارات الملتقطة. |
sct |
mss
|
كائن التقاط الشاشة من |
bs |
int
|
حجم الدفعة، مضبوط على 1. |
monitor |
dict
|
مراقبة تفاصيل التكوين. |
أساليب:
اسم | وصف |
---|---|
__iter__ |
إرجاع كائن مكرر. |
__next__ |
يلتقط لقطة الشاشة التالية ويعيدها . |
شفرة المصدر في ultralytics/data/loaders.py
__init__(source)
المصدر = [screen_number ارتفاع عرض أعلى اليسار] (بكسل).
شفرة المصدر في ultralytics/data/loaders.py
__iter__()
__next__()
التقاط شاشة MSS: احصل على وحدات بكسل أولية من الشاشة كصفيف NP.
شفرة المصدر في ultralytics/data/loaders.py
ultralytics.data.loaders.LoadImagesAndVideos
YOLOv8 محمل بيانات الصور / الفيديو.
تدير هذه الفئة التحميل والمعالجة المسبقة لبيانات الصور والفيديو ل YOLOv8. وهو يدعم التحميل من تنسيقات مختلفة ، بما في ذلك ملفات الصور الفردية وملفات الفيديو وقوائم مسارات الصور والفيديو.
سمات:
اسم | نوع | وصف |
---|---|---|
files |
list
|
قائمة مسارات ملفات الصور والفيديو. |
nf |
int
|
إجمالي عدد الملفات (الصور ومقاطع الفيديو). |
video_flag |
list
|
علامات تشير إلى ما إذا كان الملف عبارة عن فيديو (صواب) أو صورة (خطأ). |
mode |
str
|
الوضع الحالي ، "صورة" أو "فيديو". |
vid_stride |
int
|
خطوة لمعدل إطارات الفيديو، الإعدادات الافتراضية هي 1. |
bs |
int
|
حجم الدفعة ، مضبوط على 1 لهذه الفئة. |
cap |
VideoCapture
|
كائن التقاط الفيديو ل OpenCV. |
frame |
int
|
عداد الإطار للفيديو. |
frames |
int
|
إجمالي عدد الإطارات في الفيديو. |
count |
int
|
عداد للتكرار ، تمت تهيئته عند 0 أثناء |
أساليب:
اسم | وصف |
---|---|
_new_video |
إنشاء سيرة ذاتية جديدة. كائن VideoCapture لمسار فيديو معين. |
شفرة المصدر في ultralytics/data/loaders.py
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 |
|
__init__(path, batch=1, vid_stride=1)
تهيئة أداة تحميل البيانات ورفع فايلنوتفاوندخطأ إذا لم يتم العثور على الملف.
شفرة المصدر في ultralytics/data/loaders.py
__iter__()
__len__()
__next__()
إرجاع الدفعة التالية من الصور أو إطارات الفيديو مع مساراتها وبيانات التعريف الخاصة بها.
شفرة المصدر في ultralytics/data/loaders.py
ultralytics.data.loaders.LoadPilAndNumpy
قم بتحميل الصور من صفائف PIL و Numpy لمعالجة الدفعات.
تم تصميم هذه الفئة لإدارة التحميل والمعالجة المسبقة لبيانات الصورة من تنسيقات PIL و Numpy. يقوم بإجراء التحقق الأساسي وتحويل التنسيق للتأكد من أن الصور بالتنسيق المطلوب ل معالجة المصب.
سمات:
اسم | نوع | وصف |
---|---|---|
paths |
list
|
قائمة مسارات الصور أو أسماء الملفات التي تم إنشاؤها تلقائيا. |
im0 |
list
|
قائمة الصور المخزنة كمصفوفات Numpy. |
mode |
str
|
نوع البيانات التي تتم معالجتها ، يتم تعيينها افتراضيا إلى "صورة". |
bs |
int
|
حجم الدفعة ، أي ما يعادل طول |
أساليب:
اسم | وصف |
---|---|
_single_check |
التحقق من صحة صورة واحدة وتنسيقها إلى صفيف Numpy. |
شفرة المصدر في ultralytics/data/loaders.py
__init__(im0)
تهيئة PIL و Numpy Dataloader.
شفرة المصدر في ultralytics/data/loaders.py
__iter__()
__len__()
__next__()
إرجاع مسارات الدفعات، الصور، الصور المعالجة، بلا، ''.
ultralytics.data.loaders.LoadTensor
تحميل الصور من torch.Tensor بيانات.
تدير هذه الفئة التحميل والمعالجة المسبقة لبيانات الصورة من PyTorch الموترات لمزيد من المعالجة.
سمات:
اسم | نوع | وصف |
---|---|---|
im0 |
Tensor
|
المدخلات tensor تحتوي على الصورة (الصور). |
bs |
int
|
حجم الدفعة ، يستدل عليه من شكل |
mode |
str
|
الوضع الحالي ، اضبط على "صورة". |
paths |
list
|
قائمة مسارات الصور أو أسماء الملفات. |
count |
int
|
عداد للتكرار ، تمت تهيئته عند 0 أثناء |
أساليب:
اسم | وصف |
---|---|
_single_check |
التحقق من صحة وربما تعديل الإدخال tensor. |
شفرة المصدر في ultralytics/data/loaders.py
__init__(im0)
تهيئه Tensor أداة تحميل البيانات.
شفرة المصدر في ultralytics/data/loaders.py
__iter__()
__len__()
ultralytics.data.loaders.autocast_list(source)
يدمج قائمة مصادر من أنواع مختلفة في قائمة من صفائف numpy أو صور PIL.
شفرة المصدر في ultralytics/data/loaders.py
ultralytics.data.loaders.get_best_youtube_url(url, use_pafy=True)
يسترجع عنوان URL لأفضل دفق فيديو MP4 جودة من فيديو YouTube معين.
تستخدم هذه الوظيفة مكتبة pafy أو yt_dlp لاستخراج معلومات الفيديو من YouTube. ثم يجد الأعلى تنسيق MP4 عالي الجودة يحتوي على برنامج ترميز فيديو ولكن لا يوجد برنامج ترميز صوتي ، ويعيد عنوان URL لدفق الفيديو هذا.
البارامترات:
اسم | نوع | وصف | افتراضي |
---|---|---|---|
url |
str
|
عنوان URL لفيديو يوتيوب. |
مطلوب |
use_pafy |
bool
|
استخدم الحزمة pafy ، الافتراضي = True ، وإلا استخدم yt_dlp الحزمة. |
True
|
ارجاع:
نوع | وصف |
---|---|
str
|
عنوان URL لأفضل دفق فيديو MP4 جودة ، أو لا شيء إذا لم يتم العثور على دفق مناسب. |