YOLO11 预训练模型指南:Ultralytics、Roboflow Universe 与 HuggingFace

Ultralytics 官方预训练模型

YOLO11 由 Ultralytics 维护,提供五档大小(n/s/m/l/x),覆盖四类主要任务:

任务示例权重用途
目标检测yolo11n.pt ~ yolo11x.pt输出 bbox + 类别
实例分割yolo11n-seg.pt ~ yolo11x-seg.pt输出精确 Mask
姿态识别yolo11n-pose.pt ~ yolo11x-pose.pt人体关键点
旋转框(OBB)yolo11n-obb.pt ~ yolo11x-obb.pt遥感/倾斜目标

默认基于 COCO 80 类训练,可直接检测人、车、猫狗、手机、键盘等常见目标:

from ultralytics import YOLO

model = YOLO("yolo11m.pt")
results = model("image.jpg")

推理返回的 results 包含 bbox 坐标(xywhxyxy)、置信度和类别 ID。

实例分割

model = YOLO("yolo11m-seg.pt")
results = model("image.jpg")

# 获取多边形点集(可直接转 YOLO 分割格式)
for r in results:
    for mask in r.masks.xy:
        print(mask)  # [[x, y], ...]

垂直场景预训练权重

不需要从头训练的常见场景权重可在以下平台获取:

Roboflow Universe:提供数千个社区训练好的 YOLO 权重,支持直接下载或 API 调用。常见方向:

  • 安全帽检测(工地、工厂)
  • 烟火检测
  • 跌倒检测
  • 车牌识别
  • PCB 缺陷检测
  • 路面裂缝

HuggingFace:搜索 “yolo11” 或 “yolov8” 可找到专有领域微调版本,包含医学图像、农业、无人机航拍等方向。

更多可用模型

除 YOLO11 外,以下模型各有优势:

模型特点适用场景
YOLO-World文本描述直接检测任意目标零样本检测
YOLO12Attention 架构,精度更高精度优先
RT-DETRTransformer 检测器,速度快部署优先
RF-DETR小数据集效果好数据少的场景
YOLOX无 AGPL 协议争议商业部署
PP-YOLOE百度飞桨出品国内生态

YOLO-World 示例

from ultralytics import YOLO

model = YOLO("yolov8s-worldv2.pt")

# 设置类别(无需重新训练)
model.set_classes(["car", "person", "safety helmet"])

results = model("image.jpg")

输入类别名称即可检测,适合快速验证和标注辅助。

推荐工作流

没有垂直场景数据时:

1. 从 Roboflow Universe 找现有权重
2. 用预训练权重对自己数据集预标注
3. 人工审核 + 修正
4. 在预训练权重基础上微调(而非从头训练)

微调比从头训练所需的数据量少 3~5 倍,收敛速度也更快。