Liu Genchao, Wu Kun, Lan Wei, Wu Yunjie
School of Aeronautic Science and Engineering, Beihang University, Beijing 100191, China.
School of Aviation, Beihang University, Beijing 100191, China.
Sensors (Basel). 2025 Aug 6;25(15):4832. doi: 10.3390/s25154832.
Accurately identifying driver fatigue in complex driving environments plays a crucial role in road traffic safety. To address the challenge of reduced fatigue detection accuracy in complex cabin environments caused by lighting variations, we propose YOLO-FDCL, a novel algorithm specifically designed for driver fatigue detection under complex lighting conditions. This algorithm introduces MobileNetV4 into the backbone network to enhance the model's ability to extract fatigue-related features in complex driving environments while reducing the model's parameter size. Additionally, by incorporating the concept of structural re-parameterization, RepFPN is introduced into the neck section of the algorithm to strengthen the network's multi-scale feature fusion capabilities, further improving the model's detection performance. Experimental results show that on the YAWDD dataset, compared to the baseline YOLOv8-S, precision increased from 97.4% to 98.8%, recall improved from 96.3% to 97.5%, mAP@0.5 increased from 98.0% to 98.8%, and mAP@0.5:0.95 increased from 92.4% to 94.2%. This algorithm has made significant progress in the task of fatigue detection under complex lighting conditions. At the same time, this model shows outstanding performance on our self-developed Complex Lighting Driving Fatigue Dataset (CLDFD), with precision and recall improving by 2.8% and 2.2%, respectively, and improvements of 3.1% and 3.6% in mAP@0.5 and mAP@0.5:0.95 compared to the baseline model, respectively.
在复杂驾驶环境中准确识别驾驶员疲劳对道路交通安全至关重要。为应对因光照变化导致复杂驾驶室内环境中疲劳检测准确率降低的挑战,我们提出了YOLO - FDCL,这是一种专门为复杂光照条件下的驾驶员疲劳检测设计的新算法。该算法将MobileNetV4引入主干网络,以增强模型在复杂驾驶环境中提取疲劳相关特征的能力,同时减小模型的参数规模。此外,通过引入结构重参数化的概念,将RepFPN引入算法的颈部,以增强网络的多尺度特征融合能力,进一步提高模型的检测性能。实验结果表明,在YAWDD数据集上,与基线YOLOv8 - S相比,精度从97.4%提高到98.8%,召回率从96.3%提高到97.5%,mAP@0.5从98.0%提高到98.8%,mAP@0.5:0.95从92.4%提高到94.2%。该算法在复杂光照条件下的疲劳检测任务中取得了显著进展。同时,该模型在我们自主开发的复杂光照驾驶疲劳数据集(CLDFD)上表现出色,精度和召回率分别提高了2.8%和2.2%,mAP@0.5和mAP@0.5:0.95分别比基线模型提高了3.1%和3.6%。