学习python使用Ultralytics的YOLO26进行旋转框检测的基本用法

  Ultralytics的YOLO26模型提供专门的旋转框检测模型以支持旋转边界框检测,其主要预训练模型及参数如下表所示,其中的mAPtest50-95(e2e)指标指IoU从0.5到0.95多个阈值下的平均精度,用于评估评估模型在高难度、高精度需求下的性能,而mAPtest50(e2e)指标指在IoU=0.5单一阈值下的平均精度,用于快速对比基线模型,或对定位精度要求不高的任务。

  定向目标检测器的输出是一组旋转的边界框,这些边界框精确地包围了图像中的目标,以及每个框的类别标签和置信度分数。当目标以各种角度出现时,定向边界框特别有用,例如在航空图像中,传统的轴对齐边界框可能包含不必要的背景。
  YOLO26的旋转框检测模型的输出数据形状为N,7(检测)或N,8(追踪),也即result.obb. data的数据形状,其中N为从图片中检测到的目标个数,第二个维度则是目标的位置及置信度信息。

  模型检测结果中的obb.xywhr属性保存检测到所有目标的边界框及旋转角度,obb.xyxyxyxy属性的数据形状为N,4,2,保存所有目标的边界框四个角的坐标。不过将旋转框检测模型转换为onnx格式后,其输出形状变为1,300,7,具体解析方式后续再学习。

   YOLO26的旋转框检测模型是在DOTAv1数据集上进行的预训练,分类只有14种,其使用示例、运行效果(测试图片来自百度图片)、数据形式如下所示:

python 复制代码
from ultralytics import YOLO

# Load a model
model = YOLO(r"E:\MyPrograms\Python\ultralytics\yolo26m-obb.pt")

# Predict with the model
results = model("ship.jpg")

# Access the results
for result in results:
    print(result.obb.data)
    print(result.obb.xywhr)
    print(result.obb.xyxyxyxy)
    print(result.names)
    result.show()


参考文献:

1https://docs.ultralytics.com/zh/models/yolo26/

2https://docs.ultralytics.com/zh/tasks/obb/

3https://deepwiki.com/ultralytics/ultralytics/5.2-annotation-and-plotting-utilities#obb

4https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/DOTAv1.yaml

相关推荐
qq_158895411111 天前
AI 音乐生成完成后,音频 URL 为什么会过期?一套可恢复的交付方案
python·api·ai音乐
Warson_L1 天前
高性能架构设计
python·go
databook1 天前
新数据到手,先跑哪几步检查
python·数据分析
czq_26867194871 天前
Python打卡第46天
开发语言·python
Flynt1 天前
日涨4000星的OSINT工具,我拿Linus的用户名和一个随机字符串实测了一遍
python·安全
曲径海盗1 天前
【Python 学习第 53 天】列表基础
python·学习笔记·编程入门
Dumbness1 天前
群里 @ 机器人发"查订单",指令一条都没命中:微信群机器人在路由前漏掉的几步归一化
python·微信
用户1614840955451 天前
用 M4 Pro 跑羽毛球视频 AI 分析(二):TrackNet MPS 加速 3.2 倍、卡尔曼滤波补漏与自动回合剪辑
python
用户4019925568611 天前
基于对话树回溯的社媒评论流买方意图识别与噪音过滤实践
python
用户0332126663671 天前
使用 Python 加密或解密 PDF 文档
python