液晶与显示2025,Vol.40Issue(4):617-629,13.DOI:10.37188/CJLCD.2024-0192
基于改进YOLOv8n-Pose的疲劳驾驶检测
Fatigue driving detection based on improved YOLOv8n-Pose
摘要
Abstract
Aiming to address the issues of complex detection processes,numerous parameters,low accuracy,and slow execution speed in current driver fatigue detection algorithms,we propose a lightweight model based on an improved YOLOv8n-Pose.This model optimizes the structure of YOLOv8n-Pose.Firstly,Ghost convolution is introduced into the backbone network to reduce the number of model parameters and unnecessary convolution computations.Secondly,a Slim-neck is introduced to fuse features of different sizes extracted by the backbone network,accelerating network prediction calculations.Additionally,an occlusion-aware attention module(SEAM)is added to the neck part to emphasize the facial region in images and weaken the background,improving keypoint localization accuracy.Finally,a GNSC-Head structure is proposed in the detection head part,which incorporates shared convolution and optimizes the BN layers of traditional convolution with more stable GN layers,effectively saving model parameter space and computational resources.Experimental results show that compared with the original algorithm,the improved YOLOv8n-Pose increases mAP@0.5 by 0.9%,reduces parameter count and computational cost by 50%,and increases FPS by 8%.The final fatigue driving recognition rate reaches 93.5%.Verified through experiments,this algorithm maintains high detection accuracy while being lightweight and effectively recognizes driver status,providing strong support for deployment on vehicle edge devices.关键词
疲劳驾驶检测/深度学习/YOLOv8n-Pose/轻量化/注意力机制Key words
fatigue driving detection/deep learning/YOLOv8n-Pose/light weight/attention mechanism分类
计算机与自动化引用本文复制引用
蔡忠祺,林珊玲,林坚普,吕珊红,林志贤,郭太良..基于改进YOLOv8n-Pose的疲劳驾驶检测[J].液晶与显示,2025,40(4):617-629,13.基金项目
国家重点研发计划(No.2021YFB3600603)Supported by National Key R&D Program of China(No.2021YFB3600603) (No.2021YFB3600603)