Video-Based Driver Drowsiness Detection With Optimised Utilization of Key Facial Features
Lie Yang,Haohan Yang,Henglai Wei,Zhongxu Hu,Chen Lv
DOI: https://doi.org/10.1109/tits.2023.3346054
IF: 8.5
2024-01-01
IEEE Transactions on Intelligent Transportation Systems
Abstract:Driver drowsiness detection is of great significance in improving driving safety and has been widely studied in recent years. However, some existing methods have not fully utilized the drowsiness-related information, and some methods are susceptible to interference from the redundant information of input data. To address these issues, a video-based driver drowsiness detection method according to the key facial features including facial landmarks and local facial areas (VBFLLFA) is proposed in this paper. In order to fully utilize the key facial features related to drowsiness and exclude the interference of redundant information, the head movement information is obtained through facial landmark analysis and the movement information of eyes and mouth is acquired from the local facial areas. And the spatial filtering based on the common spatial pattern (CSP) algorithm is introduced to improve the discrimination of different classes of samples. To adequately extract the temporal and spatial features, a two-branch multi-head attention (TB-MHA) module is designed in this paper. Furthermore, the center loss with center vector distance penalty is introduced to further improve the discrimination of different classes of samples in the feature space. In addition to two public datasets, we specifically create a novel video-based driver drowsiness detection (VBDDD) dataset to evaluate the effectiveness of our method. The experimental results verify that our method can achieve very excellent performance in driver drowsiness detection tasks.
engineering, electrical & electronic,transportation science & technology, civil