mcpField
Y
yolo-models

yolo-models

github.com/ultralytics开发33125 次安装1 天前更新
代码仓库

关于 yolo-models

用于按任务、规模和部署约束选择 Ultralytics 视觉模型的参考 Skill,帮助用户挑选模型系列并配置自定义架构。

主要功能

  1. 01比较模型系列与规模
  2. 02按后缀匹配视觉任务
  3. 03选择开放词汇或提示式模型
  4. 04加载并定制架构 YAML

适用场景

  1. 01为新的视觉任务选择模型
  2. 02按延迟约束挑选模型规模
  3. 03决定检测或分割方案

安装

复制下面的命令,在终端中运行。

$npx skills add https://github.com/ultralytics/skills --skill yolo-models

安装后在 Claude Code 中输入/yolo-models确认已加载。

附带文件agents/openai.yamlweights-catalog.md

详细介绍

选择 Ultralytics 模型

默认推荐:预训练的 YOLO26。 最新一代,提供可选的无 NMS 检测头(nms=False),可实现最快的 CPU 推理和最简单的部署。只有在需要匹配现有代码库,或部署目标尚不支持 YOLO26 时,才使用 YOLO11/YOLOv8。大多数官方权重会在首次使用时自动下载;sam3.pt 和 sam3.1_multiplex.pt 需要手动获取并下载。

在 Platform 中选择

如需最快速地无代码启动,请在项目中点击 New Model。其 Official 选项卡会列出 YOLO26(推荐)、YOLO11、YOLOv8 和 YOLOv5 模型,这些模型的任务可以训练所选数据集;My Models 会列出已完成训练或已上传的检查点,供进一步微调。如需先浏览官方模型,请在 Platform Explore 的 Projects 下打开 @ultralytics 项目。

使用 Platform 模型页面查看指标、在 Predict 中测试、导出、部署、将公开模型克隆到工作区,或下载其 .pt 权重以用于下文的 Python/CLI 工作流。请参阅 Platform Models 和 Explore。

模型 = 系列 + 尺寸 + 任务后缀

yolo26 + n/s/m/l/x + 任务后缀 → yolo26s-seg.pt

尺寸COCO mAP50-95参数量T4 TensorRT适用场景
n40.92.4M~1.7 ms边缘设备/移动端、CPU 实时推理、首个原型
s48.69.5M~2.5 ms大多数项目的均衡默认选择
m53.120.4M~4.7 msGPU 服务端、重视准确率
l55.024.8M~6.2 ms对准确率要求极高、GPU 资源充足
x57.555.7M~11.8 ms追求最高准确率、离线/批量处理

策略:先用 n 低成本验证工作流,然后逐步增大模型,直到准确率提升不再值得增加延迟。更大的模型永远无法修正糟糕的标签。

后缀任务输出
(none)检测边界框
-seg实例分割多边形 + 边界框
-sem语义分割(YOLO26+)逐像素类别掩码
-depth单目深度估计(YOLO26+)深度图
-cls分类类别概率
-pose姿态/关键点关键点 + 边界框
-obb有向边界框旋转边界框

较新任务的说明:

  • 语义分割(-sem):数据集使用通过 masks_dir(默认为 masks/)指定的 PNG 掩码,或多边形标签;指标为 mIoU。
  • 深度估计(-depth):目标为缩放后的 uint16 PNG 图(首选)或以米为单位的浮点 .npy 图;指标为 delta1。深度模型还提供 model.calibrate(data=...),用于拟合度量尺度校正;使用 model.save(...) 持久化该校正。

各系列速查表

系列类适用场景
YOLO26 / YOLO11 / YOLO12 / YOLOv8–v10YOLO("yolo26n.pt")标准封闭集任务;默认选择
YOLO-WorldYOLOWorld("yolov8s-worldv2.pt")零样本检测任意文本类别;model.set_classes(["person", "helmet"]);优先使用 -worldv2(原始 -world 权重无法导出)
YOLOEYOLOE("yoloe-26s-seg.pt")通过文本或视觉提示词进行开放词汇检测和分割;set_classes(["person", "helmet"]),通过 predict(..., visual_prompts={"bboxes": ..., "cls": ...}, predictor=YOLOEVPSegPredictor) 使用视觉提示词(默认预测器仅返回边界框);-pf 变体无需提示词
SAM / SAM2 / SAM3 / MobileSAMSAM("sam_b.pt")可通过提示词进行分割:predict(source, bboxes=... / points=... / labels=...);SAM2/SAM3 增加了视频预测器;SAM3 增加了文本/示例概念分割(SAM3SemanticPredictor);仅支持推理
FastSAMFastSAM("FastSAM-s.pt")基于 CNN 的全分割模型,速度远快于 SAM;不支持训练
RT-DETRRTDETR("rtdetr-l.pt")Transformer 检测器,在 GPU 上具有较高准确率
YOLO-NASNAS("yolo_nas_s.pt")不支持训练(仅支持 val/predict/export)

所有类都共享同一套 Model API(train/val/predict/track/export/...)——其他 yolo-* 技能中的所有内容都适用于它们,但上述例外情况除外。

开放词汇模型选择:需要在推理时检测任意类别且无需训练 → YOLO-World(检测)或 YOLOE(检测和分割,也支持视觉提示词)。需要通过点击/边界框获得像素级精准的掩码 → SAM 系列。需要针对固定类别列表训练模型 → 微调普通 YOLO26(在该封闭集上速度更快、准确率更高)。

架构 YAML(自定义模型)

ultralytics/cfg/models/ 提供可编辑的架构定义(yolo26.yaml、yolo11.yaml、yolov8.yaml 及其变体,例如 -p2(小目标)、-p6(较大的 imgsz)和 -ghost)。加载 YOLO("yolo26n.yaml") 会从头构建模型——规模由文件名主体中的字母决定。要自定义架构,同时保留层匹配处的预训练权重:

model = YOLO("yolo26n.yaml").load("yolo26n.pt")  # transfer matching weights

仅在研究或存在特殊约束时使用此方式;常规工作请微调原版 .pt。

相关页面

  • 本文件夹中的 weights-catalog.md——了解程序包已知的权重命名模式和专用官方资源。不要猜测权重名称。

根据已安装版本进行验证

模型可用性变化很快。运行以下命令可查看已安装程序包的已知快速路径集合;在判定未列出的官方资源无效前,请先阅读 weights-catalog.md:

python -c "from ultralytics.utils.downloads import GITHUB_ASSETS_NAMES; print(*sorted(GITHUB_ASSETS_NAMES), sep='\\n')"

如果权重返回 404 或类导入失败,请检查 yolo checks,并以已安装版本报出的错误为准。

评论

支持 Markdown