Lightweight and Real-Time Improvement of Yolov8n Model for Driving Fatigue Detection
To address the challenges of high computational cost, large parameter requirements, and insufficient facial keypoint extraction capability in fatigue driving detection models, this paper proposes an optimized algorithm based on an improved YOLOv8n. A lightweight FCOS architecture is integrated into the Detect head, leveraging shared convolution to significantly reduce the number of parameters while enhancing object localization and classification performance, thereby lowering computational overhead. The model was trained for 100 epochs under the same hardware configuration. Experimental results show that the improved model achieves a mean average precision (mAP) of 80.5 %, representing a notable improvement over the baseline model. Additionally, the computational cost (GFLOPS) and model size (MB) are reduced by 19.7 % and <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$\mathbf{2 2. 2 \%}$</tex>, respectively, while latency decreases by <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$\mathbf{7 0. 2 \%}$</tex>, and detection speed increases by 148 FPS. These results demonstrate that the proposed algorithm holds significant application value in driver facial object detection tasks.
Authors
- Junhui Zhang (ORCID: https://orcid.org/0000-0002-2603-2065)
- Feng Du (ORCID: https://orcid.org/0000-0003-1035-9749)
Institutions
- Tianjin University of Technology and Education (CN)
Publication Details
- Published
- 2025-04-25
- DOI
- https://doi.org/10.1109/iccea65460.2025.11102988
- Primary Topic
- Engineering Applied Research
- Type
- article
- Field-Weighted Citation Impact
- 0.00