Yolov5旋转框(斜框)检测自己的数据集,附带代码+数据集(模型可以收敛)

文章目录

    • 1. 制作数据集
      • 1.1 标注数据集
      • 1.2标签转换
      • 1.3 数据集划分
    • 2. 环境搭建
      • 1.安装nms_rotated
      • 2.安装DOTA_devkit
    • 3. 代码讲解
      • 3.1坐标表示
      • 3.2 损失函数
    • 4.训练+测试
    • 链接

后面附上百度网盘链接,内部包含数据集。

下一篇介绍tensorRT部署yolov5-obb

1. 制作数据集

​ 标注软件为rolabelimg,标注方法:https://blog.csdn.net/jiao_mrswang/article/details/89216431

1.1 标注数据集

rolabelimg标注出的数据类型如下

你可能感兴趣的:(yolo系列改进计划,YOLO)