Conjunto de datos DOTA128#
Introducción#
Ultralytics DOTA128 es un conjunto de datos de detección de objetos orientados pequeño pero versátil, compuesto por 128 imágenes del conjunto DOTAv1 que sirven tanto para entrenamiento como para validación. Este conjunto de datos es ideal para probar y depurar modelos de cajas delimitadoras orientadas (OBB) o para experimentar con nuevos métodos de detección. Con 128 imágenes, es lo bastante pequeño para gestionarlo fácilmente y lo bastante diverso para probar si hay errores en los flujos de entrenamiento y hacer una comprobación de coherencia antes de entrenar con conjuntos de datos más grandes.
Estructura del conjunto de datos#
- Imágenes: 128 teselas aéreas (todas en la carpeta train, utilizadas tanto para train como para val) procedentes de DOTAv1.
- Clases: Hereda las 15 categorías de DOTAv1, como avión, barco y vehículo grande.
- Etiquetas: Cajas delimitadoras orientadas en formato YOLO, guardadas en archivos
.txtjunto a cada imagen. - Descarga: 34 MB, descargados automáticamente de los recursos de Ultralytics en GitHub la primera vez que entrenes.
Este conjunto de datos está diseñado para usarse con Ultralytics Platform y YOLO26.
YAML del conjunto de datos#
Se utiliza un archivo YAML para definir la configuración del conjunto de datos. Contiene información sobre las rutas, las clases y otros datos pertinentes del conjunto de datos. En el caso del conjunto de datos DOTA128, el archivo dota128.yaml se mantiene en https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/dota128.yaml.
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
# DOTA128 dataset (128 images from the DOTAv1 split) by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/obb/dota128
# Example usage: yolo train model=yolo26n-obb.pt data=dota128.yaml
# parent
# ├── ultralytics
# └── datasets
# └── dota128 ← downloads here (34 MB)
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: dota128 # dataset root dir
train: images/train # train images (relative to 'path') 128 images
val: images/train # val images (relative to 'path') 128 images
# Classes for DOTA 1.0
names:
0: plane
1: ship
2: storage tank
3: baseball diamond
4: tennis court
5: basketball court
6: ground track field
7: harbor
8: bridge
9: large vehicle
10: small vehicle
11: helicopter
12: roundabout
13: soccer ball field
14: swimming pool
# Download script/URL (optional)
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/dota128.zipUso#
Para entrenar un modelo YOLO26n-obb con el conjunto de datos DOTA128 durante 100 épocas y con un tamaño de imagen de 640, puedes usar los siguientes fragmentos de código. Para consultar una lista completa de los argumentos disponibles, visita la página de entrenamiento del modelo.
from ultralytics import YOLO
# Cargar un modelo
model = YOLO("yolo26n-obb.pt") # cargar un modelo preentrenado (recomendado para entrenar)
# Entrenar el modelo
results = model.train(data="dota128.yaml", epochs=100, imgsz=640)Imágenes de muestra y anotaciones#
Aquí tienes algunos ejemplos de imágenes del conjunto de datos DOTA128 junto con sus anotaciones correspondientes:
- Imagen en mosaico: Esta imagen muestra un lote de entrenamiento compuesto por imágenes del dataset dispuestas en mosaico. El mosaico es una técnica que se utiliza durante el entrenamiento y combina varias imágenes en una sola para aumentar la variedad de objetos y escenas de cada lote de entrenamiento. Esto ayuda a mejorar la capacidad del modelo para generalizar a distintos tamaños de objetos, relaciones de aspecto y contextos.
El ejemplo muestra la variedad y complejidad de las imágenes del conjunto de datos DOTA128, así como las ventajas de usar el mosaico durante el entrenamiento.
Citas y agradecimientos#
Si utilizas el conjunto de datos DOTA en tu trabajo de investigación o desarrollo, cita el siguiente artículo:
@article{9560031,
author={Ding, Jian and Xue, Nan and Xia, Gui-Song and Bai, Xiang and Yang, Wen and Yang, Michael and Belongie, Serge and Luo, Jiebo and Datcu, Mihai and Pelillo, Marcello and Zhang, Liangpei},
journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
title={Object Detection in Aerial Images: A Large-Scale Benchmark and Challenges},
year={2022},
volume={44},
number={11},
pages={7778-7796},
doi={10.1109/TPAMI.2021.3117983}
}Queremos expresar nuestro agradecimiento al equipo responsable de los conjuntos de datos DOTA por su encomiable labor de recopilación. Para conocer en profundidad el conjunto de datos y sus particularidades, visita el sitio web oficial de DOTA.
Preguntas frecuentes#
El conjunto de datos DOTA128 es un conjunto versátil para la detección de objetos orientados, formado por 128 imágenes del conjunto DOTAv1, todas almacenadas en la carpeta train. El entrenamiento y la validación utilizan el mismo conjunto de imágenes, por lo que es ideal para probar y depurar rápidamente modelos OBB como Ultralytics YOLO26. Gracias a su tamaño manejable y su diversidad, ayuda a identificar errores en los flujos de trabajo y a realizar comprobaciones de coherencia antes de entrenar con conjuntos de datos más grandes. Descubre más sobre la detección OBB con Ultralytics YOLO26.
Para entrenar un modelo YOLO26n-obb con el conjunto de datos DOTA128 durante 100 épocas y con un tamaño de imagen de 640, puedes usar los siguientes fragmentos de código. Para consultar las opciones de argumentos disponibles, visita la página de entrenamiento del modelo.
Ejemplo de entrenamientofrom ultralytics import YOLO # Cargar un modelo model = YOLO("yolo26n-obb.pt") # cargar un modelo preentrenado (recomendado para entrenar) # Entrenar el modelo results = model.train(data="dota128.yaml", epochs=100, imgsz=640)El conjunto de datos DOTA es conocido por su benchmark a gran escala y por los retos que plantea la detección de objetos en imágenes aéreas. El subconjunto DOTA128 ofrece más diversidad que DOTA8 y sigue siendo manejable para las pruebas iniciales. Puedes acceder al archivo
dota128.yaml, que contiene rutas, clases y detalles de configuración, en este enlace de GitHub.DOTA128 (128 imágenes) se sitúa entre DOTA8 (8 imágenes) y el conjunto de datos completo DOTA-v1 (1.869 imágenes etiquetadas de entrenamiento y validación) en cuanto a tamaño:
- DOTA8: Contiene solo 8 imágenes (4 de train y 4 de val); es ideal para pruebas rápidas y depuración.
- DOTA128: Contiene 128 imágenes (todas en la carpeta train, utilizadas tanto para train como para val); ofrece un equilibrio entre tamaño y diversidad.
- DOTA-v1 completo: Contiene 1.869 imágenes etiquetadas de entrenamiento y validación; es exhaustivo, pero requiere muchos recursos.
DOTA128 ofrece un término medio adecuado: aporta más diversidad que DOTA8 y, a la vez, es mucho más manejable que el conjunto de datos DOTA completo para experimentar y desarrollar modelos iniciales.
El mosaico combina varias imágenes en una durante el entrenamiento, lo que aumenta la variedad de objetos y contextos de cada lote. Esto mejora la capacidad del modelo para generalizar a distintos tamaños de objetos, proporciones y escenas. Esta técnica se puede mostrar visualmente mediante un lote de entrenamiento compuesto por imágenes del conjunto de datos DOTA128 combinadas en mosaico, lo que ayuda a desarrollar modelos robustos. Descubre más sobre el mosaico y las técnicas de entrenamiento en nuestra página de entrenamiento.
Ultralytics YOLO26 ofrece capacidades de detección de objetos en tiempo real de última generación, con funciones como cajas delimitadoras orientadas (OBB), segmentación de instancias y un flujo de entrenamiento muy versátil. Es adecuado para diversas aplicaciones y ofrece modelos preentrenados para un ajuste fino eficiente. Descubre más sobre sus ventajas y uso en la documentación de Ultralytics YOLO26.