Riferimento per ultralytics/data/split_dota.py
Nota
Questo file è disponibile su https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/split_dota .py. Se riscontri un problema, contribuisci a risolverlo inviando una Pull Request 🛠️. Grazie 🙏!
ultralytics.data.split_dota.bbox_iof(polygon1, bbox2, eps=1e-06)
Calcola gli iof tra bbox1 e bbox2.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
polygon1 |
ndarray
|
Coordinate del poligono, (n, 8). |
richiesto |
bbox2 |
ndarray
|
Caselle di delimitazione, (n ,4). |
richiesto |
Codice sorgente in ultralytics/data/split_dota.py
ultralytics.data.split_dota.load_yolo_dota(data_root, split='train')
Carica il set di dati DOTA.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
data_root |
str
|
Radice dei dati. |
richiesto |
split |
str
|
Il set di dati diviso può essere train o val. |
'train'
|
Note
La struttura di directory assunta per il set di dati DOTA: - data_root - immagini - addestramento - val - etichette - addestramento - val
Codice sorgente in 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)
Ottieni le coordinate delle finestre.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
im_size |
tuple
|
Dimensioni dell'immagine originale (h, w). |
richiesto |
crop_sizes |
List(int
|
Ritaglia le dimensioni delle finestre. |
[1024]
|
gaps |
List(int
|
Spazio tra le colture. |
[200]
|
im_rate_thr |
float
|
Soglia delle aree delle finestre divisa per le aree dell'immagine. |
0.6
|
Codice sorgente in ultralytics/data/split_dota.py
ultralytics.data.split_dota.get_window_obj(anno, windows, iof_thr=0.7)
Ottiene gli oggetti per ogni finestra.
Codice sorgente in ultralytics/data/split_dota.py
ultralytics.data.split_dota.crop_and_save(anno, windows, window_objs, im_dir, lb_dir)
Ritaglia le immagini e salva le nuove etichette.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
anno |
dict
|
Dettatura delle annotazioni, tra cui |
richiesto |
windows |
list
|
Un elenco di coordinate delle finestre. |
richiesto |
window_objs |
list
|
Un elenco di etichette all'interno di ogni finestra. |
richiesto |
im_dir |
str
|
Il percorso della directory di output delle immagini. |
richiesto |
lb_dir |
str
|
Il percorso della directory di output delle etichette. |
richiesto |
Note
La struttura di directory assunta per il set di dati DOTA: - data_root - immagini - addestramento - val - etichette - addestramento - val
Codice sorgente in ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_images_and_labels(data_root, save_dir, split='train', crop_sizes=[1024], gaps=[200])
Dividi sia le immagini che le etichette.
Note
La struttura di directory assunta per il set di dati DOTA: - data_root - immagini - split - etichette - split e la struttura della directory di output è - save_dir - immagini - split - etichette - split
Codice sorgente in ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_trainval(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
Treno diviso e set di val di DOTA.
Note
La struttura di directory assunta per il set di dati DOTA: - data_root - immagini - addestramento - val - etichette - addestramento - val e la struttura della directory di output è - save_dir - immagini - treno - val - etichette - treno - val
Codice sorgente in ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_test(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
Set di test diviso di DOTA, le etichette non sono incluse in questo set.
Note
La struttura di directory assunta per il set di dati DOTA: - data_root - immagini - test e la struttura della directory di output è: - save_dir - immagini - test