江大白 | 目标检测之旋转目标:YOLO11 与定向边界框(OBB)旋转检测!(附多种旋转检测模型)

本文来源公众号“江大白”,仅用于学术分享,侵权删,干货满满。

原文链接:目标检测之旋转目标:YOLO11 与定向边界框(OBB)旋转检测!(附多种旋转检测模型)

导读

面对旋转、倾斜等复杂目标,传统检测方法难以准确识别。OBB(定向边界框)技术应运而生,赋予AI识别“朝向”的能力,广泛应用于医疗影像、航空监控和农业自动化等领域,本文详解其原理与实践。

作为人类,我们天生具备识别物体的能力——无论物体如何排列或朝向哪个方向。例如,我们可以轻松辨认十字路口的汽车或港口中的船只,并判断它们的朝向。然而,对人工智能(AI)系统而言,这并非易事。

计算机视觉作为AI的一个分支,专注于理解和分析图像与视频。其中,定向边界框(Oriented Bounding Box,OBB)检测技术的出现,正在改变AI识别旋转和倾斜物体的方式。

像 Ultralytics YOLO11 这样的计算机视觉模型,原生支持OBB检测,为实时场景应用提供了可能,尤其适用于目标方向非常关键的任务,如空中监控。除此之外,OBB检测还广泛应用于医疗健康、农业和文档分析等领域。

本文将带你深入了解OBB检测的原理、工作方式以及它在现实世界中的实际应用场景。

图1.使用YOLO11对船只进行OBB检测的示例演示

什么是定向边界框(OBB)?

定向边界框(Oriented Bounding Box,简称OBB)是一种用于计算机视觉中的矩形框类型,用于在图像中表示被检测的目标。与标准边界框(只能与图像的水平和垂直轴对齐)不同,OBB可以根据目标的实际角度进行旋转。

这种可旋转的特性带来了多种优势。OBB能够更贴近目标的实际朝向,使边界框更紧密地包围物体的形状和方向,从而提高检测的精度和准确度。

OBB特别适用于目标并非垂直或水平放置的场景,例如:航拍视频中转弯的汽车、桌面上倾斜放置的书籍,或医学影像中旋转角度不一的肿瘤。通过更准确地匹配物体的角度,OBB不仅提升了检测效果,还减少了背景干扰,尤其适合那些对物体“朝向”和“位置”同等重要的应用场景。

图2.水平边界框和定向边界框

OBB检测 vs 传统目标检测

OBB检测和传统目标检测在外观上看似相似,但它们适用于不同的场景与需求。下面通过一个实际例子进行对比说明:

在工业检测等真实应用中,计算机视觉模型(如YOLO11)可用于检测和分类流水线上的各类零件。假设某工厂的装配线上,不同的机械零件在传送带上移动,有些摆放整齐,而另外一些可能因震动或速度影响而发生旋转、倾斜或重叠。

传统目标检测使用的是标准的矩形框,与图像的水平和垂直边对齐。当目标发生旋转时,矩形框无法准确贴合物体边缘,可能遗漏部分目标,或包含过多背景,导致检测精度降低,系统难以准确识别物体。

而使用OBB检测时,模型绘制的边界框可以旋转,从而匹配物体的实际角度。比如一个倾斜的齿轮或斜放的组件,OBB会以正确的方向紧密包围目标。这不仅提高了检测的精度,减少误差,还让识别结果更加可靠,特别适用于自动质检、机器人分拣等对准确度要求极高的场景。

主流OBB检测模型

许多先进的计算机视觉模型专门针对旋转或倾斜目标的检测进行了优化。其中,Ultralytics的YOLO系列模型以其可靠且高效的OBB检测能力尤为著名。

早期版本如Ultralytics YOLOv5主要用于传统的目标检测;而后续版本,比如Ultralytics YOLOv8以及最新的YOLO11,则原生支持OBB检测。特别是YOLO11,在保证高速推理的同时,实现了领先的准确率,适合实时应用场景。

预训练的YOLO11 OBB模型(如YOLO11n-obb)在DOTAv1数据集上训练,该数据集包含航空影像,标注了多类以不同角度和方向出现的目标,如飞机、船只和网球场等。

此外,YOLO11模型提供从nano(n-obb)到extra-large(x-obb)共五种规模,满足不同性能需求。这种多样性使其能够广泛应用于各行各业——从城市基础设施监测、机械设备检查,到识别扫描文档中倾斜的文字。

更多OBB开源模型

  • MMRotate:https://github.com/open-mmlab/mmrotate

  • OBBDetection:https://github.com/jbwang1997/OBBDetection

使用YOLO11进行有向边界框(OBB)检测的自定义训练

在许多现实场景中,待检测的物体往往与标准训练数据集中的目标完全不同。例如,产线上的工具、产品包装、或电路板上的元器件,可能存在旋转、不规则摆放或形状各异的情况。

当目标的朝向很关键时,为了实现精准检测,就需要使用自己的图像和标签对像YOLO11这样的模型进行自定义训练。以下是YOLO11用于OBB检测的训练步骤说明:

  1. 图像采集:收集展示目标物体的图像,确保涵盖各种角度、位置以及真实环境下的情况。

  2. 目标标注:使用支持OBB的标注工具,为每个目标打上旋转边界框(OBB),以记录其位置与朝向。

  3. 数据集准备:将图像与标签整理成YOLO所需的目录结构,并创建一个YAML配置文件,包含类别名称与数据路径。

  4. 模型训练:选择适合需求的YOLO11模型版本,启动训练过程,让模型从标注图像中学习。

  5. 评估与部署:使用新的图像对训练好的模型进行测试,评估其准确性,并将其部署到制造、空中监控或文档分析等实际应用中。

通过自定义训练,YOLO11模型可以精准适配各种角度、形状复杂、方向不定的目标,为实际项目带来更高的检测精度与可靠性。

OBB检测的三大应用场景

医疗影像分析

在X光片中,器官、骨骼、肿瘤等目标常呈现不同角度以及不规则形状,由于OBB可以旋转以匹配目标角度,因此在定位和测量上比传统的水平边界框更为准确,这对于诊断和治疗方案的制定至关重要。

例如,在用于分析儿科肘部X光片应用中,通过调整骨骼的方向,它有助于提高检测精度。

图3.X光片(a,d)目标检测(b,e)和定向边界框检测(c,f)。

航空监控

空中监视在公共安全、环境监测和城市规划等领域中是一项重要工具。无人机或卫星拍摄的图像可以用于识别诸如船只、车辆和建筑物等目标。然而,在这类图像中,目标通常较小,且具有不规则的角度,导致准确检测变得更加困难。

OBB检测通过旋转边界框以匹配目标的实际角度,有效解决了这一问题。这种方法能够更精确地测量目标的尺寸与朝向,从而为城市规划、国防、灾害响应和环境监测等领域提供更可靠的决策支持。

一个典型的应用案例是海事监视中的船只跟踪。由于天气、光照或运动等因素,卫星图像中捕捉到的船只往往存在角度各异、大小不一的情况。OBB能够适应这些变化,尤其在检测小型或部分遮挡的船只方面,表现出更高的准确性。

图4.OBB检测在海上监视的应用

农业自动化

针对特殊形状作物的分拣难题:

  • 传统方法:在处理如苹果、橙子等圆形水果时表现良好,但对于胡萝卜、茭白这类细长形作物,由于其形状多变且朝向不一,检测与准确分拣则更加困难。

  • OBB方案:能够在一张图像中检测多个作物,即使它们存在倾斜或重叠情况,也能实现实时的品质评估与位置识别,从而显著提升分拣效率和准确率。

OBB检测的优缺点分析

优点

  1. 提升下游任务性能

    • 对于实例分割、目标跟踪等计算机视觉任务,OBB检测提供了更精确的目标边界,从而提升整体处理效果。

  2. 增强空间理解能力

    • OBB能够捕捉目标的朝向角度,有助于系统更好地理解目标的排列方向与空间分布,提升空间推理能力。

  3. 减少拥挤场景中的重叠问题

    • 在目标密集或复杂背景中,OBB能更紧密贴合目标边界,从而减少边界重叠和检测歧义,提高区分能力。

缺点

  1. 对噪声更敏感

    • 由于OBB依赖角度预测,轻微的角度误差在处理细长或密集目标时可能导致显著的检测偏差。

  2. 需依赖专用工具

    • 并非所有标注和训练平台原生支持OBB,因此使用该方法可能需要额外的软件工具或自定义设置。标注工具:https://github.com/heshameraqi/labelImg_OBB 

  3. 数据集资源有限

    • 相较于标准目标检测,公开可用的OBB标注数据集数量较少,增加了模型训练和性能对比的难度。

总结

OBB检测使计算机视觉系统能够更轻松地识别那些不完全笔直或不规则排列的目标。通过同时捕捉目标的位置和朝向,OBB检测在医学影像扫描、农田监测、卫星影像分析等实际应用中显著提升了检测准确率。

随着像YOLO11这样的模型使OBB检测变得更加易用,OBB正逐渐成为多个行业的实用选择。无论面对倾斜、重叠或形状奇特的目标,OBB检测都能提供标准方法难以达到的额外精度保障。

如需实践OBB检测,可参考Ultralytics官方GitHub仓库,其中提供了完整的训练教程和预训练模型。

  • 官网:https://github.com/ultralytics/ultralytics

参考文献:https://www.ultralytics.com/blog/what-is-oriented-bounding-box-obb-detection-a-quick-guide,内容有改动

THE END !

文章结束,感谢阅读。您的点赞,收藏,评论是我继续更新的动力。大家有推荐的公众号可以评论区留言,共同学习,一起进步。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值