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+

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐