مرجع ل ultralytics/data/build.py
ملاحظه
هذا الملف متاح في https://github.com/ultralytics/ultralytics/ نقطة / الرئيسية /ultralytics/البيانات/build.py. إذا اكتشفت مشكلة ، فيرجى المساعدة في إصلاحها من خلال المساهمة في طلب 🛠️ سحب. شكرا لك 🙏!
ultralytics.data.build.InfiniteDataLoader
قواعد: DataLoader
أداة تحميل البيانات التي تعيد استخدام العمال.
يستخدم نفس بناء الجملة مثل الفانيليا DataLoader.
شفرة المصدر في ultralytics/data/build.py
__init__(*args, **kwargs)
Dataloader الذي يعيد تدوير العمال بلا حدود ، يرث من DataLoader.
شفرة المصدر في ultralytics/data/build.py
__iter__()
__len__()
reset()
ultralytics.data.build._RepeatSampler
أخذ العينات التي تتكرر إلى الأبد.
البارامترات:
اسم | نوع | وصف | افتراضي |
---|---|---|---|
sampler |
sampler
|
أخذ العينات لتكرار. |
مطلوب |
شفرة المصدر في ultralytics/data/build.py
__init__(sampler)
ultralytics.data.build.seed_worker(worker_id)
تعيين https:// البذور لعامل أداة تحميل البياناتpytorch.org/docs/stable/notes/randomness.html#dataloader.
ultralytics.data.build.build_yolo_dataset(cfg, img_path, batch, data, mode='train', rect=False, stride=32, multi_modal=False)
بنى YOLO مجموعة البيانات.
شفرة المصدر في ultralytics/data/build.py
ultralytics.data.build.build_grounding(cfg, img_path, json_file, batch, mode='train', rect=False, stride=32)
بنى YOLO مجموعة البيانات.
شفرة المصدر في ultralytics/data/build.py
ultralytics.data.build.build_dataloader(dataset, batch, workers, shuffle=True, rank=-1)
إرجاع إنفينيتيداتولودر أو داتا لودر للتدريب أو مجموعة التحقق من الصحة.
شفرة المصدر في ultralytics/data/build.py
ultralytics.data.build.check_source(source)
تحقق من نوع المصدر وقم بإرجاع قيم العلم المقابلة.
شفرة المصدر في ultralytics/data/build.py
ultralytics.data.build.load_inference_source(source=None, batch=1, vid_stride=1, buffer=False)
يقوم بتحميل مصدر استدلال لاكتشاف الكائن ويطبق التحويلات الضرورية.
البارامترات:
اسم | نوع | وصف | افتراضي |
---|---|---|---|
source |
(str, Path, Tensor, Image, ndarray)
|
مصدر الإدخال للاستدلال. |
None
|
batch |
int
|
حجم الدفعة لأدوات تحميل البيانات. الافتراضي هو 1. |
1
|
vid_stride |
int
|
الفاصل الزمني للإطار لمصادر الفيديو. الافتراضي هو 1. |
1
|
buffer |
bool
|
تحديد ما إذا كان سيتم تخزين إطارات الدفق مؤقتا. الافتراضي هو خطأ. |
False
|
ارجاع:
اسم | نوع | وصف |
---|---|---|
dataset |
Dataset
|
كائن مجموعة بيانات لمصدر الإدخال المحدد. |