Referencia para ultralytics/data/split_dota.py
Nota
Este archivo está disponible en https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/split_dota .py. Si detectas algún problema, por favor, ayuda a solucionarlo contribuyendo con una Pull Request 🛠️. ¡Gracias 🙏!
ultralytics.data.split_dota.bbox_iof(polygon1, bbox2, eps=1e-06)
Calcula los iofs entre bbox1 y bbox2.
Parámetros:
Nombre | Tipo | Descripción | Por defecto |
---|---|---|---|
polygon1 |
ndarray
|
Coordenadas del polígono, (n, 8). |
necesario |
bbox2 |
ndarray
|
Cajas delimitadoras, (n ,4). |
necesario |
Código fuente en ultralytics/data/split_dota.py
ultralytics.data.split_dota.load_yolo_dota(data_root, split='train')
Carga el conjunto de datos DOTA.
Parámetros:
Nombre | Tipo | Descripción | Por defecto |
---|---|---|---|
data_root |
str
|
Raíz de los datos. |
necesario |
split |
str
|
El conjunto de datos dividido, puede ser entrenar o val. |
'train'
|
Notas
La estructura de directorios asumida para el conjunto de datos DOTA: - raíz_datos - imágenes - entrenamiento - val - etiquetas - entrenar - val
Código fuente en 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)
Obtén las coordenadas de las ventanas.
Parámetros:
Nombre | Tipo | Descripción | Por defecto |
---|---|---|---|
im_size |
tuple
|
Tamaño de la imagen original, (h, w). |
necesario |
crop_sizes |
List(int
|
Recorta el tamaño de las ventanas. |
[1024]
|
gaps |
List(int
|
Espacio entre cultivos. |
[200]
|
im_rate_thr |
float
|
Umbral de las áreas de las ventanas dividido por las áreas de la imagen. |
0.6
|
Código fuente en ultralytics/data/split_dota.py
ultralytics.data.split_dota.get_window_obj(anno, windows, iof_thr=0.7)
Obtener objetos para cada ventana.
Código fuente en ultralytics/data/split_dota.py
ultralytics.data.split_dota.crop_and_save(anno, windows, window_objs, im_dir, lb_dir)
Recorta imágenes y guarda nuevas etiquetas.
Parámetros:
Nombre | Tipo | Descripción | Por defecto |
---|---|---|---|
anno |
dict
|
Dictado de anotación, incluyendo |
necesario |
windows |
list
|
Una lista de las coordenadas de las ventanas. |
necesario |
window_objs |
list
|
Una lista de etiquetas dentro de cada ventana. |
necesario |
im_dir |
str
|
La ruta del directorio de salida de las imágenes. |
necesario |
lb_dir |
str
|
La ruta del directorio de salida de las etiquetas. |
necesario |
Notas
La estructura de directorios asumida para el conjunto de datos DOTA: - raíz_datos - imágenes - entrenamiento - val - etiquetas - entrenar - val
Código fuente en ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_images_and_labels(data_root, save_dir, split='train', crop_sizes=[1024], gaps=[200])
Divide las imágenes y las etiquetas.
Notas
La estructura de directorios asumida para el conjunto de datos DOTA: - raíz_datos - imágenes - dividir - etiquetas - dividir y la estructura del directorio de salida es - guardar_directorio - imágenes - dividir - etiquetas - dividir
Código fuente en ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_trainval(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
Tren partido y val set de DOTA.
Notas
La estructura de directorios asumida para el conjunto de datos DOTA: - raíz_datos - imágenes - entrenamiento - val - etiquetas - entrenar - val y la estructura del directorio de salida es - directorio_guardar - imágenes - entrenar - val - etiquetas - tren - val
Código fuente en ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_test(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
Conjunto de pruebas dividido de DOTA, las etiquetas no se incluyen en este conjunto.
Notas
La estructura de directorios asumida para el conjunto de datos DOTA: - raíz_datos - imágenes - prueba y la estructura del directorio de salida es - directorio_guardar - imágenes - prueba