Referans için ultralytics/data/build.py
Not
Bu dosya https://github.com/ultralytics/ultralytics/blob/main/ ultralytics/data/build .py adresinde mevcuttur. Bir sorun tespit ederseniz lütfen bir Çekme İsteği 🛠️ ile katkıda bulunarak düzeltilmesine yardımcı olun. Teşekkürler 🙏!
ultralytics.data.build.InfiniteDataLoader
Üsler: DataLoader
Çalışanları yeniden kullanan veri yükleyici.
Vanilla DataLoader ile aynı sözdizimini kullanır.
Kaynak kodu ultralytics/data/build.py
__init__(*args, **kwargs)
İşçileri sonsuza kadar geri dönüştüren Dataloader, DataLoader'dan miras alır.
Kaynak kodu ultralytics/data/build.py
__iter__()
__len__()
reset()
Yineleyiciyi sıfırla.
Bu, eğitim sırasında veri kümesinin ayarlarını değiştirmek istediğimizde kullanışlıdır.
ultralytics.data.build._RepeatSampler
Sonsuza kadar tekrar eden örnekleyici.
Parametreler:
İsim | Tip | Açıklama | Varsayılan |
---|---|---|---|
sampler |
sampler
|
Tekrarlanacak örnekleyici. |
gerekli |
Kaynak kodu ultralytics/data/build.py
__init__(sampler)
ultralytics.data.build.seed_worker(worker_id)
Set dataloader worker seed 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 Veri Kümesini Oluşturun.
Kaynak kodu ultralytics/data/build.py
ultralytics.data.build.build_grounding(cfg, img_path, json_file, batch, mode='train', rect=False, stride=32)
YOLO Veri Kümesini Oluşturun.
Kaynak kodu ultralytics/data/build.py
ultralytics.data.build.build_dataloader(dataset, batch, workers, shuffle=True, rank=-1)
Eğitim veya doğrulama kümesi için bir InfiniteDataLoader veya DataLoader döndürür.
Kaynak kodu ultralytics/data/build.py
ultralytics.data.build.check_source(source)
Kaynak türünü kontrol edin ve ilgili bayrak değerlerini döndürün.
Kaynak kodu ultralytics/data/build.py
ultralytics.data.build.load_inference_source(source=None, batch=1, vid_stride=1, buffer=False)
Nesne algılama için bir çıkarım kaynağı yükler ve gerekli dönüşümleri uygular.
Parametreler:
İsim | Tip | Açıklama | Varsayılan |
---|---|---|---|
source |
(str, Path, Tensor, Image, ndarray)
|
Çıkarım için girdi kaynağı. |
None
|
batch |
int
|
Veri yükleyicileri için toplu iş boyutu. Varsayılan değer 1'dir. |
1
|
vid_stride |
int
|
Video kaynakları için kare aralığı. Varsayılan değer 1'dir. |
1
|
buffer |
bool
|
Akış çerçevelerinin tamponlanıp tamponlanmayacağını belirler. Varsayılan değer False'dir. |
False
|
İade:
İsim | Tip | Açıklama |
---|---|---|
dataset |
Dataset
|
Belirtilen giriş kaynağı için bir veri kümesi nesnesi. |
Kaynak kodu ultralytics/data/build.py
Oluşturuldu 2023-11-12, Güncellendi 2024-05-08
Yazarlar: Burhan-Q (1), Laughing-q (1), glenn-jocher (3)