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

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐