DOTAv2.yaml 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. # Ultralytics YOLO 🚀, AGPL-3.0 license
  2. # DOTA 2.0 dataset https://captain-whu.github.io/DOTA/index.html for object detection in aerial images by Wuhan University
  3. # Example usage: yolo train model=yolov8n-obb.pt data=DOTAv2.yaml
  4. # parent
  5. # ├── ultralytics
  6. # └── datasets
  7. # └── dota2 ← downloads here (2GB)
  8. # 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, ..]
  9. path: ../datasets/DOTAv2 # dataset root dir
  10. train: images/train # train images (relative to 'path') 1411 images
  11. val: images/val # val images (relative to 'path') 458 images
  12. test: images/test # test images (optional) 937 images
  13. # Classes for DOTA 2.0
  14. names:
  15. 0: plane
  16. 1: ship
  17. 2: storage tank
  18. 3: baseball diamond
  19. 4: tennis court
  20. 5: basketball court
  21. 6: ground track field
  22. 7: harbor
  23. 8: bridge
  24. 9: large vehicle
  25. 10: small vehicle
  26. 11: helicopter
  27. 12: roundabout
  28. 13: soccer ball field
  29. 14: swimming pool
  30. 15: container crane
  31. 16: airport
  32. 17: helipad
  33. # Download script/URL (optional)
  34. download: https://github.com/ultralytics/yolov5/releases/download/v1.0/DOTAv2.zip