Yolov8 obb旋转目标检测训练自己的数据集

原文：Yolov8 obb旋转目标检测训练自己的数据集 - 知乎 (zhihu.com)

yolov8继续增加旋转目标检测

Detect:

YOLOV8血细胞检测-CSDN博客

Segment:

YOLOV8花朵实例分割实战_yolov8分割任务result.plot-CSDN博客

Classify:

yolov8 分类太阳能板-CSDN博客

Pose:

Yolov8 姿态估计-CSDN博客

Track:

YOLOv8目标跟踪deepsort-CSDN博客

YOLOv8_ ByteTrack目标跟踪、模型部署-CSDN博客

obb数据样例：

模型训练：

from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n-obb.yaml')  # build a new model from YAML
model = YOLO('yolov8n-obb.pt')  # load a pretrained model (recommended for training)
model = YOLO('yolov8n-obb.yaml').load('yolov8n.pt')  # build from YAML and transfer weights

# Train the model
results = model.train(data='dota8-obb.yaml', epochs=100, imgsz=640)

模型测试：

from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n-obb.pt')  # load an official model
#model = YOLO('path/to/best.pt')  # load a custom model

# Predict with the model
results = model('https://ultralytics.com/images/bus.jpg')  # predict on an image

模型导出onnx:

from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n-obb.pt')  # load an official model
model = YOLO('path/to/best.pt')  # load a custom trained model

# Export the model
model.export(format='onnx')

训练自己的数据集

DroneVenicle数据集

DroneVenicle训练集下载地址：https://pan.baidu.com/s/1ptZCJ1mKYqFnMnsgqEyoGg（密码：ngar）

DroneVenicle验证集下载地址：https://pan.baidu.com/s/1e6e9mESZecpME4IEdU8t3Q（密码：jnj6）

DroneVenicle测试集下载地址：