Riferimento per ultralytics/data/build.py
Nota
Questo file è disponibile su https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/build .py. Se noti un problema, contribuisci a risolverlo inviando una Pull Request 🛠️. Grazie 🙏!
ultralytics.data.build.InfiniteDataLoader
Basi: DataLoader
Dataloader che riutilizza i lavoratori.
Utilizza la stessa sintassi del DataLoader vanilla.
Codice sorgente in ultralytics/data/build.py
__init__(*args, **kwargs)
Dataloader che ricicla all'infinito i lavoratori, eredita da DataLoader.
Codice sorgente in ultralytics/data/build.py
__iter__()
__len__()
reset()
Azzeramento dell'iteratore.
Questo è utile quando si vogliono modificare le impostazioni del dataset durante l'addestramento.
ultralytics.data.build._RepeatSampler
Un campionario che si ripete all'infinito.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
sampler |
sampler
|
Il campionatore da ripetere. |
richiesto |
Codice sorgente in ultralytics/data/build.py
__init__(sampler)
ultralytics.data.build.seed_worker(worker_id)
Imposta il seme del lavoratore del 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)
Costruisci il dataset YOLO .
Codice sorgente in ultralytics/data/build.py
ultralytics.data.build.build_grounding(cfg, img_path, json_file, batch, mode='train', rect=False, stride=32)
Costruisci il dataset YOLO .
Codice sorgente in ultralytics/data/build.py
ultralytics.data.build.build_dataloader(dataset, batch, workers, shuffle=True, rank=-1)
Restituisce un InfiniteDataLoader o un DataLoader per il set di allenamento o di validazione.
Codice sorgente in ultralytics/data/build.py
ultralytics.data.build.check_source(source)
Controlla il tipo di sorgente e restituisce i valori dei flag corrispondenti.
Codice sorgente in ultralytics/data/build.py
ultralytics.data.build.load_inference_source(source=None, batch=1, vid_stride=1, buffer=False)
Carica una fonte di inferenza per il rilevamento degli oggetti e applica le trasformazioni necessarie.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
source |
(str, Path, Tensor, Image, ndarray)
|
La fonte di input per l'inferenza. |
None
|
batch |
int
|
Dimensione del batch per i dataloader. L'impostazione predefinita è 1. |
1
|
vid_stride |
int
|
L'intervallo dei fotogrammi per le sorgenti video. Il valore predefinito è 1. |
1
|
buffer |
bool
|
Determina se i fotogrammi del flusso saranno bufferizzati. L'impostazione predefinita è False. |
False
|
Restituzione:
Nome | Tipo | Descrizione |
---|---|---|
dataset |
Dataset
|
Un oggetto dataset per la sorgente di input specificata. |