YOLOv8_obb训练流程-原理解析[旋转目标检测理论篇]

吃鱼不卡次

已于 2024-06-02 23:23:10 修改

阅读量1.3k

点赞数 5

分类专栏： YOLOv8深度剖析文章标签： YOLOv8 旋转目标检测 dist2rbox Rotated TAA Prob Loss

于 2024-06-02 23:06:44 首次发布

本文链接：https://blog.csdn.net/weixin_45144684/article/details/139399075

版权

YOLOv8深度剖析专栏收录该内容

23 篇文章 20 订阅 ¥99.90 ¥299.90

订阅专栏

本文详细介绍了YOLOv8_obb在旋转目标检测中的训练流程，包括标签转换、预测结果解码、RotatedTaskAlignedAssigner以及Loss计算。重点讲解了如何将四个点的标签转换为旋转角度、宽高和中心点，并探讨了预测结果解码过程中的dist2rbox方法和旋转中心点的确定。

摘要由CSDN通过智能技术生成

在旋转目标检测网络中，换了个顺序，先把训练流程捋一遍，然后再取捋一下测试的流程。由下图的YOLOv8l_obb网络结构图可以看到相对于目标检测网络，旋转目标检测网络只是在Head层不相同，在每个尺度特征层中增加了Angle分支（浅蓝色），通过两个卷积组和一个Conv卷积得到得到通道数为1的特征图，1表示的就是每个Grid cell预测的角度。

1.YOLOv8_obb标签转换

首先来认识一下v8中的标签格式，如下图所示，标注完经过转换后得到的标签一共有9列，其中第一列是类别，后面八列表示四个角点的X、Y坐标，这四个点不用按照顺序排列。然后对这四个点使用插值来补点，和实例分割标签处理的方法一样，只不过实例分割标签会把点补到1000个，旋转目标检测只需要补充到100个点；最后使用cv2.minAreaRect()对这100个轮廓点进行处理，得到旋转的中心点、宽高以及旋转角度。

这里要注意一下使用cv2.minAreaRect()生成的中心点、宽高、角度的规律：（1）首先生成的角度范围是[0,90];（2）由下图可知，角度由两种表示方式：其一是在图像坐标系中（原点在左上角），角度是指平行于x轴的直线顺时针旋转直到碰到第一条边得到的度数，W即为碰到的第一条边的长度，H为另一条

了解本专栏