Referência para ultralytics/data/build.py
Nota
Este ficheiro está disponível em https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/build .py. Se detectares um problema, por favor ajuda a corrigi-lo contribuindo com um Pull Request 🛠️. Obrigado 🙏!
ultralytics.data.build.InfiniteDataLoader
Bases: DataLoader
Carregador de dados que reutiliza trabalhadores.
Usa a mesma sintaxe do DataLoader básico.
Código fonte em ultralytics/data/build.py
__init__(*args, **kwargs)
Dataloader que recicla infinitamente os trabalhadores, herda de DataLoader.
Código fonte em ultralytics/data/build.py
__iter__()
__len__()
reset()
Repõe o iterador.
Isto é útil quando queremos modificar as definições do conjunto de dados durante o treino.
ultralytics.data.build._RepeatSampler
Um sampler que se repete para sempre.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
sampler |
sampler
|
O sampler para repetir. |
necessário |
Código fonte em ultralytics/data/build.py
__init__(sampler)
ultralytics.data.build.seed_worker(worker_id)
Define a semente do trabalhador do 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)
Constrói o conjunto de dados YOLO .
Código fonte em ultralytics/data/build.py
ultralytics.data.build.build_grounding(cfg, img_path, json_file, batch, mode='train', rect=False, stride=32)
Constrói o conjunto de dados YOLO .
Código fonte em ultralytics/data/build.py
ultralytics.data.build.build_dataloader(dataset, batch, workers, shuffle=True, rank=-1)
Devolve um InfiniteDataLoader ou DataLoader para o conjunto de treino ou validação.
Código fonte em ultralytics/data/build.py
ultralytics.data.build.check_source(source)
Verifica o tipo de fonte e devolve os valores de marcação correspondentes.
Código fonte em ultralytics/data/build.py
ultralytics.data.build.load_inference_source(source=None, batch=1, vid_stride=1, buffer=False)
Carrega uma fonte de inferência para deteção de objectos e aplica as transformações necessárias.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
source |
(str, Path, Tensor, Image, ndarray)
|
A fonte de entrada para a inferência. |
None
|
batch |
int
|
Tamanho do lote para os carregadores de dados. A predefinição é 1. |
1
|
vid_stride |
int
|
O intervalo de fotogramas para fontes de vídeo. A predefinição é 1. |
1
|
buffer |
bool
|
Determina se os quadros de fluxo serão armazenados em buffer. A predefinição é Falso. |
False
|
Devolve:
Nome | Tipo | Descrição |
---|---|---|
dataset |
Dataset
|
Um objeto de conjunto de dados para a fonte de entrada especificada. |
Código fonte em ultralytics/data/build.py
Criado em 2023-11-12, Atualizado em 2024-05-08
Autores: Burhan-Q (1), Laughing-q (1), glenn-jocher (3)