YOLO[室内办公与个人携带场景][护照]目标检测数据集
YOLO[室内办公与个人携带场景][护照]目标检测数据集
YOLO26 目标检测模型训练全流程
📊 数据集基本信息
- 目标类别: [‘passport - v1 2025-05-15 4-39pm’]
- 中文类别:[‘护照’]
- 训练集:187 张
- 验证集:87 张
- 测试集:0 张
- 总计:274 张
📄 data.yaml 配置信息
该数据集提供了data.yaml文件,内容如下:
train: ../train/images
val: ../valid/images
test: ../test/images
nc: 1
names: ['passport - v1 2025-05-15 4-39pm']
🖼️ 标注可视化
数据集下载
小郭AI日志:
https://mp.weixin.qq.com/mp/appmsgalbum?__biz=MzYzNTg4ODk4NA==&action=getalbum&album_id=4497813105477042181&scene=126#wechat_redirect @image:image.dbe2d44a14.png



数据集详细介绍
该数据集名称为“YOLO[室内办公与个人携带场景][护照]目标检测数据集”,从名称可推测其主要用于识别在室内办公环境或个人携带场景中出现的护照图像。此类数据集通常应用于需要快速定位和识别证件类物品的场景,如安检、身份核验、智能办公设备等。在实际应用中,该数据集可以支持开发针对特定场景的自动化检测系统,提高证件识别的效率和准确性,尤其适用于需要高精度识别的封闭或半封闭环境。
数据集包含274张图片,覆盖了室内办公和个人携带场景下的护照图像,类别数量为1,表明该数据集专注于单一目标——护照的检测任务。尽管类别数量较少,但其应用场景明确,且图像内容可能包含不同角度、光照条件以及背景复杂度的变化,这为模型的泛化能力提供了基础支持。由于类别单一,该数据集更适合用于特定目标的精调训练,而非通用目标检测任务。
标注规范方面,数据集中的每张图片均对护照位置进行了精确标注,通常以边界框(bounding box)形式呈现。这种标注方式符合主流目标检测任务的标准,便于直接用于YOLO等经典检测算法的训练与评估。同时,高质量的标注数据有助于提升模型在实际部署中的准确性和稳定性,尤其适合需要高精度检测的工业级应用。
该数据集的典型应用方向包括智能安防系统的证件识别模块、办公场景中的自动登记系统以及个人电子设备中的身份验证功能。对于希望在特定场景下实现快速检测和识别的开发者而言,建议结合具体应用场景进行数据增强,例如增加不同光照、遮挡或姿态变化的样本,以进一步提升模型的鲁棒性。此外,也可考虑与其他数据集进行融合,拓展至更广泛的证件检测任务。
YOLO26 目标检测算法原理
相比于 YOLOv11 等前代版本,YOLO26 在架构层面做出了几个重要改变,让推理流程更简洁高效。
端到端无 NMS 推理:传统 YOLO 模型在推理后需要执行非极大值抑制(NMS)来去除冗余检测框,这增加了后处理延迟和部署复杂度。YOLO26 默认采用一对一(one-to-one)检测头,直接输出每张图像最多 300 个检测结果 (N, 300, 6),完全省去 NMS 步骤,推理流水线大幅简化。在 CPU 上的 ONNX 推理速度相比 YOLO11n 提升高达 43%。
无 DFL 回归:YOLO26 移除了分布焦点损失(Distribution Focal Loss, DFL)结构,检测头更加轻量。边界框回归不再受限于预设的离散区间,既降低了模型复杂度又保持了同等甚至更优的回归精度。
Progressive Loss 与 STAL:训练过程中通过渐进损失(Progressive Loss)将监督信号逐步从辅助头部转移到推理阶段的主头部。小目标感知标签分配(STAL)策略特别提升小目标的正样本标签覆盖率,改善小物体检测性能。
MuSGD 混合优化器:创新性地将大语言模型常用的 Muon 优化器与 SGD 结合,实现更稳定高效的训练收敛。相比纯 SGD,MuSGD 在大 batch 训练时收敛更快且不易震荡。
模型尺度与 COCO 性能:YOLO26 提供 N/S/M/L/X 五种尺度:
| 模型 | mAP50-95 | CPU ONNX(ms) | T4 TensorRT(ms) | 参数量 |
|---|---|---|---|---|
| YOLO26n | 40.9 | 38.9 | 1.7 | 2.4M |
| YOLO26s | 48.6 | 87.2 | 2.5 | 9.5M |
| YOLO26m | 53.1 | 220.0 | 4.7 | 20.4M |
| YOLO26l | 55.0 | 286.2 | 6.2 | 24.8M |
| YOLO26x | 57.5 | 525.8 | 11.8 | 55.7M |
Nano 版本参数量仅 2.4M,CPU 推理速度比 YOLO11n 快 43%,适合边缘设备和小数据集;X 版本 mAP 达 57.5,适用于对准确率要求极高的场景。
双头架构设计:YOLO26 检测模型内置两个检测头——一对一(默认,无 NMS,输出 300 个检测框)和一对多(需 NMS,输出 8400 个候选框)。训练时一对多头作为辅助监督信号加速收敛,推理时默认使用一对一头实现端到端输出。
YOLO26 支持全部七大视觉任务:检测、实例分割、语义分割、深度估计、分类、姿态估计和定向边界框检测(OBB),一个框架覆盖从 2D 检测到 3D 感知的完整需求。
YOLO26 训练步骤详解
不管你用的是命令行还是 Python 脚本,YOLO26 的训练接口都保持一致。下面用最常见的 CLI 方式演示完整流程。
环境准备:
pip install ultralytics
yolo checks
python -c "import torch; print(torch.cuda.is_available())"
命令行训练:
yolo detect train data=data.yaml model=yolo26n.pt epochs=100 imgsz=416 batch=16 device=0 lr0=0.005 patience=50 cos_lr=True close_mosaic=10
Python API 训练:
from ultralytics import YOLO
model = YOLO("yolo26n.pt")
results = model.train(
data="data.yaml",
epochs=100,
imgsz=416,
batch=16,
device=0,
lr0=0.005,
patience=50,
cos_lr=True,
close_mosaic=10,
augment=True,
amp=True,
workers=8,
)
训练输出文件:
runs/detect/train/weights/best.pt:验证集上 mAP 最高的权重runs/detect/train/weights/last.pt:最后一轮保存的权重runs/detect/train/results.csv:每轮 loss 和指标数据runs/detect/train/confusion_matrix.png:混淆矩阵图runs/detect/train/results.png:训练曲线图runs/detect/train/val_batch*_pred.jpg:验证集预测效果图
单张推理测试:
model = YOLO("runs/detect/train/weights/best.pt")
results = model("test_image.jpg")
results[0].show()
多 GPU 分布式训练:
yolo detect train data=data.yaml model=yolo26n.pt device=0,1 epochs=100
恢复中断的训练:
yolo detect train model=path/to/last.pt data=data.yaml resume=True
关键训练参数配置详解
训练过程中最容易被忽视的参数其实是 patience——早停机制能帮你省掉大量无效训练时间,尤其是小数据集上。
| 参数 | 默认值 | 本项目设置 | 含义 |
|---|---|---|---|
model | - | yolo26n.pt | 预训练权重,n/s/m/l/x 五种尺度 |
epochs | 100 | 100 | 训练总轮数 |
imgsz | 640 | 416 | 输入图像尺寸 |
batch | 16 | 16 | 批次大小,根据显存调整 |
lr0 | 0.01 | 0.005 | 初始学习率 |
lrf | 0.01 | 0.01 | 最终学习率因子(lr0 × lrf) |
momentum | 0.937 | 0.937 | SGD 动量 |
weight_decay | 0.0005 | 0.0005 | 权重衰减(L2 正则化) |
warmup_epochs | 3.0 | 3.0 | 预热轮数 |
warmup_momentum | 0.8 | 0.8 | 预热期间动量初始值 |
box | 7.5 | 7.5 | 边界框回归损失权重 |
cls | 0.5 | 0.5 | 分类损失权重 |
dfl | 1.5 | 1.5 | DFL 损失权重(YOLO26 可置 0) |
patience | 100 | 50 | 早停轮数 |
cos_lr | False | True | 余弦学习率衰减 |
close_mosaic | 10 | 10 | 最后 N 轮关闭 Mosaic 增强 |
optimizer | auto | SGD | 优化器选择 |
amp | True | True | 自动混合精度训练 |
freeze | None | 0 | 冻结前 N 层骨干网络参数 |
dropout | 0.0 | 0.1 | 分类头 Dropout 率,防过拟合 |
损失权重里最需要关注的是 box=7.5 和 cls=0.5 的比例。box/cls 比越大,模型越在意框的位置而非类别正确性。如果你的场景对定位精度要求极高(如精密零件检测),可以考虑把 box 提到 10。
学习率调参经验:
- 训练初期 loss 不下降 → 调大
lr0或增加warmup_epochs - 训练后期 loss 震荡 → 降低
lr0,增大weight_decay - 验证集 loss 先降后升(过拟合)→ 增大
dropout、weight_decay或减小epochs - 小数据集(<100张)→
lr0=0.001,weight_decay=5e-4,dropout=0.2 - 大数据集(>1000张)→
lr0=0.01,weight_decay=5e-4,batch=32+
训练可视化与TensorBoard:YOLO26 默认开启 TensorBoard 日志记录,训练过程中可实时查看 loss 曲线和指标变化。在命令行执行 tensorboard --logdir runs/detect/train 即可打开 Web 看板,每轮训练完成后自动刷新。训练结束后 results.png 生成的六合一曲线图是最直观的诊断工具——一眼就能看出是否过拟合、是否收敛、哪个 loss 在震荡。
断点续训机制:如果训练意外中断(断电、OOM 等),不用从头开始。YOLO 每轮保存 last.pt,恢复训练时指定 model=last.pt 并设置 resume=True,学习率和优化器状态都会从断点恢复,训练曲线也和连续训练一样平滑。
YOLO 系列算法演进史
从 YOLOv1 的一刀切 7×7 网格到 YOLO26 的无 NMS 自适应检测头,这条进化路线本质上是在不断移除人工设计的先验假设,让模型自己学会判断。
YOLOv1 (2015):将检测问题建模为单一回归问题,直接在输出层预测边界框和类别概率。速度极快但定位精度较差。
YOLOv2/YOLOv3 (2016-2018):引入 anchor box 机制、特征金字塔(FPN)和多尺度训练。YOLOv3 的 Darknet-53 骨干网络成为经典架构,至今仍被广泛使用。
YOLOv5 (2020):Ultralytics 在 PyTorch 上的实现,带来了完善的训练框架、自动锚框聚类和数据增强 pipeline,大幅降低了使用门槛。
YOLOv8 (2023):统一了检测、分割、分类和关键点任务的框架,引入 C2f 模块和无锚框(anchor-free)检测头。
YOLOv9 (2024):提出 GELAN(通用高效层聚合网络)和 PGI(可编程梯度信息),同等参数量下精度提升显著。
YOLOv10/YOLOv11 (2024):YOLOv10 首次提出无 NMS 端到端检测。YOLOv11 进一步优化骨干网络和训练策略,在速度和精度之间取得新平衡。
YOLO26 (2026):集历代之大成,核心创新包括:① 端到端无 NMS 一对一检测头 ② 无 DFL 的简化回归头 ③ Progressive Loss 渐进式监督转移 ④ STAL 小目标感知标签分配 ⑤ MuSGD 混合优化器。
关键设计演进总结:
- Anchor-based → Anchor-free(v8 起)
- C3 → C2f → C3k2(骨干模块持续优化)
- 多任务统一框架(v8 起)
- NMS 后处理 → 无 NMS 端到端(v10/v26)
- 单一损失函数 → 多任务联合损失(Box + Cls + DFL,v26 可去 DFL)
YOLO 数据增强策略
YOLO26 默认的训练增强已经覆盖了 90% 的场景。你需要做的不是加新增强,而是理解每个参数在哪调、怎么关。
Mosaic 拼接:将 4 张训练图像随机缩放后拼接为一张,大幅增加场景组合的多样性。对背景复杂、目标尺寸差异大的场景特别有效。训练最后 10 轮通过 close_mosaic=10 自动关闭 Mosaic,让模型适应真实数据分布。
MixUp 混合:两张图像按一定比例(α=0.3)线性混合,标签也按同比例混合。能平滑模型决策边界,对小数据集防过拟合效果显著。
HSV 色彩空间扰动:在 HSV 通道上对图像进行随机偏移(色相 ±0.015、饱和度 ±0.7、明度 ±0.4),模拟不同光照和色彩条件下的拍摄效果,提升模型对光照变化的鲁棒性。
几何变换:随机水平翻转(fliplr=0.5)、随机缩放(scale=0.5)、随机平移(translate=0.1),帮助模型学习位置不变性特征。
Copy-Paste 增强:将目标实例复制到另一张图像的随机位置,增加目标实例多样性。通过 copy_paste=0.3 控制概率。
增强参数配置(data.yaml 中可选覆盖):
hsv_h: 0.015
hsv_s: 0.7
hsv_v: 0.4
degrees: 0.0
translate: 0.1
scale: 0.5
shear: 0.0
perspective: 0.0
flipud: 0.0
fliplr: 0.5
mosaic: 1.0
mixup: 0.1
copy_paste: 0.1
部署与应用建议
从训练到上线的检查清单:
- 确认验证集 mAP50 ≥ 0.80(达标才能上线)
- 导出 ONNX 并验证 input/output shape 正确
- 在测试集上跑一遍推理,抽样检查预测框合理性
- 准备 5-10 张真实场景的图做最终验收
# 一键导出所有常用格式
from ultralytics import YOLO
model = YOLO("runs/detect/train/weights/best.pt")
model.export(format="onnx", imgsz=416)
model.export(format="engine", device=0, imgsz=416, half=True)
YOLO26 无 NMS 的部署红利:由于检测头直接输出 [N,300,6] 格式结果,后端代码可以精简到几十行。不需要实现 NMS 算法、不需要调 IoU 阈值、不用担心 NMS 导致的漏检。对于需要快速集成到 C++/Java 后端的场景,这个简化是最实在的优势。
量化 vs 蒸馏:如果需要在性能有限的设备上运行,int8 量化(int8=True)是首选,操作简单精度损失小。如果量化后的精度不满足要求,再考虑知识蒸馏(大模型教小模型),但蒸馏需要额外训练一次,工程成本更高。
通用训练经验:274 张图训练目标检测模型,遇到的问题是典型的——数据不多不少,精度不上不下。
这种规模的数据集,调参的边际收益其实很高。几个值得优先尝试的方向:
一是学习率的余弦退火。默认 YOLO26 用的是 cos_lr 衰减,但周期很关键。如果训练 80 轮,lr0=0.01 搭配 cos_lr=True,学习率会从 0.01 平滑降到接近 0。这个曲线比阶梯式衰减更适合中等数据集,收敛更稳定。
二是 mosaic=1.0 时最后 10 轮关闭(close_mosaic=10)。这个设置让模型在训练末期接触真实数据分布,对部署时的精度很重要。很多人忽视这个参数,训完发现部署精度掉了一大截。
三是关注训练过程中的 val_box_loss 拐点。通常 30-40 轮附近会出现一个平台期,如果 50 轮后仍无明显下降,基本可以判断当前参数组合不够理想。
实践建议:以上内容整理了 YOLO26 官方文档和社区反馈的核心要点。真实训练中不同的数据集总会有意料之外的问题,因此跑起来后重点盯三个指标:val/box_loss(边界框回归质量)、metrics/mAP50(B)(粗粒度检测精度)、metrics/mAP50-95(B)(精确定位水平)。
如果训练到一半 val/box_loss 开始抬头,第一时间检查是否过拟合——把 weight_decay 加倍、dropout 拉到 0.2,通常能刹住车。反过来,如果两个 loss 都居高不下且训练了 30 轮都没改善,那大概率是学习率太低或者标注有硬伤,建议先排查标签质量再回头调参。
训练完成后的操作清单:① 在测试集上确认 mAP 符合预期 ② 检查 confusion_matrix.png 有无异常的模式 ③ 抽取几张 val_batch*_pred.jpg 用肉眼判断预测效果是否合理。这些都做完后,再决定是否导出部署。
另外一个容易踩的坑是标注类别名称不统一——比如有些标签文件里同一类目标叫 ‘car’,有些叫 ‘Car’,YOLO 会当成两个不同的类。训练前用 grep 检查一遍 labels/ 下的 txt 文件就能避免这种低级错误。
DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐

所有评论(0)