改进YOLO11的轻量化驾驶员疲劳检测算法OA
Improved lightweight driver fatigue detection algorithm for YOLO11
针对驾驶员疲劳检测模型存在的参数量、计算量较大,难以在移动设备上部署的问题,提出一种基于深度学习的轻量级驾驶员疲劳检测模型YOLO11-SFL.引入StarNet 作为主干网络来简化网络的结构,以降低整体的参数量和计算成本,提高模型的运行效率;使用 FasterNet Block 模块对 C3k2 进行改进,优化其卷积结构,减少模型冗余计算;此外,采用具有非对称多级压缩技术的LADH 检测头,提升模型检测不同尺寸目标的适应度,在降低模型参数量的同时补偿了轻量化带来的部分精度损失.实验结果表明,提出的疲劳检测模型 YOLO11-SFL 在保持较高检测精度的同时,其参数量(Parameters)、计算量(FLOPs)和模型体积(Size)分别为原 YOLO11 模型的 57.7%、57.1%和60.2%.通过与其他轻量级目标检测模型对比,验证了该方法在轻量化和性能平衡方面的有效性和优越性.
To address the challenges posed by large number of parameters and high computational re-quirements of driver fatigue detection models,which make deployment on mobile devices difficult,a lightweight deep learning-based driver fatigue detection model,YOLO11-SFL,is proposed.The model introduces StarNet as the backbone network to simplify the overall network structure,thereby reducing the total number of parameters and computational costs while improving operational efficiency.A FasterNet Block is adopted to enhance the C3k2,optimizing its convolutional structure to eliminate redundant com-putations.Furthermore,a Lightweight Asymmetric Detection Head(LADH)with multi-level compression techniques is employed to improve the model's adaptability to detect objects of various sizes.This ap-proach reduces the number of model parameters while compensating for some of the accuracy loss associ-ated with lightweight models.Experimental results indicate that the proposed fatigue detection model,YOLO11-SFL,maintains high accuracy while significantly reducing its parameter count,computational load,and model size to 57.7%,57.1%,and 60.2%of those of the original YOLO11 model,respec-tively.Comparisons with other lightweight object detection models validate the effectiveness and superiority of this method in balancing lightweight design and performance.
张洋;朱泽德;李宸翔;王道斌
武汉科技大学 汽车与交通工程学院,湖北 武汉 430065中国科学院 合肥物质科学研究院,安徽 合肥 230031武汉科技大学 汽车与交通工程学院,湖北 武汉 430065武汉科技大学 汽车与交通工程学院,湖北 武汉 430065
交通工程
疲劳检测深度学习轻量化非对称多级压缩技术
fatigue detectiondeep learninglightweightasymmetric multi-level compression technology
《山东理工大学学报(自然科学版)》 2026 (5)
15-21,7
评论