You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何用旋转边界框训练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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.08 22:28:09