A Real-Time and Lightweight Driver Fatigue Detection Model Using Anchor-Free and Visual-Attention Mechanisms

Ji Wang,Baoming Li,Zhaoye Li,Peiquan Xu,Leijun Li
DOI: https://doi.org/10.1007/s10489-024-05696-4
IF: 5.3
2024-01-01
Applied Intelligence
Abstract:Fatigue driving poses a serious threat to road safety, and research on how to effectively perceive driver fatigue and provide friendly reminders under the limited computing resources of vehicle-mounted platforms has attracted much attention. This study aims to address this limitation by developing edge computing-friendly operators and lightweight network structures. The proposed model, EMFastDet, enhances the efficiency and accuracy of driver fatigue detection. It integrates an attention module within edge computing-friendly operation blocks to capture features of the mouth and eyes. Anchor-free methods and a single detection head layer are employed for position and category predictions. The eye and mouth states in video streams are evaluated based on the metrics of Percentage of Eye Closure (PERCLOS) and Percentage of Yawning (POY). Extensive experiments were conducted using the YFDMS dataset collected in a real driving cabin environment with an infrared camera. Testing on the Qualcomm Snapdragon SA8155P chip, the DSP-accelerated EMFastDet 0.5 × version achieved an inference time of 3.02 ms and a quantized model size of 0.91 MB. The model achieved an mAP0.5 accuracy of 66.1 % , meeting the deployment requirements of in-vehicle platforms.
What problem does this paper attempt to address?