Ссылка для ultralytics/data/build.py
Примечание
Этот файл доступен по адресу https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/build .py. Если ты обнаружил проблему, пожалуйста, помоги исправить ее, создав Pull Request 🛠️. Спасибо 🙏!
ultralytics.data.build.InfiniteDataLoader
Базы: DataLoader
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)
Установи семя рабочего dataloader 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 Dataset.
Исходный код в ultralytics/data/build.py
ultralytics.data.build.build_grounding(cfg, img_path, json_file, batch, mode='train', rect=False, stride=32)
Построй YOLO Dataset.
Исходный код в ultralytics/data/build.py
ultralytics.data.build.build_dataloader(dataset, batch, workers, shuffle=True, rank=-1)
Возвращай InfiniteDataLoader или DataLoader для тренировочного или проверочного набора.
Исходный код в 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. |
False
|
Возвращается:
Имя | Тип | Описание |
---|---|---|
dataset |
Dataset
|
Объект набора данных для указанного источника входного сигнала. |