from ultralytics import YOLO
model = YOLO('/kaggle/input/model-v11-obb/yolo11n-obb.pt')
model.train(data='/kaggle/input/hrscobb4/HRSC-YOLO/data.yaml', epochs=100)
1使用的训练平台为Kaggle
数据集连接:一瓶可乐
数据集:HRSC的三种形式
一级分类:船
有水平框版本,旋转框版本。
二级分类:航母、军舰、商船、潜艇四大类
主要制作了旋转框版本
最终结果:
可以看到训练100轮后总的mp50可以达到60+