绿色检查
链接复制到剪贴板

如何使用Ultralytics YOLO11 检测 OBB 对象

了解Ultralytics YOLO11 如何利用定向边界框 (OBB) 增强物体检测,以及这项计算机视觉任务的理想应用。

UltralyticsYOLO Vision 2024 (YV24),重点讨论人工智能和计算机视觉领域最新突破。这是介绍我们最新机型的绝佳机会、 Ultralytics YOLO11.该模型支持的计算机视觉任务Ultralytics YOLOv8的计算机视觉任务,使用户可以毫不费力地转向新模型。

假设您正在使用YOLOv8 进行定向包围盒 (OBB) 物体检测,以从不同角度检测物体。现在,您只需对代码稍作改动,就可以切换到YOLO11 ,并从YOLO11的改进中获益,包括提高准确性、效率和处理速度。如果您还没有使用过 YOLO11 这样的模型,OBB 检测就是一个很好的例子,说明 YOLO11 可以应用于各行各业,提供实用的解决方案,产生真正的影响。

在本文中,我们将介绍什么是 OBB 物体检测、它可以应用于哪些领域以及如何使用YOLO11 检测 OBB。我们还将介绍YOLO11 的新功能如何改进这些流程,以及如何运行推论和训练自定义模型,以充分利用其 OBB 检测功能。

下一代YOLO11 功能包括 OBB 对象检测等任务

OBB 物体检测通过检测不同角度的物体,在传统物体检测的基础上更进一步。与保持与图像轴线对齐的常规边界框不同,OBB 可根据物体的方向进行旋转。OBB 物体检测可用于分析航空或卫星图像,因为这些图像中的物体并不总是直的。在城市规划、能源和交通等行业中,准确检测建筑物、车辆或基础设施等倾斜物体的能力可为计算机视觉应用奠定基础,并带来切实的好处。 

图 1.法线包围盒与定向包围盒的比较。

YOLO11 YOLO11 支持 OBB 检测,并在DOTA v1.0 数据集上进行了训练,可从不同角度检测飞机、船只和储油罐等物体。该软件有多种模型可供选择,以满足不同需求,包括 YOLO11n-obb (纳米)、YOLO11s-obb(小型)、YOLO11m-obb(中型)、YOLO11l-obb(大型)和 YOLO11x-obb(超大型)。每个型号都有不同的尺寸,速度、精确度和计算能力也各不相同。用户可以根据自己的应用选择在速度和精确度之间取得适当平衡的型号。 

YOLO11 为检测引入新角度的用例

YOLO11的对象检测功能,尤其是其对定向边界框的支持,为各行各业带来了更高的精度。接下来,我们将举例说明YOLO11 和 OBB 检测如何在实际应用中使不同领域的流程更高效、更精确、更易于管理。

城市规划和基础设施监测YOLO11

如果您曾经欣赏过一座城市的设计和布局,那要归功于城市规划和基础设施监控的细致工作。基础设施监测的众多方面之一就是识别和管理储罐、管道和工业场地等重要结构。YOLO11 可以帮助城市规划者分析航空图像,快速准确地检测这些关键部件。 

定向边界框对象检测在此特别有用,因为它可以检测从不同角度观察的对象(航空图像通常如此)。精确度对于跟踪工业区、管理环境影响以及确保基础设施得到妥善维护至关重要。OBB 使检测过程更加可靠,有助于规划人员就城市的发展、安全可持续性做出明智的决策。使用YOLO11 ,可以帮助规划者监控和管理保持城市平稳运行的基础设施。

图 2.使用YOLO11 在航拍镜头中探测储油罐。

利用无人机、YOLO11 和边缘人工智能检查太阳能电池板

随着可再生能源和太阳能发电场等创新技术的普及,定期检查变得越来越重要。需要对太阳能电池板进行检查,以确保其有效工作。随着时间的推移,裂缝、污垢堆积或偏差等问题都会降低它们的性能。定期检查有助于及早发现这些问题,以便进行维护,使其保持平稳运行。

例如,可以使用集成了边缘人工智能和YOLO11 的无人机检查太阳能电池板的损坏情况。边缘图像分析为检查过程带来了更高的精度和效率。由于无人机的移动和视角,监控镜头可能经常从不同角度拍摄太阳能电池板。在这种情况下,YOLO11的 OBB 检测可以帮助无人机准确识别太阳能电池板。 

YOLO11 可为车队管理提供洞察力

港口和码头每周要处理数百艘船只,管理如此庞大的船队是一项挑战。在分析航拍图像中的船只时还会遇到额外的困难,因为船只通常会以不同的角度出现。这时,YOLO11的 OBB 检测支持就派上了用场。 

与标准矩形框相比,OBB 检测使模型能够更准确地检测到各种角度的船只。通过使用带有 OBB 的YOLO11 ,航运公司可以更轻松地识别其船队的位置和状况,跟踪船队移动和供应链物流等重要细节。这种支持视觉的解决方案有助于优化航线、减少延误并改善航运路线上的整体船队管理。

图 3.使用YOLO11 以一定角度探测船只和港口。

YOLO11 人工智能开发人员:检测定向边框

如果你是一名人工智能开发人员,希望使用YOLO11 进行 OBB 检测,有两种简单的方法可以入门。如果您喜欢使用代码,Ultralytics Python 软件包是一个不错的选择。如果您更喜欢用户友好、无需代码、具有云培训功能的解决方案,Ultralytics HUB 是专门为此设计的内部平台。欲了解更多详情,请参阅我们关于使用Ultralytics HUB培训和部署Ultralytics YOLO11 的指南。

现在,我们已经看到了YOLO11OBB 支持的应用实例,让我们来探索一下Ultralytics Python 软件包,看看如何使用它运行推论训练自定义模型。 

使用YOLO11

首先,要将YOLO11 与Python 结合使用,需要安装Ultralytics 软件包。根据自己的偏好,可以选择使用 pip、conda 或 Docker 安装。有关分步说明,请参阅我们的Ultralytics 安装指南。如果在安装过程中遇到任何问题,我们的《常见问题指南》提供了有用的故障排除技巧。

安装Ultralytics 软件包后,使用YOLO11 就变得非常简单。运行推理指的是使用训练有素的模型对新图像进行预测的过程,比如实时使用 OBB 检测物体。它不同于模型训练,后者是指您教模型识别新物体或提高其在特定任务中的性能。当您想将模型应用于未见数据时,就需要使用推理。

下面的示例将教你如何加载模型并用它来预测图像上的定向包围盒。有关更详细的示例和高级使用技巧,请务必查看Ultralytics 官方 文档,了解最佳实践和进一步说明。

图 4.使用YOLO11 运行推理的代码片段。

培训自定义YOLO11 模型

训练YOLO11 模型意味着您可以微调其在特定数据集任务上的性能,例如定向边界框对象检测。虽然像YOLO11 这样的预训练模型可以用于一般的物体检测,但当你需要模型来检测独特的物体或优化特定数据集的性能时,训练自定义模型就显得非常重要了。

在下面的代码片段中,我们介绍了训练YOLO11 模型进行 OBB 检测的步骤。 

首先,使用预先训练好的YOLO11 OBB 特定权重(yolo11n-obb.pt)对模型进行初始化。然后,使用训练函数在自定义数据集上训练模型,参数包括数据集配置文件、训练周期数、训练图像大小以及运行训练的硬件(如CPU 或GPU )。训练完成后,对模型的性能进行验证,检查准确率和损失等指标。 

使用训练有素的模型,您可以在新图像上运行推理,以检测带有 OBB 的物体,并将其可视化。此外,还可以使用导出功能将训练有素的模型转换为 ONNX 等格式,以便使用导出功能进行部署。

图 5.用于 OBB 物体检测的训练YOLO11 示例。

YOLO11 人工智能进步的未来之路

Ultralytics YOLO11 通过对定向边界框的支持,它将物体检测提升到了一个新的水平。YOLO11 能够检测不同角度的物体,因此可用于不同行业的各种应用。例如,它非常适合城市规划、能源和航运等行业,在这些行业中,精确度对于太阳能电池板检测或车队监控等任务至关重要。凭借更快的性能和更高的精度,YOLO11 可以帮助人工智能开发人员解决现实世界中的难题。 

随着人工智能越来越广泛地被采用并融入我们的日常生活,像YOLO11 这样的模型将塑造人工智能解决方案的未来。

要了解 Vision AI 的潜力,请加入我们不断壮大的社区,并浏览我们的GitHub 存储库。我们通过重塑医疗保健自动驾驶汽车等行业来推动创新。

Facebook 徽标Twitter 徽标LinkedIn 徽标复制链接符号

在此类别中阅读更多内容

让我们共同打造人工智能的未来

开始您的未来机器学习之旅