1. 为什么驾驶员行为检测突然成了刚需做智能驾驶方向的朋友应该都有体会这两年行业对“人”的关注度已经悄悄超过了“车”。传感器、融合感知、规控这些赛道卷到头之后车祸发生的最后一道防线反倒落在了驾驶员自己身上。业内共识是L2和L3级辅助驾驶在相当长时间内仍然是主流而这中间有一个长期无法回避的问题——驾驶员会疲劳、走神、看手机、甚至睡着。车企要保证主动安全系统在关键时刻接得住方向盘就必须先知道驾驶员此刻在干嘛。所以驾驶员行为监测Driver Monitoring SystemDMS从一个可选功能变成了新车评测里的常客也成了法规和碰撞安全评级里越来越重视的必选项。我自己在实际项目里跑过好几版驾驶员行为检测方案从最早用MediaPipe做手势粗筛到后来切到YOLO系列重新整理数据、训练、剪枝、部署最大的体会是算法模型的好坏反而不是最卡脖子的数据集的质量和结构才是决定项目上限的那道闸门。尤其是当我拿到这类规模为2万多张的驾驶员行为数据集时感受最深。这类数据集的标注类别、样本分布、场景覆盖度直接决定了模型有没有可能收敛、泛化到真实车内环境里。这篇文章就从一套典型的22600张驾驶员行为检测数据集出发拆解它的结构逻辑、标注方式、YOLO训练流程以及一个完整DMS项目落地时绕不开的细节和坑。内容不偏理论全是实际动手过程中会用到的东西适合正在做智能驾驶算法、毕设课题选型或者想快速搭建一套DMS原型的同学参考。2. 数据集整体设计与类别拆解2.1 22600张这个规模意味着什么先说一个关键判断22600张图片对于DMS任务来说是个什么量级我的结论是刚好够用且偏紧巴很多。做过检测项目的人都知道通用目标检测比如COCO是12万张起步自动驾驶的BDD100K有10万张但驾驶员行为检测是个窄域任务画面背景固定——基本都是驾驶舱内部、座椅、方向盘、仪表台目标形态变化没有开放世界那么剧烈。所以在这种约束下2万张是一个“能训练出可用的模型但必须把每一张样本的利用率和标注质量拉满”的量级。如果这22600张里存在明显的场景重复比如都是同一台模拟器拍的连续帧那么有效独立样本可能就只有几千模型过拟合的风险极大。反过来如果数据来自不同光照、不同人脸朝向、不同配件眼镜、口罩、帽子那这2万多张的含金量就非常高了。我在实际项目中见过不少标称几万张的数据集去掉重复帧和无效样本后真正能进训练集的可能不到一半。所以拿到任何数据集第一步不是急着跑模型而是先做一次彻底的样本清洗。2.2 典型类别划分与语义设计驾驶员行为检测数据集的类别设计行业内没有完全统一的标准但主流的分类方式基本围绕“安全相关动作”和“常规驾驶动作”展开。一套有代表性的类别划分大致如下类别编号类别名称行为含义检测难度0normal_driving正常驾驶较低1drinking喝水中等2texting发短信/用手机较高3talking_on_phone打电话中等4operating_radio操作中控台中等5reaching_behind回头取后座物品较高6hair_and_makeup照镜子/整理仪容较高7drowsiness疲劳闭眼较高8smoking吸烟中等9yawning打哈欠中等有些数据集会把抽烟和喝水合并成“hand_on_mouth”有的会额外加“用手遮挡面部”“刷短视频”等长尾行为。类别设计背后其实反映的是产品方的决策逻辑如果这个数据集用于商用车卡车、公交的主动安全监管那疲劳、分神、抽烟、打电话这类行为的权重就明显更高如果用于乘用车后装行车记录仪则会把操作中控、看手机这类日常高频场景做得更细。我给新人的建议是拿到数据集先画出每个类别的样本数量分布别急着训练。如果某个类别的样本数量少于500并且这个类别又是安全关键行为比如闭眼、打哈欠那你基本可以预测到模型在验证集上的表现会很难看除非后续做数据增强或采样策略补偿。2.3 标注格式与YOLO的对接方式当前22700张左右的驾驶行为数据集绝大多数会采用Pascal VOC或YOLO的标注格式。YOLO格式的标注文件是txt每行对应一个目标框内容是class_id center_x center_y width height注意这里的四个坐标值都是归一化后的相对值除以了图片宽高。比如一张1920x1080的图某个目标框的左上角是(480, 270)右下角是(960, 810)那么中心的相对坐标就是(0.375, 0.5)宽度相对值是0.25高度相对值是0.5对应标注行就是0 0.375 0.5 0.25 0.5这个转换逻辑看起来很简单但实际踩坑率很高。很多人从标注平台导出的坐标是绝对像素没有归一化直接喂给YOLO训练就会导致损失函数直接崩掉。我一般会写一个小脚本做格式校验先确认所有txt里的数值范围都在0到1之间再确认没有出现坐标反挂左下角右下角的异常值。3. 这套数据集在DMS项目中的实际定位3.1 算法选型YOLOv5、YOLOv8还是YOLO11和数据集绑定的算法分类讨论一下因为不同版本的YOLO对这个场景的适应度差别很大。如果是跑在云端或者有独立域控制器的设备上YOLOv8和YOLO11都是合理选择。YOLOv8的C2f模块能提升对细粒度特征的提取能力用在面部朝向、手部遮挡检测这种“小差别人”的目标上表现优于YOLOv5在相同训练条件下的效果。YOLO11推出来之后主打的就是骨干网络效率和检测头解耦在推理速度上做了优化适合需要实时处理的要求。但如果你的部署目标是一块Jetson Nano、RK3588这类中低算力芯片我会建议回归到YOLOv5s或者YOLOv8n这种轻量级变体。做DMS项目比拼的不是mAP那零点几的涨点而是帧率和误报率的平衡。单片设备上既要做驾驶员人脸检测又要做行为分类有时还叠加视线估计和人脸关键点留给每个模型的推理预算不高。一个n大小的模型能把单帧推理压在10毫秒内整个系统的实时性才有余量。3.2 数据集的适用边界和局限这套22600张的数据集典型应用场景有两种。一种是离线统计型任务。比如车队管理系统摄像头拍到的视频经过端侧检测后将驾驶员行为标签上传后台分析一天内有多少次分神、多少次抽烟、疲劳驾驶的总时长占比。这种场景下2万张数据训练出的模型足以支撑因为单帧图像的检测错误率可以通过连续帧的投票机制来平滑掉。另一种是实时预警型任务这在DMS里更常见也更难。要求在驾驶员闭眼超过一定时长后、或者视线偏离道路中心超过阈值时立刻触发报警。这种实时预警场景对模型有两个隐性要求第一单帧检测鲁棒性要高。我们不能靠“这一帧没检出来没关系下一秒再检”来做兜底因为下一秒可能就撞了。第二是检测延迟必须可控。闭眼这个动作如果到第0.5秒才被捕捉到预警的价值就打了折扣。22600张的数据集在训练这样一个实时预警模型时能提供足够的基线效果但无法覆盖所有极端情况。比如夜间几乎全黑的环境、强逆光下司机完全变成剪影、戴着墨镜导致眼睛关键点不可见等。这些情况往往需要额外补充针对性数据或者设计红外补光方案来规避。我自己处理这类问题的做法是先拿这套2万多张的数据训练一版基线模型然后在真实车载环境或者台架上采集一定量的“困难样本”二次微调。这样既发挥了数据集覆盖面广的优势又针对部署场景做了适配。千万别指望任何公开数据一次性满足真实生产环境DMS尤其如此。4. 实操用YOLO训练驾驶员行为检测模型4.1 环境准备与数据目录结构动手之前先把训练环境理一理。我的基础环境组合如下大家可以参照Python 3.10 PyTorch 2.1.2 CUDA 11.8 ultralytics 8.3.x 显卡最低建议RTX 3060 12G或以上数据集的目录结构建议这样组织driver_behavior_dataset/ ├── images/ │ ├── train/ │ ├── val/ │ └── test/ ├── labels/ │ ├── train/ │ ├── val/ │ └── test/ └── driver_behavior.yaml注意YOLO训练时要求图像和标签一一对应最好保持文件名一致只是后缀不同。图片是00001.jpg标注文件就是00001.txt。如果发现有的图片没有对应标签或者反向缺失训练前必须处理掉否则ultralytics会报DatasetError。4.2 一个直接可用的YAML配置数据集的YAML配置文件是训练入口这里给一份可以直接套用的配置path: ./driver_behavior_dataset train: images/train val: images/val test: images/test names: 0: normal_driving 1: drinking 2: texting 3: talking_on_phone 4: operating_radio 5: reaching_behind 6: hair_and_makeup 7: drowsiness 8: smoking 9: yawning如果标签里的类别编号和这个列表对不上要么改YAML要么写脚本统一转换。我的建议是优先改数据端因为names的顺序一旦变了最后部署时的类别映射表也要跟着改容易引入低级错误。4.3 训练启动命令与关键参数训练命令本身不复杂复杂的是参数的选择。一个适合DMS任务的基准训练命令如下yolo detect train \ modelyolov8n.pt \ datadriver_behavior.yaml \ epochs300 \ imgsz640 \ batch32 \ device0 \ patience20 \ cos_lrTrue \ workers8 \ project./runs/driver_behavior \ nameyolov8n_dms_baseline几个参数的选择逻辑我想展开说说。imgsz640是绝大多数YOLO官方权重的主训练尺寸。DMS的图像通常来自车内摄像头分辨率从720P到1080P不等但真正输入模型前都会缩放。640尺寸能保证计算量与精度的平衡。如果你希望提升小目标比如远处的手机屏幕的召回率可以考虑768甚至896但训练速度会下降。batch32在12G显存上搭配yolov8n是可行的如果模型换成更大的比如yolov8m那batch就要降到16甚至8否则会爆显存。patience20是早停策略连续20个epoch验证集指标没有提升就停止训练。这套数据集如果出现某些类别样本严重不足早停能避免越过拟合那个拐点之后还傻傻训练下去。4.4 训练完成后的评估指标怎么读训练结束后会生成result.png和混淆矩阵很多人只盯着mAP值我觉得关键要看的并不是总体mAP。更值得关注的是安全关键类别的单类recall。比如drowsiness这一类如果recall能到95%以上说明绝大多数闭眼状态都能被框出来哪怕precision稍低一点也无妨——DMS产品里漏报的代价远高于误报。误报顶多是给驾驶员一个不必要的提醒漏报则是安全事故。训练日志里还可以看到PR曲线。我建议多关注小目标的AP因为驾驶员行为检测中很多目标框面积本来就小。比如手持手机这个目标框在画面里可能不到整个图像面积的5%如果AP不高说明模型对远距离手持物的识别存在短板需要额外补数据或者调阈值。5. 我总结的DMS数据集使用避坑清单5.1 类别不均衡是最大的敌人22600张数据集中正常驾驶类别的样本数通常会占大头而吸烟、打哈欠这类行为的样本可能只有几百张。如果不做任何处理模型会倾向于把绝大多数目标预测为多数类也就是“正常驾驶”因为这样总损失最小。我常用以下几种手段来缓解不均衡第一做类别加权采样。给少数类的每个样本设置更高的采样权重使得每个epoch里看到的少数类样本更多。ultralytics在训练时支持通过数据集配置或采样器来调整实际操作中也可以用oversample的方式手动干预。第二使用马赛克和复制粘贴增强。我试过把吸烟、打电话这类小目标样本复制粘贴到正常驾驶的背景图中生成额外的合成样本。效果提升显著特别是对训练过程中的正样本多样性有帮助。第三对关键类别使用略微提高的损失权重。这样能让模型更关注少数类即使分类错误的代价变高也值得。5.2 光照与遮挡问题不能靠增强完全解决数据增强能模拟旋转、缩放、平移、翻转但复杂的车内光照变化——比如太阳光从侧窗打入在驾驶员脸上形成高亮区域、隧道内光线骤变、夜间仪表盘反光——这些增强难以真实模拟。一个可行的做法是采用亮度对比度调整增强来近似极端光照更好的方案是用红外训练集或伪灰度图补充。很多量产DMS摄像头本身就是红外方案图像是近红外的灰度图所以如果数据集只有彩色照片推理时最好先做灰度归一化再喂给模型我发现这样能够提升夜间场景下的泛化性。5.3 目标框边缘与遮挡目标的标注习惯DMS场景中人物和物体经常被方向盘、座椅头枕遮挡如果标注时习惯性地将遮挡物也画进框内模型学习到的特征就会包含遮挡物信息泛化性变差。我处理这类情况时统一采用一个标注规范目标框只包含可见部分遮挡严重时宁可漏标也不把遮挡物画进去。同时统一规定小目标比如手机的框必须包含完整屏幕和手部握持区域不能只框屏幕。这些规范如果不在标注阶段统一后面模型输出的框会忽大忽小评估时的IoU指标也难以解释。5.4 训练时默认增强在DMS上的反效果ultralytics默认开启HSV色彩增强和仿射变换。HSV增强在DMS场景需要谨慎调节特别是色相偏移过大时会把肤色变成奇怪的绿色这会让模型对肤色特征的建模出现偏差。肤色在驾驶员检测中是一个非常重要的特征线索尤其是在手部检测和脸部检测中。我实际测试下来的经验是将hsv_h从默认的0.015降到0.005hsv_s保持不变或略微降低hsv_v基本不动。因为DMS的核心是识别人的状态色彩畸变轻微即可过度增强反而有害。5.5 验证集的划分策略适用于DMS数据集的划分方式并不是完全随机的。因为同一段连续视频中前后帧非常相似直接随机划分会将几乎相同的画面同时分到训练集和验证集导致验证指标虚高。推荐的做法是按视频片段聚类划分。对数据集中属于同一拍摄序列的帧做分组整组划分到训练或验证。如果数据集没有标注视频分组信息可以使用图像相似度做近似聚合把相同场景、相同人员、相似光照的帧归为一组。6. 实际项目里的典型问题与排查经验6.1 训练Loss正常下降但检测结果全是乱框这个现象多数和标注坐标格式错误有关。如果标签txt里的坐标是绝对像素而非归一化值模型在训练初期很可能发散了。排查方法很简单import numpy as np label_path xxx.txt data np.loadtxt(label_path).reshape(-1, 5) print(data.min(axis0)) print(data.max(axis0)) # 检查类别ID assert set(data[:, 0].astype(int)) set(range(10))如果发现某个坐标值大于1基本可以判断标注没归一化。处理方式是遍历所有标签文件将坐标除以对应的图片宽高。6.2 验证集mAP很高但实际摄像头测试效果很差出现这种问题首先要查看验证集和测试环境的数据分布差异。比如训练集全是白天光线充足的图片但实际测试在夜间环境进行那效果差是必然的。解决办法是引入夜间数据微调或者对输入图像做预处理如自动白平衡、去噪、对比度增强使输入分布接近白天场景。同时模型的设计要尽量耐光照变化可以考虑在网络前加一个轻量的自适应归一化层。6.3 频繁误报出“使用手机”而实际不在用手机如果模型把正常握方向盘的手误判为手机很可能是数据集中normal_driving类别里“手部靠近方向盘但手持物品”的样本太少。模型学到的不是“使用手机”这个行为而是“手部有物体”这个浅层特征。这个问题的解决思路有两个一是增加负样本的干扰数据在正常驾驶类别中多留一些手持水杯、拿纸巾的样本二是后处理逻辑配合对“使用手机”的报警需要连续若干帧都命中才触发单帧误报不立即提醒。后者在工程上更加可靠也是DMS产品落地时普遍采用的降误报策略。6.4 闭眼检测在戴墨镜时完全失效在驾驶员戴墨镜时眼睛区域的可见度急剧下降闭眼与睁眼的差异极其微弱。如果数据集缺乏墨镜样本的支持模型几乎不可能正确分类。针对这种情况有两条路可走。一条是增加墨镜样本进行专门的数据扩充另一种是绕开眼部状态判断改用头部姿态估计来综合判断疲劳状态。比如当驾驶员头部低头角度过大且持续一段时间即便看不到眼睛也可以判定为疑似疲劳。这两者结合使用在实际工程中才是稳的。7. 最终建议驾驶员行为检测数据集的选用和训练本质上是一次“数据质量、类别结构、工程约束”三者的平衡。22600张的YOLO数据集在这个任务里是一个很好的起步基准但我始终建议把它视为基线而不是终点。真正能让模型在实车上跑得稳的往往是后续针对目标车型、目标摄像头的补采数据以及一套合理可靠的报警策略。我在实际项目里把模型从YOLOv5的baseline换成经过数据清洗和困难样本微调的YOLOv8n之后误报率降低了接近一半漏报率基本保持一致。这个提升完全不是靠调参调出来的而是靠一点点啃数据、理清类别边界、把训练流程中的隐含问题逐个暴露并修正出来的。最后一个小建议训练之前务必抽出半天时间把数据标签全部过一遍用可视化脚本随机抽查几千张。你花在这个环节上的时间会在后面评估、调优、部署阶段成倍地赚回来。
驾驶员行为检测DMS实战：从数据集到YOLO训练全攻略
1. 为什么驾驶员行为检测突然成了刚需做智能驾驶方向的朋友应该都有体会这两年行业对“人”的关注度已经悄悄超过了“车”。传感器、融合感知、规控这些赛道卷到头之后车祸发生的最后一道防线反倒落在了驾驶员自己身上。业内共识是L2和L3级辅助驾驶在相当长时间内仍然是主流而这中间有一个长期无法回避的问题——驾驶员会疲劳、走神、看手机、甚至睡着。车企要保证主动安全系统在关键时刻接得住方向盘就必须先知道驾驶员此刻在干嘛。所以驾驶员行为监测Driver Monitoring SystemDMS从一个可选功能变成了新车评测里的常客也成了法规和碰撞安全评级里越来越重视的必选项。我自己在实际项目里跑过好几版驾驶员行为检测方案从最早用MediaPipe做手势粗筛到后来切到YOLO系列重新整理数据、训练、剪枝、部署最大的体会是算法模型的好坏反而不是最卡脖子的数据集的质量和结构才是决定项目上限的那道闸门。尤其是当我拿到这类规模为2万多张的驾驶员行为数据集时感受最深。这类数据集的标注类别、样本分布、场景覆盖度直接决定了模型有没有可能收敛、泛化到真实车内环境里。这篇文章就从一套典型的22600张驾驶员行为检测数据集出发拆解它的结构逻辑、标注方式、YOLO训练流程以及一个完整DMS项目落地时绕不开的细节和坑。内容不偏理论全是实际动手过程中会用到的东西适合正在做智能驾驶算法、毕设课题选型或者想快速搭建一套DMS原型的同学参考。2. 数据集整体设计与类别拆解2.1 22600张这个规模意味着什么先说一个关键判断22600张图片对于DMS任务来说是个什么量级我的结论是刚好够用且偏紧巴很多。做过检测项目的人都知道通用目标检测比如COCO是12万张起步自动驾驶的BDD100K有10万张但驾驶员行为检测是个窄域任务画面背景固定——基本都是驾驶舱内部、座椅、方向盘、仪表台目标形态变化没有开放世界那么剧烈。所以在这种约束下2万张是一个“能训练出可用的模型但必须把每一张样本的利用率和标注质量拉满”的量级。如果这22600张里存在明显的场景重复比如都是同一台模拟器拍的连续帧那么有效独立样本可能就只有几千模型过拟合的风险极大。反过来如果数据来自不同光照、不同人脸朝向、不同配件眼镜、口罩、帽子那这2万多张的含金量就非常高了。我在实际项目中见过不少标称几万张的数据集去掉重复帧和无效样本后真正能进训练集的可能不到一半。所以拿到任何数据集第一步不是急着跑模型而是先做一次彻底的样本清洗。2.2 典型类别划分与语义设计驾驶员行为检测数据集的类别设计行业内没有完全统一的标准但主流的分类方式基本围绕“安全相关动作”和“常规驾驶动作”展开。一套有代表性的类别划分大致如下类别编号类别名称行为含义检测难度0normal_driving正常驾驶较低1drinking喝水中等2texting发短信/用手机较高3talking_on_phone打电话中等4operating_radio操作中控台中等5reaching_behind回头取后座物品较高6hair_and_makeup照镜子/整理仪容较高7drowsiness疲劳闭眼较高8smoking吸烟中等9yawning打哈欠中等有些数据集会把抽烟和喝水合并成“hand_on_mouth”有的会额外加“用手遮挡面部”“刷短视频”等长尾行为。类别设计背后其实反映的是产品方的决策逻辑如果这个数据集用于商用车卡车、公交的主动安全监管那疲劳、分神、抽烟、打电话这类行为的权重就明显更高如果用于乘用车后装行车记录仪则会把操作中控、看手机这类日常高频场景做得更细。我给新人的建议是拿到数据集先画出每个类别的样本数量分布别急着训练。如果某个类别的样本数量少于500并且这个类别又是安全关键行为比如闭眼、打哈欠那你基本可以预测到模型在验证集上的表现会很难看除非后续做数据增强或采样策略补偿。2.3 标注格式与YOLO的对接方式当前22700张左右的驾驶行为数据集绝大多数会采用Pascal VOC或YOLO的标注格式。YOLO格式的标注文件是txt每行对应一个目标框内容是class_id center_x center_y width height注意这里的四个坐标值都是归一化后的相对值除以了图片宽高。比如一张1920x1080的图某个目标框的左上角是(480, 270)右下角是(960, 810)那么中心的相对坐标就是(0.375, 0.5)宽度相对值是0.25高度相对值是0.5对应标注行就是0 0.375 0.5 0.25 0.5这个转换逻辑看起来很简单但实际踩坑率很高。很多人从标注平台导出的坐标是绝对像素没有归一化直接喂给YOLO训练就会导致损失函数直接崩掉。我一般会写一个小脚本做格式校验先确认所有txt里的数值范围都在0到1之间再确认没有出现坐标反挂左下角右下角的异常值。3. 这套数据集在DMS项目中的实际定位3.1 算法选型YOLOv5、YOLOv8还是YOLO11和数据集绑定的算法分类讨论一下因为不同版本的YOLO对这个场景的适应度差别很大。如果是跑在云端或者有独立域控制器的设备上YOLOv8和YOLO11都是合理选择。YOLOv8的C2f模块能提升对细粒度特征的提取能力用在面部朝向、手部遮挡检测这种“小差别人”的目标上表现优于YOLOv5在相同训练条件下的效果。YOLO11推出来之后主打的就是骨干网络效率和检测头解耦在推理速度上做了优化适合需要实时处理的要求。但如果你的部署目标是一块Jetson Nano、RK3588这类中低算力芯片我会建议回归到YOLOv5s或者YOLOv8n这种轻量级变体。做DMS项目比拼的不是mAP那零点几的涨点而是帧率和误报率的平衡。单片设备上既要做驾驶员人脸检测又要做行为分类有时还叠加视线估计和人脸关键点留给每个模型的推理预算不高。一个n大小的模型能把单帧推理压在10毫秒内整个系统的实时性才有余量。3.2 数据集的适用边界和局限这套22600张的数据集典型应用场景有两种。一种是离线统计型任务。比如车队管理系统摄像头拍到的视频经过端侧检测后将驾驶员行为标签上传后台分析一天内有多少次分神、多少次抽烟、疲劳驾驶的总时长占比。这种场景下2万张数据训练出的模型足以支撑因为单帧图像的检测错误率可以通过连续帧的投票机制来平滑掉。另一种是实时预警型任务这在DMS里更常见也更难。要求在驾驶员闭眼超过一定时长后、或者视线偏离道路中心超过阈值时立刻触发报警。这种实时预警场景对模型有两个隐性要求第一单帧检测鲁棒性要高。我们不能靠“这一帧没检出来没关系下一秒再检”来做兜底因为下一秒可能就撞了。第二是检测延迟必须可控。闭眼这个动作如果到第0.5秒才被捕捉到预警的价值就打了折扣。22600张的数据集在训练这样一个实时预警模型时能提供足够的基线效果但无法覆盖所有极端情况。比如夜间几乎全黑的环境、强逆光下司机完全变成剪影、戴着墨镜导致眼睛关键点不可见等。这些情况往往需要额外补充针对性数据或者设计红外补光方案来规避。我自己处理这类问题的做法是先拿这套2万多张的数据训练一版基线模型然后在真实车载环境或者台架上采集一定量的“困难样本”二次微调。这样既发挥了数据集覆盖面广的优势又针对部署场景做了适配。千万别指望任何公开数据一次性满足真实生产环境DMS尤其如此。4. 实操用YOLO训练驾驶员行为检测模型4.1 环境准备与数据目录结构动手之前先把训练环境理一理。我的基础环境组合如下大家可以参照Python 3.10 PyTorch 2.1.2 CUDA 11.8 ultralytics 8.3.x 显卡最低建议RTX 3060 12G或以上数据集的目录结构建议这样组织driver_behavior_dataset/ ├── images/ │ ├── train/ │ ├── val/ │ └── test/ ├── labels/ │ ├── train/ │ ├── val/ │ └── test/ └── driver_behavior.yaml注意YOLO训练时要求图像和标签一一对应最好保持文件名一致只是后缀不同。图片是00001.jpg标注文件就是00001.txt。如果发现有的图片没有对应标签或者反向缺失训练前必须处理掉否则ultralytics会报DatasetError。4.2 一个直接可用的YAML配置数据集的YAML配置文件是训练入口这里给一份可以直接套用的配置path: ./driver_behavior_dataset train: images/train val: images/val test: images/test names: 0: normal_driving 1: drinking 2: texting 3: talking_on_phone 4: operating_radio 5: reaching_behind 6: hair_and_makeup 7: drowsiness 8: smoking 9: yawning如果标签里的类别编号和这个列表对不上要么改YAML要么写脚本统一转换。我的建议是优先改数据端因为names的顺序一旦变了最后部署时的类别映射表也要跟着改容易引入低级错误。4.3 训练启动命令与关键参数训练命令本身不复杂复杂的是参数的选择。一个适合DMS任务的基准训练命令如下yolo detect train \ modelyolov8n.pt \ datadriver_behavior.yaml \ epochs300 \ imgsz640 \ batch32 \ device0 \ patience20 \ cos_lrTrue \ workers8 \ project./runs/driver_behavior \ nameyolov8n_dms_baseline几个参数的选择逻辑我想展开说说。imgsz640是绝大多数YOLO官方权重的主训练尺寸。DMS的图像通常来自车内摄像头分辨率从720P到1080P不等但真正输入模型前都会缩放。640尺寸能保证计算量与精度的平衡。如果你希望提升小目标比如远处的手机屏幕的召回率可以考虑768甚至896但训练速度会下降。batch32在12G显存上搭配yolov8n是可行的如果模型换成更大的比如yolov8m那batch就要降到16甚至8否则会爆显存。patience20是早停策略连续20个epoch验证集指标没有提升就停止训练。这套数据集如果出现某些类别样本严重不足早停能避免越过拟合那个拐点之后还傻傻训练下去。4.4 训练完成后的评估指标怎么读训练结束后会生成result.png和混淆矩阵很多人只盯着mAP值我觉得关键要看的并不是总体mAP。更值得关注的是安全关键类别的单类recall。比如drowsiness这一类如果recall能到95%以上说明绝大多数闭眼状态都能被框出来哪怕precision稍低一点也无妨——DMS产品里漏报的代价远高于误报。误报顶多是给驾驶员一个不必要的提醒漏报则是安全事故。训练日志里还可以看到PR曲线。我建议多关注小目标的AP因为驾驶员行为检测中很多目标框面积本来就小。比如手持手机这个目标框在画面里可能不到整个图像面积的5%如果AP不高说明模型对远距离手持物的识别存在短板需要额外补数据或者调阈值。5. 我总结的DMS数据集使用避坑清单5.1 类别不均衡是最大的敌人22600张数据集中正常驾驶类别的样本数通常会占大头而吸烟、打哈欠这类行为的样本可能只有几百张。如果不做任何处理模型会倾向于把绝大多数目标预测为多数类也就是“正常驾驶”因为这样总损失最小。我常用以下几种手段来缓解不均衡第一做类别加权采样。给少数类的每个样本设置更高的采样权重使得每个epoch里看到的少数类样本更多。ultralytics在训练时支持通过数据集配置或采样器来调整实际操作中也可以用oversample的方式手动干预。第二使用马赛克和复制粘贴增强。我试过把吸烟、打电话这类小目标样本复制粘贴到正常驾驶的背景图中生成额外的合成样本。效果提升显著特别是对训练过程中的正样本多样性有帮助。第三对关键类别使用略微提高的损失权重。这样能让模型更关注少数类即使分类错误的代价变高也值得。5.2 光照与遮挡问题不能靠增强完全解决数据增强能模拟旋转、缩放、平移、翻转但复杂的车内光照变化——比如太阳光从侧窗打入在驾驶员脸上形成高亮区域、隧道内光线骤变、夜间仪表盘反光——这些增强难以真实模拟。一个可行的做法是采用亮度对比度调整增强来近似极端光照更好的方案是用红外训练集或伪灰度图补充。很多量产DMS摄像头本身就是红外方案图像是近红外的灰度图所以如果数据集只有彩色照片推理时最好先做灰度归一化再喂给模型我发现这样能够提升夜间场景下的泛化性。5.3 目标框边缘与遮挡目标的标注习惯DMS场景中人物和物体经常被方向盘、座椅头枕遮挡如果标注时习惯性地将遮挡物也画进框内模型学习到的特征就会包含遮挡物信息泛化性变差。我处理这类情况时统一采用一个标注规范目标框只包含可见部分遮挡严重时宁可漏标也不把遮挡物画进去。同时统一规定小目标比如手机的框必须包含完整屏幕和手部握持区域不能只框屏幕。这些规范如果不在标注阶段统一后面模型输出的框会忽大忽小评估时的IoU指标也难以解释。5.4 训练时默认增强在DMS上的反效果ultralytics默认开启HSV色彩增强和仿射变换。HSV增强在DMS场景需要谨慎调节特别是色相偏移过大时会把肤色变成奇怪的绿色这会让模型对肤色特征的建模出现偏差。肤色在驾驶员检测中是一个非常重要的特征线索尤其是在手部检测和脸部检测中。我实际测试下来的经验是将hsv_h从默认的0.015降到0.005hsv_s保持不变或略微降低hsv_v基本不动。因为DMS的核心是识别人的状态色彩畸变轻微即可过度增强反而有害。5.5 验证集的划分策略适用于DMS数据集的划分方式并不是完全随机的。因为同一段连续视频中前后帧非常相似直接随机划分会将几乎相同的画面同时分到训练集和验证集导致验证指标虚高。推荐的做法是按视频片段聚类划分。对数据集中属于同一拍摄序列的帧做分组整组划分到训练或验证。如果数据集没有标注视频分组信息可以使用图像相似度做近似聚合把相同场景、相同人员、相似光照的帧归为一组。6. 实际项目里的典型问题与排查经验6.1 训练Loss正常下降但检测结果全是乱框这个现象多数和标注坐标格式错误有关。如果标签txt里的坐标是绝对像素而非归一化值模型在训练初期很可能发散了。排查方法很简单import numpy as np label_path xxx.txt data np.loadtxt(label_path).reshape(-1, 5) print(data.min(axis0)) print(data.max(axis0)) # 检查类别ID assert set(data[:, 0].astype(int)) set(range(10))如果发现某个坐标值大于1基本可以判断标注没归一化。处理方式是遍历所有标签文件将坐标除以对应的图片宽高。6.2 验证集mAP很高但实际摄像头测试效果很差出现这种问题首先要查看验证集和测试环境的数据分布差异。比如训练集全是白天光线充足的图片但实际测试在夜间环境进行那效果差是必然的。解决办法是引入夜间数据微调或者对输入图像做预处理如自动白平衡、去噪、对比度增强使输入分布接近白天场景。同时模型的设计要尽量耐光照变化可以考虑在网络前加一个轻量的自适应归一化层。6.3 频繁误报出“使用手机”而实际不在用手机如果模型把正常握方向盘的手误判为手机很可能是数据集中normal_driving类别里“手部靠近方向盘但手持物品”的样本太少。模型学到的不是“使用手机”这个行为而是“手部有物体”这个浅层特征。这个问题的解决思路有两个一是增加负样本的干扰数据在正常驾驶类别中多留一些手持水杯、拿纸巾的样本二是后处理逻辑配合对“使用手机”的报警需要连续若干帧都命中才触发单帧误报不立即提醒。后者在工程上更加可靠也是DMS产品落地时普遍采用的降误报策略。6.4 闭眼检测在戴墨镜时完全失效在驾驶员戴墨镜时眼睛区域的可见度急剧下降闭眼与睁眼的差异极其微弱。如果数据集缺乏墨镜样本的支持模型几乎不可能正确分类。针对这种情况有两条路可走。一条是增加墨镜样本进行专门的数据扩充另一种是绕开眼部状态判断改用头部姿态估计来综合判断疲劳状态。比如当驾驶员头部低头角度过大且持续一段时间即便看不到眼睛也可以判定为疑似疲劳。这两者结合使用在实际工程中才是稳的。7. 最终建议驾驶员行为检测数据集的选用和训练本质上是一次“数据质量、类别结构、工程约束”三者的平衡。22600张的YOLO数据集在这个任务里是一个很好的起步基准但我始终建议把它视为基线而不是终点。真正能让模型在实车上跑得稳的往往是后续针对目标车型、目标摄像头的补采数据以及一套合理可靠的报警策略。我在实际项目里把模型从YOLOv5的baseline换成经过数据清洗和困难样本微调的YOLOv8n之后误报率降低了接近一半漏报率基本保持一致。这个提升完全不是靠调参调出来的而是靠一点点啃数据、理清类别边界、把训练流程中的隐含问题逐个暴露并修正出来的。最后一个小建议训练之前务必抽出半天时间把数据标签全部过一遍用可视化脚本随机抽查几千张。你花在这个环节上的时间会在后面评估、调优、部署阶段成倍地赚回来。
