Ultralytics YOLO11 多模式框架指南

Ultralytics YOLO11是一个多功能框架,旨在覆盖机器学习模型的整个生命周期——从数据摄取和模型训练验证部署和现实世界跟踪。每种模式都服务于特定目的,并且设计得灵活高效,以满足不同任务和用例的需求。本指南旨在为提供每种模式的概述和实用见解,帮助充分发挥YOLO11的潜力。

模式概览

了解Ultralytics YOLO11支持的不同模式对于充分利用模型至关重要:

  • 训练模式:在自定义或预加载的数据集上微调模型。
  • 验证模式:训练后的检查点,用于验证模型性能。
  • 预测模式:在现实世界数据上释放模型的预测能力。
  • 导出模式:使模型以各种格式部署就绪。
  • 跟踪模式:将目标检测模型扩展到实时跟踪应用中。
  • 基准测试模式:在多样化的部署环境中分析模型的速度和准确性。

训练模式

训练模式用于在自定义数据集上训练YOLO11模型。在此模式下,模型使用指定的数据集和超参数进行训练。训练过程涉及优化模型的参数,以便它能够准确预测图像中对象的类别和位置。

from ultralytics import YOLO model = YOLO("yolo11n.pt") model.train(data="path/to/dataset.yaml", epochs=100, imgsz=640)

验证模式

验证模式用于在训练后验证YOLO11模型。在此模式下,模型在验证集上进行评估,以衡量其准确性和泛化性能。此模式可用于调整模型的超参数以提高其性能。

from ultralytics import YOLO model = YOLO("yolo11n.pt") model.val(data="path/to/validation.yaml")

预测模式

预测模式用于使用训练过的YOLO11模型在新图像或视频上进行预测。在此模式下,模型从检查点文件中加载,用户可以提供图像或视频以执行推理。模型预测输入图像或视频中对象的类别和位置。

from ultralytics import YOLO model = YOLO("yolo11n.pt") model.predict(source="path/to/video.mp4")

导出模式

导出模式用于将YOLO11模型导出为可用于部署的格式。在此模式下,模型被转换为其他软件应用程序或硬件设备可以使用的格式。此模式在将模型部署到生产环境时非常有用。

from ultralytics import YOLO model = YOLO("yolo11n.pt") model.export(format="onnx")

跟踪模式

跟踪模式用于使用YOLO11模型实时跟踪对象。在此模式下,模型从检查点文件中加载,用户可以提供实时视频流以执行实时对象跟踪。此模式适用于监控系统或自动驾驶汽车等应用。

from ultralytics import YOLO model = YOLO("yolo11n.pt") model.track(source="path/to/video.mp4")

基准测试模式

基准测试模式用于分析YOLO11各种导出格式的速度和准确性。基准测试提供了导出格式的大小、mAP50-95(对于目标检测、分割和姿态)或accuracy_top5(对于分类)指标以及每张图像的推理时间(毫秒)等信息。这些信息可以帮助用户根据对速度和准确性的要求选择最适合的导出格式。

from ultralytics.utils.benchmarks import benchmark benchmark(model="yolo11n.pt", data="coco8.yaml", imgsz=640, half=False, device=0)

常见问题解答

如何使用Ultralytics YOLO11训练自定义目标检测模型?

使用Ultralytics YOLO11训练自定义目标检测模型涉及使用训练模式。需要一个格式化为YOLO格式的数据集,包含图像和相应的注释文件。使用以下命令开始训练过程:

yolo train data=path/to/dataset.yaml epochs=100 imgsz=640

有关更详细的说明,请参阅。

Ultralytics YOLO11在验证模型性能时使用哪些指标?

Ultralytics YOLO11在验证过程中使用各种指标来评估模型性能。这些包括:

  • mAP(平均精度均值):评估目标检测的准确性。
  • IOU(交集比并集):测量预测和真实边界框之间的重叠。
  • 精确度和召回率:精确度测量真正例检测与总检测阳性的比率,而召回率测量真正例检测与总实际阳性的比率。
yolo val data=path/to/validation.yaml

有关更多详细信息,请参阅。

如何导出YOLO11模型以进行部署?

Ultralytics YOLO11提供导出功能,将训练好的模型转换为各种部署格式,如ONNX、TensorRT、CoreML等。使用以下示例导出模型:

yolo export model=yolo11n.pt format=onnx

有关每种导出格式的详细步骤,请参阅。

Ultralytics YOLO11中的基准测试模式的目的是什么?

Ultralytics YOLO11中的基准测试模式用于分析各种导出格式如ONNX、TensorRT和OpenVINO的速度和准确性。它提供了模型大小、mAP50-95(用于目标检测)等指标以及不同硬件设置下的推理时间,帮助选择最适合的格式以满足部署需求。

yolo benchmark model=yolo11n.pt data='coco8.yaml' imgsz=640 half=False device=0

有关更多详细信息,请参阅。

如何使用Ultralytics YOLO11进行实时目标跟踪?

使用Ultralytics YOLO11的跟踪模式可以实现实时目标跟踪。此模式将目标检测功能扩展到视频帧或实时馈送中的目标跟踪。使用以下示例启用跟踪:

yolo track source=path/to/video.mp4
沪ICP备2024098111号-1
上海秋旦网络科技中心:上海市奉贤区金大公路8218号1幢 联系电话:17898875485