准备DOTA格式数据集:
dota_dataset
-- images
|----- train
|----- val
-- labels
|----- train
|----- train_original
|----- val
|----- val_original
修改class_mapping和图片格式:
ultralytics/data/converter.py
convert_dota_to_yolo_obb()
转换标注:
python
from ultralytics.data.converter import convert_dota_to_yolo_obb
convert_dota_to_yolo_obb('path\to\dota_dataset')
参考:[1]