Ultralytics YOLO27:
Get Started

使用 Ultralytics YOLO 进行目标检测#

Ultralytics YOLO object detection with bounding boxes

目标检测是一项用于识别图像或视频流中目标位置和类别的任务。

目标检测器的输出是一组包围图像中目标的边界框,以及每个框对应的类别标签和置信度分数。如果你需要识别场景中感兴趣的目标,但不需要了解目标的确切形状,那么目标检测就是合适的选择。



观看: 使用预训练的 Ultralytics YOLO 模型进行目标检测。
提示

YOLO26 Detect 模型是默认的 YOLO26 模型,即 yolo26n.pt,并且是在 COCO 上预训练的。

模型#

以下展示了在 COCO 数据集上预训练的 YOLO26 Detect 模型。

首次使用时,模型会自动从 Ultralytics 最新版本下载。

模型尺寸
(像素)
mAP验证集
50-95
mAP验证
50-95(端到端)
速度
CPU ONNX
(毫秒)
速度
T4 TensorRT10
(ms)
参数
(M)
FLOPs
(B)
YOLO26n64040.940.138.9 ± 0.71.7 ± 0.02.45.5
YOLO26s64048.647.887.2 ± 0.92.5 ± 0.09.520.9
YOLO26m64053.152.5220.0 ± 1.44.7 ± 0.120.468.4
YOLO26l64055.054.4286.2 ± 2.06.2 ± 0.224.886.8
YOLO26x64057.556.9525.8 ± 4.011.8 ± 0.255.7194.4
  • mAPval 数值是在 COCO val2017 数据集上以单模型、单尺度方式测得的。
    使用 yolo detect val data=coco.yaml device=0 复现。添加 nms=False 可复现 e2e 列。
  • 速度是在 COCO val 图像上使用 ONNX(CPU)和 TensorRT10(NVIDIA T4 GPU)测得的平均值。
    使用 yolo detect val data=coco.yaml batch=1 device=0|cpu nms=False 复现
  • 参数量和 FLOPs 数值对应融合后的模型,该模型经过 Conv/BatchNorm 折叠并移除了未使用的检测分支。预训练检查点保留完整的训练架构,因此显示的数值可能更高。

查看尚未发布的 YOLO27 预览版,了解初步的 COCO 检测结果。

训练#

在 COCO8 数据集上以 640 的图像尺寸训练 YOLO26n,训练 100 个轮次。如需查看所有可用参数,请参阅配置页面。

示例
from ultralytics import YOLO

# 加载模型
model = YOLO("yolo26n.yaml")  # 从 YAML 构建新模型
model = YOLO("yolo26n.pt")  # 加载预训练模型(推荐用于训练)
model = YOLO("yolo26n.yaml").load("yolo26n.pt")  # 从 YAML 构建模型并迁移权重

# 训练模型
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)

请参阅训练页面,了解完整的 train 模式详情。检测模型也可以通过 Ultralytics Platform 云训练进行训练。

数据集格式#

数据集指南详细介绍了 YOLO 检测数据集格式。若要将现有的 COCO JSON 数据集转换为 YOLO 格式,请使用COCO 转 YOLO 指南中介绍的内置 convert_coco 函数。你也可以使用 Ultralytics Platform 的 AI 辅助标注工具标注和管理检测数据集。

验证#

验证训练好的 YOLO26n 模型精度。无需提供参数,因为 model 会将训练时的 data 及参数保留为模型属性:训练示例中的 path/to/best.pt 会在 COCO8 上进行验证。官方权重记录的训练数据集路径在你的机器上不存在,因此会发出警告并回退到任务默认的 coco8.yaml。传入 data 可在其他数据集上进行验证。

示例
from ultralytics import YOLO

# 加载模型
model = YOLO("yolo26n.pt")  # 加载官方模型
model = YOLO("path/to/best.pt")  # 加载自定义模型

# 验证模型
metrics = model.val()  # 无需参数,数据集和设置会自动记住
metrics.box.map  # map50-95
metrics.box.map50  # map50
metrics.box.map75  # map75
metrics.box.maps  # 包含每个类别 mAP50-95 的列表
metrics.box.image_metrics  # 包含精确率、召回率、F1、TP、FP 和 FN 的逐图像指标字典

预测#

使用训练好的 YOLO26n 模型对图像进行预测。

示例
from ultralytics import YOLO

# 加载模型
model = YOLO("yolo26n.pt")  # 加载官方模型
model = YOLO("path/to/best.pt")  # 加载自定义模型

# 使用模型进行预测
results = model("https://ultralytics.com/images/bus.jpg")  # 对图像进行预测

# 访问结果
for result in results:
    xywh = result.boxes.xywh  # 中心点 x、中心点 y、宽度、高度
    xywhn = result.boxes.xywhn  # normalized
    xyxy = result.boxes.xyxy  # 左上角 x、左上角 y、右下角 x、右下角 y
    xyxyn = result.boxes.xyxyn  # normalized
    names = [result.names[cls.item()] for cls in result.boxes.cls.int()]  # 每个框的类别名称
    confs = result.boxes.conf  # 每个框的置信度分数

请参阅预测页面,了解完整的 predict 模式详情。

结果输出#

目标检测会为每张图像返回一个 Results 对象。主要的预测字段是 result.boxes,其中包含 每个检测目标的边界框坐标、类别 ID 和置信度分数。

属性类型形状说明
result.boxesBoxes(N)检测框。
result.boxes.datatorch.float32(N,6/7)原始 [x1,y1,x2,y2,conf,cls],以及可选的跟踪 ID。
result.boxes.xyxytorch.float32(N,4)xyxy 像素框。
result.boxes.conftorch.float32(N,)置信度分数。
result.boxes.clstorch.float32(N,)类别 ID;转换为 int 可获取类别名称。

如需了解所有任务中特定于任务的 Results 字段,请参阅按任务查看预测结果部分。

导出#

将 YOLO26n 模型导出为 ONNX、CoreML 等其他格式。

示例
from ultralytics import YOLO

# 加载模型
model = YOLO("yolo26n.pt")  # 加载官方模型
model = YOLO("path/to/best.pt")  # 加载自定义模型

# 导出模型
model.export(format="onnx")

下表列出了可用的 YOLO26 导出格式。你可以使用 format 参数导出为任意格式,例如 format='onnx' 或 format='engine'。你可以直接使用导出的模型进行预测或验证,例如 yolo predict model=yolo26n.onnx。导出完成后,系统会显示适用于你模型的用法示例。

格式format 参数模型元数据参数
PyTorch-yolo26n.pt✅-
TorchScripttorchscriptyolo26n.torchscript✅imgsz, quantize, dynamic, nms, batch, device
ONNXonnxyolo26n.onnx✅imgsz, quantize, dynamic, simplify, opset, nms, batch, data, fraction, device
OpenVINOopenvinoyolo26n_openvino_model/✅imgsz, quantize, dynamic, nms, batch, data, fraction, device
TensorRTengineyolo26n.engine✅imgsz, quantize, dynamic, simplify, opset, workspace, nms, batch, data, fraction, device
CoreMLcoremlyolo26n.mlpackage✅imgsz, dynamic, quantize, nms, batch, device
Apple Core AIcoreaiyolo26n.aimodel✅imgsz, batch, quantize
TF SavedModelsaved_modelyolo26n_saved_model/✅imgsz, quantize, opset, nms, batch, data, fraction, device
TF GraphDefpbyolo26n.pb❌imgsz, opset, batch, device
TF Edge TPUedgetpuyolo26n_edgetpu.tflite✅imgsz, quantize, opset, data, fraction, device
LiteRTlitertyolo26n.tflite✅imgsz, quantize, batch, data, fraction, device
PaddlePaddlepaddleyolo26n_paddle_model/✅imgsz, batch, device
MNNmnnyolo26n.mnn✅imgsz, batch, dynamic, quantize, simplify, opset, nms, device
NCNNncnnyolo26n_ncnn_model/✅imgsz, quantize, batch, device
IMX500imxyolo26n_imx_model/✅imgsz, quantize, data, fraction, nms, device
RKNNrknnyolo26n_rknn_model/✅imgsz, batch, name, quantize, simplify, opset, data, fraction, device
ExecuTorchexecutorchyolo26n_executorch_model/✅imgsz, batch, device
Axeleraaxelerayolo26n_axelera_model/✅imgsz, batch, quantize, data, fraction, device
DEEPXdeepxyolo26n_deepx_model/✅imgsz, quantize, simplify, opset, data, optimize, device
Qualcomm QNNqnnyolo26n_qnn.onnx✅imgsz, batch, name, quantize, simplify, opset, data, fraction, device
Hailohailoyolo26n_hailo_model/✅imgsz, name, quantize, data, fraction, simplify, conf, iou, device
Huawei Ascendascendyolo26n_ascend_model/✅imgsz, batch, name, quantize, opset, simplify, nms, device
AMD Xilinxxilinxyolo26n_xilinx_model/✅imgsz, name, quantize, data, fraction, opset, simplify, device

nms=None 默认输出原始结果,以供外部 NMS 使用。设置 nms=False 可选择可用的无 NMS 检测头;不支持的格式会回退到其原生输出路径。上方的 nms 条目表示可以通过 nms=True 嵌入 NMS 的格式。

请参阅 Export 页面,了解完整的 export 详情。

常见问题#

  • 可以。Ultralytics Platform 快速入门介绍了基于浏览器的工作流程,你可以通过它标注数据集、在云端 GPU 上训练检测模型,并将模型部署到推理端点。

  • 在自定义数据集上训练 YOLO26 模型需要几个步骤:

    1. 准备数据集:确保数据集采用 YOLO 格式。请参阅我们的数据集指南了解相关说明。
    2. 加载模型:使用 Ultralytics YOLO 库加载预训练模型,或根据 YAML 文件创建新模型。
    3. 训练模型:在 Python 中调用 train 方法,或在 CLI 中执行 yolo detect train 命令。
    示例
    from ultralytics import YOLO
    
    # 加载预训练模型
    model = YOLO("yolo26n.pt")
    
    # 在你的自定义数据集上训练模型
    model.train(data="my_custom_dataset.yaml", epochs=100, imgsz=640)

    如需查看详细的配置选项,请访问配置页面。

  • Ultralytics YOLO26 提供多种预训练模型,可用于目标检测、实例分割、语义分割、深度估计、分类、姿态估计和有向边界框。这些模型分别在 COCO 数据集、用于语义分割的 Cityscapes、多数据集混合数据集(用于深度估计)、ImageNet(用于分类)或 DOTAv1(用于 OBB)上进行了预训练。以下是部分可用模型:

    如需查看详细列表和性能指标,请参阅模型部分。

  • 若要验证训练好的 YOLO26 模型的精度,可以在 Python 中使用 .val() 方法,或在 CLI 中使用 yolo detect val 命令。这会提供 mAP50-95、mAP50 等指标。

    示例
    from ultralytics import YOLO
    
    # 加载模型
    model = YOLO("path/to/best.pt")
    
    # 验证模型
    metrics = model.val()
    print(metrics.box.map)  # mAP50-95

    如需了解更多验证详情,请访问验证页面。

  • Ultralytics YOLO26 支持将模型导出为多种格式,例如 ONNX、TensorRT、CoreML 等,以确保模型兼容不同平台和设备。

    示例
    from ultralytics import YOLO
    
    # 加载模型
    model = YOLO("yolo26n.pt")
    
    # 将模型导出为 ONNX 格式
    model.export(format="onnx")

    请在导出页面查看完整的受支持格式列表和相关说明。

  • Ultralytics YOLO26 旨在为目标检测、实例分割、语义分割和姿态估计提供先进的性能。以下是一些主要优势:

    1. 预训练模型:使用在 COCO 和 ImageNet 等热门数据集上预训练的模型,加快开发进程。
    2. 高精度:取得出色的 mAP 分数,确保目标检测结果可靠。
    3. 速度:针对实时推理进行了优化,非常适合需要快速处理的应用。
    4. 灵活性:将模型导出为 ONNX 和 TensorRT 等多种格式,以便部署到各种平台。

    在我们的博客中了解 YOLO26 实际应用的案例和成功经验。

评论