Referencia para ultralytics/data/build.py
Nota
Este archivo está disponible en https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/build .py. Si detectas algún problema, por favor, ayuda a solucionarlo contribuyendo con una Pull Request 🛠️. ¡Gracias 🙏!
ultralytics.data.build.InfiniteDataLoader
Bases: DataLoader
Cargador de datos que reutiliza trabajadores.
Utiliza la misma sintaxis que DataLoader vainilla.
Código fuente en ultralytics/data/build.py
__init__(*args, **kwargs)
Cargador de datos que recicla infinitamente los trabajadores, hereda de DataLoader.
Código fuente en ultralytics/data/build.py
__iter__()
__len__()
reset()
Restablecer iterador.
Esto es útil cuando queremos modificar los ajustes del conjunto de datos durante el entrenamiento.
ultralytics.data.build._RepeatSampler
Muestrario que se repite eternamente.
Parámetros:
Nombre | Tipo | Descripción | Por defecto |
---|---|---|---|
sampler |
sampler
|
El muestreador para repetir. |
necesario |
Código fuente en ultralytics/data/build.py
__init__(sampler)
ultralytics.data.build.seed_worker(worker_id)
Establece la semilla del trabajador 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)
Construye YOLO Conjunto de datos.
Código fuente en ultralytics/data/build.py
ultralytics.data.build.build_grounding(cfg, img_path, json_file, batch, mode='train', rect=False, stride=32)
Construye YOLO Conjunto de datos.
Código fuente en ultralytics/data/build.py
ultralytics.data.build.build_dataloader(dataset, batch, workers, shuffle=True, rank=-1)
Devuelve un InfiniteDataLoader o un DataLoader para el conjunto de entrenamiento o validación.
Código fuente en ultralytics/data/build.py
ultralytics.data.build.check_source(source)
Comprueba el tipo de fuente y devuelve los valores de bandera correspondientes.
Código fuente en ultralytics/data/build.py
ultralytics.data.build.load_inference_source(source=None, batch=1, vid_stride=1, buffer=False)
Carga una fuente de inferencia para la detección de objetos y aplica las transformaciones necesarias.
Parámetros:
Nombre | Tipo | Descripción | Por defecto |
---|---|---|---|
source |
(str, Path, Tensor, Image, ndarray)
|
La fuente de entrada para la inferencia. |
None
|
batch |
int
|
Tamaño del lote para los cargadores de datos. Por defecto es 1. |
1
|
vid_stride |
int
|
El intervalo de fotogramas para las fuentes de vídeo. Por defecto es 1. |
1
|
buffer |
bool
|
Determina si las tramas de flujo se almacenarán en la memoria intermedia. Por defecto es Falso. |
False
|
Devuelve:
Nombre | Tipo | Descripción |
---|---|---|
dataset |
Dataset
|
Un objeto conjunto de datos para la fuente de entrada especificada. |