Reference for ultralytics/data/build.py
Note
Full source code for this file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/data/build.py. Help us fix any issues you see by submitting a Pull Request 🛠️. Thank you 🙏!
ultralytics.data.build.InfiniteDataLoader
Bases: DataLoader
Dataloader that reuses workers. Uses same syntax as vanilla DataLoader.
Source code in ultralytics/data/build.py
__init__(*args, **kwargs)
Dataloader that infinitely recycles workers, inherits from DataLoader.
Source code in ultralytics/data/build.py
__iter__()
__len__()
reset()
ultralytics.data.build._RepeatSampler
Sampler that repeats forever.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
sampler |
sampler
|
The sampler to repeat. |
required |
Source code in 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)
Build YOLO Dataset
Source code in ultralytics/data/build.py
ultralytics.data.build.build_dataloader(dataset, batch, workers, shuffle=True, rank=-1)
Return an InfiniteDataLoader or DataLoader for training or validation set.
Source code in ultralytics/data/build.py
ultralytics.data.build.check_source(source)
Check source type and return corresponding flag values.
Source code in ultralytics/data/build.py
ultralytics.data.build.load_inference_source(source=None, imgsz=640, vid_stride=1, buffer=False)
Loads an inference source for object detection and applies necessary transformations.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
source |
(str, Path, Tensor, Image, ndarray)
|
The input source for inference. |
None
|
imgsz |
int
|
The size of the image for inference. Default is 640. |
640
|
vid_stride |
int
|
The frame interval for video sources. Default is 1. |
1
|
buffer |
bool
|
Determined whether stream frames will be buffered. Default is False. |
False
|
Returns:
Name | Type | Description |
---|---|---|
dataset |
Dataset
|
A dataset object for the specified input source. |