Referentie voor ultralytics/data/split_dota.py
Opmerking
Dit bestand is beschikbaar op https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/split_dota .py. Als je een probleem ziet, help het dan oplossen door een Pull Request 🛠️ bij te dragen. Bedankt 🙏!
ultralytics.data.split_dota.bbox_iof(polygon1, bbox2, eps=1e-06)
Bereken iofs tussen bbox1 en bbox2.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
polygon1 |
ndarray
|
Polygooncoördinaten, (n, 8). |
vereist |
bbox2 |
ndarray
|
Begrenzende vakken, (n ,4). |
vereist |
Broncode in ultralytics/data/split_dota.py
ultralytics.data.split_dota.load_yolo_dota(data_root, split='train')
Laad DOTA dataset.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
data_root |
str
|
Gegevenswortel. |
vereist |
split |
str
|
De gesplitste gegevensset kan train of val zijn. |
'train'
|
Opmerkingen
De aangenomen mapstructuur voor de DOTA dataset: - data_root - afbeeldingen - trainen - waarde - labels - trainen - val
Broncode 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)
Krijg de coördinaten van vensters.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
im_size |
tuple
|
Origineel afbeeldingsformaat, (h, b). |
vereist |
crop_sizes |
List(int
|
Grootte van vensters bijsnijden. |
[1024]
|
gaps |
List(int
|
Kloof tussen gewassen. |
[200]
|
im_rate_thr |
float
|
Drempelwaarde van vensteroppervlakken gedeeld door beeldoppervlakken. |
0.6
|
Broncode in ultralytics/data/split_dota.py
ultralytics.data.split_dota.get_window_obj(anno, windows, iof_thr=0.7)
Objecten ophalen voor elk venster.
Broncode in ultralytics/data/split_dota.py
ultralytics.data.split_dota.crop_and_save(anno, windows, window_objs, im_dir, lb_dir)
Afbeeldingen bijsnijden en nieuwe labels opslaan.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
anno |
dict
|
Annotatiedict, inclusief |
vereist |
windows |
list
|
Een lijst met coördinaten van vensters. |
vereist |
window_objs |
list
|
Een lijst met labels in elk venster. |
vereist |
im_dir |
str
|
Het pad van de uitvoermap met afbeeldingen. |
vereist |
lb_dir |
str
|
Het pad van de uitvoermap van labels. |
vereist |
Opmerkingen
De aangenomen mapstructuur voor de DOTA dataset: - data_root - afbeeldingen - trainen - waarde - labels - trainen - val
Broncode 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])
Splits zowel afbeeldingen als labels.
Opmerkingen
De aangenomen mapstructuur voor de DOTA dataset: - data_root - afbeeldingen - splitsing - labels - gesplitst en de uitvoermapstructuur is: - opslaan_dir - afbeeldingen - gesplitst - labels - splitsen
Broncode in ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_trainval(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
Gesplitste trein en val set van DOTA.
Opmerkingen
De aangenomen mapstructuur voor de DOTA dataset: - data_root - afbeeldingen - trainen - waarde - labels - trainen - waarde en de uitvoermapstructuur is: - opslaan_dir - afbeeldingen - trein - waarde - labels - trein - val
Broncode in ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_test(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
Gesplitste testset van DOTA, labels zijn niet opgenomen in deze set.
Opmerkingen
De aangenomen mapstructuur voor de DOTA dataset: - data_root - afbeeldingen - test en de uitvoermapstructuur is: - save_dir - afbeeldingen - test