面向疲劳驾驶的渐进扁平化超轻量检测模型
DOI:
CSTR:
作者:
作者单位:

山东理工大学机械工程学院 淄博 255022

作者简介:

通讯作者:

中图分类号:

TP391.4; TN911.73

基金项目:

2024年山东省博士后创新项目(SDCX-ZG-202400240)资助


Progressive flattening ultra-lightweight detection model for driver fatigue monitoring
Author:
Affiliation:

School of Mechanical Engineering, Shandong University of Technology,Zibo 255022, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    疲劳驾驶检测对实时性与可靠性要求高,但现有目标检测模型参数量与计算复杂度较大,难以在车载边缘设备上部署。为此,提出面向车载环境的超轻量级检测模型Flat-YOLO,通过渐进式扁平化通道标度降低参数冗余,并设计边界框感知通道自注意力(bounding box-aware channel self-attention,BBoxCSA)与IoU感知多尺度融合(IoU-aware multi-scale fusion,IoUMSF)两类配套补偿模块以增强边缘表征与多尺度融合。在自建驾驶员疲劳检测(driver drowsiness detection,DDD)数据集(9 751张,7类)上,Flat-YOLO相较YOLOv8n参数量减少69.4%,mAP@0.5:0.95持平,精确率和召回率分别提升0.3%和1.0%;相较YOLOv8s参数量减少91.7%,计算量减少76.8%。分类别结果进一步表明,在安全关键类别上,Flat-YOLO对head_low的召回率提升4.9%(0.951 vs 0.902),对open_eyes召回率提升2.2%(0.913 vs 0.891);在细粒度疲劳线索上,mouth的mAP@0.5:0.95提升0.9%(0.479 vs 0.470),yawn的精确率提升3.3%(0.947 vs 0.914)。此外,在公开公开的哈欠检测数据集(yawning detection dataset,YawDD)数据集上,Flat-YOLO的mAP@0.5:0.95达到0.901(YOLOv8n为0.862),验证了所提方法的跨数据源泛化能力。

    Abstract:

    Fatigue driving detection demands high real-time performance and reliability, yet existing object detection models suffer from large parameter counts and high computational complexity, making them difficult to deploy on in-vehicle edge devices. To tackle this problem, this paper proposes Flat-YOLO, an ultra-lightweight detection model tailored for in-vehicle environments, which reduces parameter redundancy through progressive flattened channel scaling and incorporates two complementary modules, bounding box-aware channel self-attention (BBoxCSA) and IoU-aware multi-scale fusion (IoUMSF), to enhance edge representation and multi-scale fusion. On the self-constructed driver drowsiness detection (DDD) dataset (9 751 images, 7 classes), Flat-YOLO reduces parameters by 69.4% compared to YOLOv8n while maintaining comparable mAP@0.5:0.95, with precision and recall improvements of 0.3% and 1.0%, respectively; compared to YOLOv8s, it achieves a 91.7% parameter reduction and 76.8% computational cost reduction. Class-wise results further demonstrate that for safety-critical categories, Flat-YOLO improves recall for head_low by 4.9% (0.951 vs 0.902) and for open_eyes by 2.2% (0.913 vs 0.891); for fine-grained fatigue cues, it enhances mouth′s mAP@0.5:0.95 by 0.9% (0.479 vs 0.470) and yawn′s precision by 3.3% (0.947 vs 0.914). Furthermore, on the public yawning detection dataset (YawDD) dataset, Flat-YOLO achieves an mAP@0.5:0.95 of 0.901 (compared to 0.862 for YOLOv8n), validating the cross-dataset generalization capability of the proposed method.

    参考文献
    相似文献
    引证文献
引用本文

解峪坤,刘强,高名旺,曹万里,马圆捷.面向疲劳驾驶的渐进扁平化超轻量检测模型[J].电子测量技术,2026,49(15):193-205

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-10-23
  • 出版日期:
文章二维码

重要通知公告

①《电子测量技术》期刊收款账户变更公告