Ultralytics 支持的模型#
欢迎阅读 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 还提供 LLM,这是一个兼容 OpenAI 的大语言模型和视觉模型接口,可用于文本和图像理解。它本身不包含任务或模式,因此未列入上表,但可以与任何 YOLO 流水线配合使用,并连接到 OpenAI 等云服务提供商,也可以在本地部署的服务器上完全运行。
Watch: Run Ultralytics YOLO models in just a few lines of code.
入门:使用示例#
此示例展示了简单的 YOLO 训练和推理用法。如需了解这些及其他模式的完整文档,请参阅 Predict、Train、Val 和 Export 文档页面。
请注意,以下示例重点介绍用于目标检测的 YOLO26 Detect 模型。如需了解其他支持的任务,请参阅 Segment、Semantic、Depth、Classify、Pose 和 OBB 文档。
可以将预训练的 PyTorch *.pt 模型以及配置 *.yaml 文件传递给 YOLO()、SAM()、NAS() 和 RTDETR() 类,以在 Python 中创建模型实例:
from ultralytics import YOLO
# 加载经过 COCO 预训练的 YOLO26n 模型
model = YOLO("yolo26n.pt")
# 显示模型信息(可选)
model.info()
# 使用 COCO8 示例数据集训练模型 100 个周期
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 克隆到本地计算机,并新建一个分支进行开发。
-
实现你的模型:按照贡献指南中的编码标准和规范添加模型。
-
全面测试:确保对模型进行严格测试,包括单独测试以及作为流水线一部分进行测试。
-
创建拉取请求:对模型满意后,向主仓库创建拉取请求以供审核。
-
代码审查与合并:审核通过后,如果模型符合我们的标准,就会合并到主仓库。
如需了解详细步骤,请参阅我们的贡献指南。
常见问题#
使用 Ultralytics 库可以轻松地在自定义数据上训练 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 模型。它简化了复杂的工作流,让用户能够专注于模型性能和应用。此外,平台还提供云端训练功能、全面的数据集管理,以及适合初学者和资深开发者的易用界面。