Référence pour ultralytics/data/build.py
Note
Ce fichier est disponible à l'adresse https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/build .py. Si tu repères un problème, aide à le corriger en contribuant à une Pull Request 🛠️. Merci 🙏 !
ultralytics.data.build.InfiniteDataLoader
Bases : DataLoader
Chargeur de données qui réutilise les travailleurs.
Utilise la mĂŞme syntaxe que le DataLoader vanille.
Code source dans ultralytics/data/build.py
__init__(*args, **kwargs)
Dataloader qui recycle infiniment les travailleurs, hérite de DataLoader.
Code source dans ultralytics/data/build.py
__iter__()
__len__()
reset()
Réinitialise l'itérateur.
C'est utile lorsque nous voulons modifier les paramètres de l'ensemble de données pendant la formation.
ultralytics.data.build._RepeatSampler
Un échantillonnage qui se répète à l'infini.
Paramètres :
Nom | Type | Description | DĂ©faut |
---|---|---|---|
sampler |
sampler
|
L'échantillonneur à répéter. |
requis |
Code source dans ultralytics/data/build.py
__init__(sampler)
ultralytics.data.build.seed_worker(worker_id)
DĂ©finis la graine du travailleur du 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)
Construis l'ensemble de données YOLO .
Code source dans ultralytics/data/build.py
ultralytics.data.build.build_grounding(cfg, img_path, json_file, batch, mode='train', rect=False, stride=32)
Construis l'ensemble de données YOLO .
Code source dans ultralytics/data/build.py
ultralytics.data.build.build_dataloader(dataset, batch, workers, shuffle=True, rank=-1)
Renvoie un InfiniteDataLoader ou un DataLoader pour l'ensemble de formation ou de validation.
Code source dans ultralytics/data/build.py
ultralytics.data.build.check_source(source)
VĂ©rifie le type de source et renvoie les valeurs des drapeaux correspondants.
Code source dans ultralytics/data/build.py
ultralytics.data.build.load_inference_source(source=None, batch=1, vid_stride=1, buffer=False)
Charge une source d'inférence pour la détection des objets et applique les transformations nécessaires.
Paramètres :
Nom | Type | Description | DĂ©faut |
---|---|---|---|
source |
(str, Path, Tensor, Image, ndarray)
|
La source d'entrée pour l'inférence. |
None
|
batch |
int
|
Taille du lot pour les chargeurs de données. La valeur par défaut est 1. |
1
|
vid_stride |
int
|
L'intervalle de trame pour les sources vidéo. La valeur par défaut est 1. |
1
|
buffer |
bool
|
Détermine si les trames de flux seront mises en mémoire tampon. La valeur par défaut est False. |
False
|
Retourne :
Nom | Type | Description |
---|---|---|
dataset |
Dataset
|
Un objet de jeu de données pour la source d'entrée spécifiée. |
Code source dans ultralytics/data/build.py
Créé le 2023-11-12, Mis à jour le 2024-05-08
Auteurs : Burhan-Q (1), Laughing-q (1), glenn-jocher (3)