Référence pour ultralytics/data/split_dota.py
Note
Ce fichier est disponible à l'adresse https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/split_dota .py. Si tu repères un problème, aide à le corriger en contribuant à une Pull Request 🛠️. Merci 🙏 !
ultralytics.data.split_dota.bbox_iof(polygon1, bbox2, eps=1e-06)
Calcule les iofs entre bbox1 et bbox2.
Paramètres :
Nom | Type | Description | Défaut |
---|---|---|---|
polygon1 |
ndarray
|
Coordonnées du polygone, (n, 8). |
requis |
bbox2 |
ndarray
|
Boîtes de délimitation, (n ,4). |
requis |
Code source dans ultralytics/data/split_dota.py
ultralytics.data.split_dota.load_yolo_dota(data_root, split='train')
Charge l'ensemble de données DOTA.
Paramètres :
Nom | Type | Description | Défaut |
---|---|---|---|
data_root |
str
|
Racine des données. |
requis |
split |
str
|
L'ensemble de données divisé, peut être train ou val. |
'train'
|
Notes
La structure de répertoire supposée pour l'ensemble de données DOTA : - data_root - images - formation - val - étiquettes - train - val
Code source dans ultralytics/data/split_dota.py
ultralytics.data.split_dota.get_windows(im_size, crop_sizes=[1024], gaps=[200], im_rate_thr=0.6, eps=0.01)
Obtiens les coordonnées des fenêtres.
Paramètres :
Nom | Type | Description | Défaut |
---|---|---|---|
im_size |
tuple
|
Taille de l'image originale, (h, w). |
requis |
crop_sizes |
List(int
|
Taille des fenêtres. |
[1024]
|
gaps |
List(int
|
Écart entre les cultures. |
[200]
|
im_rate_thr |
float
|
Seuil des zones de fenêtres divisé par les zones de l'image. |
0.6
|
Code source dans ultralytics/data/split_dota.py
ultralytics.data.split_dota.get_window_obj(anno, windows, iof_thr=0.7)
Obtiens des objets pour chaque fenêtre.
Code source dans ultralytics/data/split_dota.py
ultralytics.data.split_dota.crop_and_save(anno, windows, window_objs, im_dir, lb_dir)
Recadre les images et enregistre de nouvelles étiquettes.
Paramètres :
Nom | Type | Description | Défaut |
---|---|---|---|
anno |
dict
|
Dictée d'annotations, y compris |
requis |
windows |
list
|
Une liste de coordonnées de fenêtres. |
requis |
window_objs |
list
|
Une liste d'étiquettes à l'intérieur de chaque fenêtre. |
requis |
im_dir |
str
|
Le chemin du répertoire de sortie des images. |
requis |
lb_dir |
str
|
Le chemin du répertoire de sortie des étiquettes. |
requis |
Notes
La structure de répertoire supposée pour l'ensemble de données DOTA : - data_root - images - formation - val - étiquettes - train - val
Code source dans ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_images_and_labels(data_root, save_dir, split='train', crop_sizes=[1024], gaps=[200])
Divise à la fois les images et les étiquettes.
Notes
La structure de répertoire supposée pour l'ensemble de données DOTA : - data_root - images - diviser - étiquettes - split et la structure du répertoire de sortie est : - save_dir - images - diviser - étiquettes - diviser
Code source dans ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_trainval(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
Train divisé et ensemble de val de DOTA.
Notes
La structure de répertoire supposée pour l'ensemble de données DOTA : - data_root - images - formation - val - étiquettes - train - val et la structure du répertoire de sortie est : - save_dir - images - train - val - étiquettes - train - val
Code source dans ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_test(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
Jeu de test divisé de DOTA, les étiquettes ne sont pas incluses dans ce jeu.
Notes
La structure de répertoire supposée pour l'ensemble de données DOTA : - data_root - images - test et la structure du répertoire de sortie est : - save_dir - images - test