Ultralytics 支持的模型#
Ultralytics 支持多种模型,每种模型都针对特定任务进行了优化,例如目标检测、实例分割、语义分割、深度估计、图像分类、姿态估计和有向边界框。

精选模型#
新项目可以从 YOLO26 开始:这是 Ultralytics 的最新版本,也是唯一覆盖全部七种任务的模型。YOLO11 是成熟的替代方案,五种任务均有预训练检查点。只有在需要可提示分割(SAM 3)、开放词汇检测(YOLOE、YOLO-World)或 Transformer 检测器(RT-DETR)时,才选择专用模型系列。
YOLO27 正在进行最后阶段的 R&D,预计将于今年晚些时候推出,但尚未确定发布日期。预览其架构、计划支持的七种任务和初步基准测试结果。模型权重和软件包支持尚未开放;当前项目请使用 YOLO26。
此表列出了已发布的模型和架构参考资料,以及它们支持的任务、支持的 train、val、predict 和 export 模式和推荐用例。Track 未单独列出:它基于 Detect、Segment、Pose 和 OBB 模型的 predict 运行,而 SAM 2 和 SAM 3 则通过各自的视频预测器进行跟踪。Benchmark 也未单独列出,因为它会对多种格式执行 export 和 val,而不会新增自身支持的功能。
| 模型 | 任务 | 模式 | 适用场景 |
|---|---|---|---|
| YOLO26 🚀 新品 | 检测、分割、语义分割、深度估计、分类、姿态估计、OBB | 训练、验证、预测、导出 | 新项目:可选的无 NMS 推理、针对边缘设备优化的部署,以及最广泛的任务覆盖 |
| YOLO12 | 检测、分割、分类、姿态估计、OBB | 训练、验证、预测、导出 | 用于基准测试的注意力机制社区版本;预训练权重仅覆盖检测 |
| YOLO11 | 检测、分割、分类、姿态估计、OBB | 训练、验证、预测、导出 | 适用于稳定的生产工作负载,支持的每种任务都有预训练检查点 |
| YOLOv10 | 检测 | 训练、验证、预测、导出 | 清华大学开展的无 NMS 检测研究 |
| YOLOv9 | 检测、分割 | 训练、验证、预测、导出 | 可编程梯度信息(PGI),基于 Ultralytics YOLOv5 代码库实现 |
| YOLOv8 | 检测、分割、分类、姿态估计、OBB | 训练、验证、预测、导出 | 现有 YOLOv8 流程和成熟的第三方集成 |
| YOLOv7 | 检测 | 预测 | 运行上游训练的 YOLOv7 导出模型:该软件包可使用兼容的 ONNX 或 TensorRT 模型进行预测,但无法使用原生检查点进行预测 |
| YOLOv6 | 检测 | 训练、验证、预测、导出 | 根据 YAML 训练 美团 的架构;Ultralytics 未托管 YOLOv6 .pt 检查点 |
| YOLOv5 | 检测 | 训练、验证、预测、导出 | 适用于旧版 Ultralytics 项目;当前版本会加载更新后的 YOLOv5u 检查点 |
| YOLOv4 | 无 | 无 | 仅供参考的架构:Alexey Bochkovskiy 基于 Darknet 的原生模型不受该软件包支持 |
| YOLOv3 | 检测 | 训练、验证、预测、导出 | 基于 Joseph Redmon 原始架构的旧版项目;当前版本会加载更新后的 YOLOv3u 检查点 |
| SAM 3 🚀 新品 | 分割 | 预测 | Meta 推出的可提示概念分割模型,可依据文本或图像示例对图像和视频进行分割;sam3.pt 和 SAM 3.1 sam3.1_multiplex.pt(图像)需要获得 Hugging Face 访问批准 |
| SAM 2 | 分割 | 预测 | Meta 推出的可提示分割模型,可跨视频帧跟踪物体 |
| SAM | 分割 | 预测 | Meta 推出的原始可提示分割模型,支持自动标注 |
| MobileSAM | 分割 | 预测 | 适用于移动设备和其他资源受限设备的可提示分割模型(庆熙大学) |
| FastSAM | 分割 | 验证、预测、导出 | 当 SAM 的延迟成为瓶颈时,可使用基于 CNN 的可提示分割模型(中国科学院) |
| YOLO-NAS | 检测 | 验证、预测、导出 | Deci 的 NAS 优化型检测器,保留用于推理和导出;被 NVIDIA 收购后,Deci 不再维护这些模型 |
| RT-DETR | 检测 | 训练、验证、预测、导出 | 百度的实时 DETR 检测器:采用卷积骨干网络和混合式 Transformer 编码器 |
| YOLO-World | 检测 | 训练、验证、预测、导出 | 根据文本提示进行开放词汇检测(腾讯 AI 实验室);导出需要 -worldv2 检查点 |
| YOLOE | 检测、分割 | 训练、验证、预测、导出 | 通过文本、视觉提示或无提示推理进行开放词汇检测和分割 |
观看: 只需几行代码即可运行 Ultralytics YOLO 模型。
入门:用法示例#
以下示例使用基于 COCO 预训练的 YOLO26n 模型进行目标检测,并在示例数据集 COCO8 上训练该模型。相同的 API 也适用于上文列出的 Segment、Semantic、Depth、Classify、Pose 和 OBB 任务。
PyTorch 预训练的 *.pt 模型可传递给 YOLO()、SAM()、NAS() 和 RTDETR() 类,以便在 Python 中创建模型实例(YOLO() 和 RTDETR() 也接受 *.yaml 配置文件):
from ultralytics import YOLO
# 加载经过 COCO 预训练的 YOLO26n 模型
model = YOLO("yolo26n.pt")
# 显示模型信息(可选)
model.info()
# 使用 COCO8 示例数据集训练模型 100 个 epoch
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)
# 使用 YOLO26n 模型对“bus.jpg”图像进行推理
results = model("path/to/bus.jpg")贡献新模型#
Ultralytics 欢迎贡献新模型。要添加模型:
-
Fork 仓库:首先,Fork Ultralytics GitHub 仓库。
-
克隆你的 Fork:将你的 Fork 克隆到本地计算机,并创建一个新分支进行开发。
-
实现你的模型:按照贡献指南中的编码标准和指南添加模型。
-
充分测试:分别单独测试模型,并将模型作为完整流水线的一部分进行测试。
-
创建拉取请求:对模型感到满意后,向主仓库提交拉取请求以供审核。
-
代码审查与合并:审查后,如果模型符合我们的标准,就会合并到主仓库。
常见问题#
只需几行 Python 代码或一条 CLI 命令,即可使用自定义数据训练 YOLO 模型:
示例from ultralytics import YOLO # 加载 YOLO 模型 model = YOLO("yolo26n.pt") # 或任何其他 YOLO 模型 # 使用自定义数据集训练模型 results = model.train(data="custom_data.yaml", epochs=100, imgsz=640)如需更详细的说明,请访问 Train 文档页面。
Ultralytics 原生支持 YOLOv3、YOLOv5、YOLOv6、YOLOv8、YOLOv9、YOLOv10、YOLO11、YOLO12 和 YOLO26,以及 SAM 系列(SAM 3、SAM 2、SAM、MobileSAM 和 FastSAM)、YOLO-NAS、RT-DETR、YOLO-World 和 YOLOE。该软件包不提供 YOLOv4 或 YOLOv7 的权重或 YAML 文件:YOLOv4 仅作为架构参考进行文档说明,而 YOLOv7 则以导出的 ONNX 或 TensorRT 模型运行。请参阅精选模型,了解每种模型支持的任务和模式。
Ultralytics Platform 是一个无需编写代码的端到端环境,可用于标注数据集,以及训练、部署和管理 YOLO 机器学习模型。它提供云端训练、数据集管理功能,以及适合初学者和经验丰富开发者的界面。
Ultralytics YOLO 模型可执行任务,包括目标检测、实例分割、语义分割、深度估计、分类、姿态估计和有向目标检测(OBB)。最新模型 YOLO26 支持全部七项任务,还支持开放词汇检测。