Vai al contenuto

Dataset DOTA8

Introduzione

Ultralytics DOTA8 è un piccolo ma versatile dataset orientato di object detection composto dalle prime 8 immagini del set DOTAv1 suddiviso, 4 per il training e 4 per la validation. Questo dataset è ideale per testare ed eseguire il debug di modelli di object detection o per sperimentare nuovi approcci di detection. Con 8 immagini, è abbastanza piccolo da essere facilmente gestibile, ma sufficientemente diversificato da testare le pipeline di training per individuare errori e fungere da controllo di integrità prima di addestrare dataset più grandi.

Questo dataset è destinato all'uso con Ultralytics HUB e YOLO11.

YAML del set di dati

Un file YAML (Yet Another Markup Language) viene utilizzato per definire la configurazione del dataset. Contiene informazioni sui percorsi, le classi e altre informazioni rilevanti del dataset. Nel caso del dataset DOTA8, il dota8.yaml file è mantenuto in https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/dota8.yaml.

ultralytics/cfg/datasets/dota8.yaml

# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license

# DOTA8 dataset 8 images from split DOTAv1 dataset by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/obb/dota8/
# Example usage: yolo train model=yolov8n-obb.pt data=dota8.yaml
# parent
# ├── ultralytics
# └── datasets
#     └── dota8 ← downloads here (1MB)

# 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: dota8 # dataset root dir
train: images/train # train images (relative to 'path') 4 images
val: images/val # val images (relative to 'path') 4 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/dota8.zip

Utilizzo

Per addestrare un modello YOLO11n-obb sul dataset DOTA8 per 100 epoche con una dimensione dell'immagine di 640, puoi utilizzare i seguenti frammenti di codice. Per un elenco completo degli argomenti disponibili, consulta la pagina di addestramento del modello.

Esempio di addestramento

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n-obb.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="dota8.yaml", epochs=100, imgsz=640)
# Start training from a pretrained *.pt model
yolo obb train data=dota8.yaml model=yolo11n-obb.pt epochs=100 imgsz=640

Immagini di esempio e annotazioni

Ecco alcuni esempi di immagini del dataset DOTA8, insieme alle relative annotazioni:

Immagine di esempio del set di dati

  • Immagine mosaicata: questa immagine mostra un batch di addestramento composto da immagini del dataset mosaicato. Il mosaicing è una tecnica utilizzata durante l'addestramento che combina più immagini in un'unica immagine per aumentare la varietà di oggetti e scene all'interno di ogni batch di addestramento. Ciò contribuisce a migliorare la capacità del modello di generalizzare a diverse dimensioni degli oggetti, proporzioni e contesti.

L'esempio mostra la varietà e la complessità delle immagini nel dataset DOTA8 e i vantaggi dell'utilizzo del mosaicing durante il processo di training.

Citazioni e ringraziamenti

Se utilizzi il dataset DOTA nel tuo lavoro di ricerca o sviluppo, cita il seguente articolo:

@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={2021},
  volume={},
  number={},
  pages={1-1},
  doi={10.1109/TPAMI.2021.3117983}
}

Un ringraziamento speciale al team dietro i dataset DOTA per il loro lodevole impegno nella cura di questo dataset. Per una comprensione esaustiva del dataset e delle sue sfumature, visitare il sito web ufficiale di DOTA.

FAQ

Cos'è il dataset DOTA8 e come può essere utilizzato?

Il dataset DOTA8 è un piccolo e versatile dataset per il rilevamento di oggetti orientati, composto dalle prime 8 immagini del set diviso DOTAv1, con 4 immagini designate per il training e 4 per la convalida. È ideale per testare ed eseguire il debug di modelli di rilevamento oggetti come Ultralytics YOLO11. Grazie alle sue dimensioni gestibili e alla sua diversità, aiuta a identificare gli errori della pipeline ed eseguire controlli di integrità prima di implementare dataset più grandi. Ulteriori informazioni sul rilevamento di oggetti con Ultralytics YOLO11.

Come posso addestrare un modello YOLO11 utilizzando il dataset DOTA8?

Per addestrare un modello YOLO11n-obb sul dataset DOTA8 per 100 epoche con una dimensione dell'immagine di 640, puoi utilizzare i seguenti frammenti di codice. Per opzioni di argomenti complete, consulta la pagina di addestramento del modello.

Esempio di addestramento

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n-obb.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="dota8.yaml", epochs=100, imgsz=640)
# Start training from a pretrained *.pt model
yolo obb train data=dota8.yaml model=yolo11n-obb.pt epochs=100 imgsz=640

Quali sono le caratteristiche principali del dataset DOTA e dove posso accedere al file YAML?

Il dataset DOTA è noto per il suo benchmark su larga scala e le sfide che presenta per il rilevamento di oggetti in immagini aeree. Il sottoinsieme DOTA8 è un dataset più piccolo e gestibile, ideale per i test iniziali. È possibile accedere al dota8.yaml file, che contiene percorsi, classi e dettagli di configurazione, a questo Link GitHub.

In che modo il mosaicing migliora il training del modello con il dataset DOTA8?

Il mosaicing combina più immagini in una sola durante l'addestramento, aumentando la varietà di oggetti e contesti all'interno di ogni batch. Ciò migliora la capacità di un modello di generalizzare a diverse dimensioni degli oggetti, proporzioni e scene. Questa tecnica può essere dimostrata visivamente attraverso un batch di addestramento composto da immagini del dataset DOTA8 in mosaico, aiutando nello sviluppo di un modello robusto. Scopri di più sul mosaicing e sulle tecniche di addestramento nella nostra pagina Addestramento.

Perché dovrei usare Ultralytics YOLO11 per attività di rilevamento oggetti?

Ultralytics YOLO11 offre funzionalità di object detection in tempo reale all'avanguardia, tra cui caratteristiche come oriented bounding boxes (OBB), segmentazione delle istanze e una pipeline di training estremamente versatile. È adatto a diverse applicazioni e offre modelli pre-addestrati per una messa a punto efficiente. Scopri di più sui vantaggi e sull'utilizzo nella documentazione di Ultralytics YOLO11.



📅 Creato 1 anno fa ✏️ Aggiornato 5 mesi fa

Commenti