如何用旋转边界框训练YOLOR?标注导出相关问题咨询
旋转边界框标注与YOLOR适配问题解答
1. 如何在Label Studio导出时保留旋转角度参数
标准YOLO格式(class_id cx cy w h)本身不包含旋转角度字段,直接导出会丢失theta参数。要保留角度,需自定义Label Studio的导出逻辑:
- 在导出界面选择「Custom (JSON)」或自定义导出脚本,提取标注数据里的
rotation字段(Label Studio存储旋转角度的字段)。 - 将导出的txt格式调整为旋转YOLO(OBB)格式:每行内容改为
class_id center_x center_y width height theta,注意theta的单位(弧度/角度)要和后续模型要求统一。
2. YOLOR默认是否识别旋转角度参数
默认的YOLOR代码仅支持轴对齐边界框(AABB),不会读取或处理旋转角度参数。若要让它支持旋转目标检测,必须对核心代码做针对性修改:
- 修改数据集加载模块:新增对标注txt中theta字段的解析逻辑。
- 调整模型输出结构:在检测头部分新增角度预测分支,输出theta值。
- 替换损失函数:将原本适配AABB的IoU损失,替换为旋转框专用的损失(如旋转GIoU、OBB IoU损失等)。
内容的提问来源于stack exchange,提问作者marco
相关产品推荐
相关产品推荐

