Skip to content
2000
  1. Home
  2. A-Z Publications
  3. Recent Advances in Computer Science and Communications
  4. Fast Track Articles
  5. Fast Track Article
image of Lightweight Research on Fatigue Driving Face Detection Based on YOLOv8

Abstract

Introduction

With the rapid development of society, motor vehicles have become one of the main means of transportation. However, as the number of motor vehicles continues to increase, traffic safety accidents also continue to appear, bringing serious threats to people's lives, and property safety. Fatigue driving is one of the important causes of traffic safety accidents.

Method

To address this problem, a target detection algorithm called VA-YOLO is designed to improve the speed and accuracy of facial recognition for fatigue checking. The algorithm employs a lightweight backbone network, VanillaNet, instead of the traditional backbone network, which reduces the computational and parametric quantities of the model. The SE attention mechanism is also introduced to enhance the model's attention to the target features, which further improves the accuracy of target detection. Finally, in terms of the bounding box regression loss function, the SIoU loss function is used to reduce the error.

Result

The experimental results show that, compared toYolov8n, the VA-YOLO algorithm improves the accuracy by 1.3% while the number of parameters decreases by 30%.

Conclusion

This shows that the VA-YOLO algorithm has a significant advantage in realizing the balance between the number of parameters and accuracy, which is important for improving the speed and accuracy of fatigue driving detection.

© 2025 Bentham Science Publishers
Loading

Article metrics loading...

/content/journals/rascs/10.2174/0126662558315127241210053411
2024-12-23
2025-01-31

  • From This Site

    /content/journals/rascs/10.2174/0126662558315127241210053411
    dcterms_title,dcterms_subject,pub_keyword
    -contentType:Contributor -contentType:Concept -contentType:Institution
    10
    5
Loading full text...

Full text loading...

References

  1. Jianzhe C.H.E.N. Research progress of fatigue driving detection methods. Automobile Practical Technology 2023 48 21 179 186 10.16638/j.cnki.1671‑7988.2023.021.036
    [Google Scholar]
  2. MACLEAN A. W. Sleep and driving Elsevier London 2019 611 622
    [Google Scholar]
  3. Feilin G.O.N.G. The hazards of fatigue driving and preventive measures. Times Automobile 2020 03 94 95
    [Google Scholar]
  4. HASAN M M WATLING C N, LARUE G S.Physiological signal-based drowsiness detection using machine learning: Singular and hybridsignal approaches. J. Safety Res. 2022 80 215 225 10.1016/j.jsr.2021.12.001 35249601
    [Google Scholar]
  5. Gharagozlou F. Nasl Saraji G. Mazloumi A. Nahvi A. Motie Nasrabadi A. Rahimi Foroushani A. Arab Kheradmand A. Ashouri M. Samavati M. Detecting driver mental fatigue based on eeg alpha power changes during simulated driving. Iran. J. Public Health 2015 44 12 1693 1700 26811821
    [Google Scholar]
  6. MU Z. Driving fatigue detecting based on eeg signals of forehead area. Int. J. Pattern Recognit. Artif. Intell. 2017 31 5
    [Google Scholar]
  7. Xin L.I. Hui Z.H.A.N.G. Chaozhong W.U. Driving fatigue detection method based on pulse wave feature fusion. China Journal of Highway 2020 33 6 168 181
    [Google Scholar]
  8. Wang C. Tian Y. Jia H. Driving fatigue detection based on feature fusion of information entropy. JCMSE 2018 18 4 977 988 10.3233/JCM‑180839
    [Google Scholar]
  9. Gang ZHANG Tianjun ZHU Xuemin LI Research on driver fatigue detection based on deep learning and facial multi-feature fusion. Comput. Meas. Control 2023 1 10
    [Google Scholar]
  10. Ahmed M Intelligent driver drowsiness detection for traffic safety based on multi CNN deep model and facial subsampling. IEEE Trans. Intell. Transp. Syst. 2021 23 10 1 10
    [Google Scholar]
  11. Qingmei GUO A review of deep learning based target detection algorithms. J. Detect. Control 2023 45 06 10 20
    [Google Scholar]
  12. Girshick Ross B. Rich feature hierarchies for accurate object detection and semantic segmentation Tech report. 2014 IEEE Conference on Computer Vision and Pattern Recognition, Columbus, OH, USA, 23-28 June 2014, pp. 580-587.
    [Google Scholar]
  13. He K. Deep residual learning for image recognition. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) , Las Vegas, NV, USA, 27-30 June 2016, pp. 770-778.
    [Google Scholar]
  14. Girshick R. Fast R-CNN. 2015 IEEE International Conference on Computer Vision (ICCV), Santiago, Chile, 2015 pp. 1440-1448.
    [Google Scholar]
  15. Sermanet Pierre OverFeat: Integrated recognition, localization and detection using convolutional networks. arXiv:1312.6229 2013
    [Google Scholar]
  16. Wu B. SqueezeDet: Unified, small, low power fully convolutional neural networks for real-time object detection for autonomous driving. 2017 IEEE Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), Honolulu, HI, USA, 21-26 July 2017, pp. 446-454.
    [Google Scholar]
  17. Redmon J. You Only Look Once: Unified, Real-Time Object Detection. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA, 27-30 June 2016, pp. 779-788.
    [Google Scholar]
  18. Jocher G. Chaurasia A. Qiu J. Computer software. 2023 Available from: https://github.com/ultralytics/ultralytics
  19. Ang L.I. Shijie S.U.N. Chaoyang Z.H.A.N.G. Lightweight research on track obstacle detection model with improved YOLOv5s. Computer Engineering and Applications 2023 59 04 197 207
    [Google Scholar]
  20. Jianqi L.I.U. He Y.A.N. Xiaotang W.A.N.G. YOLOv5 target detection network with improved pyramid and hopping connection. Control and Decision 2023 38 6 1730 1736
    [Google Scholar]
  21. Song L.I. Tao S.H.I. Fangke J.I.N.G. Improved road damage detection algorithm for YOLOv8. Computer Engineering and Application 2023 59 23 165 174
    [Google Scholar]
  22. Yunfeng J.I.N. Zhizhan L.U. Ruili W.A.N.G. Fatigue driving detection based on improved YOLOv5s. Journal of Beihua University 2024 25 02 255 261
    [Google Scholar]
  23. Lei ZHAO Lightweight detection algorithm for bottle cap package defects based on YOLOv5. Adv. Laser Optoelectron. 2023 60 22 139 148
    [Google Scholar]
  24. Junfeng H.U. Baicong L.I. Hao Z.H.U. Improved lightweight UAV target detection algorithm for YOLOv8. Available from: http://106.52.218.247:8085/kcms/detail/11.2127.TP.、20240129.1335.026.html
  25. Liu S. Qi L. Qin H. Path aggregation network for instance segmentation. 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA, 18-23 June 2018, pp. 8759-8768.
    [Google Scholar]
  26. Lin T.Y. Dollár P. Girshick R. Feature pyramid networks for object detection. 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA, 21-26 July 2017, pp. 936-944. 2017
    [Google Scholar]
  27. CHEN H. VanillaNet: The power of minimalism in deep learning arXiv:2305.12972 2023
    [Google Scholar]
  28. Chunmei W.A.N.G. Huan L.I.U. YOLOv8-VSC: A lightweight surface defect detection algorithm for strip steel. Computer Science and Exploration 2024 18 01 151 160
    [Google Scholar]
  29. Hu J. Shen L. Albanie S. Sun G. Wu E. Squeeze-and-excitation networks. IEEE Trans. Pattern Anal. Mach. Intell. 2020 42 8 2011 2023 10.1109/TPAMI.2019.2913372 31034408
    [Google Scholar]
  30. Hui W.U. Yuzhu Y.A.N.G. Xianfeng B.U. Research on urban fire detection algorithm based on improved YOLOv5. Available from: http://kns.cnki.net/kcms/detail/13.1097.TN.20240111.1703.006.html
  31. SIoU loss: More powerful learning for bounding box regression. Available from: https://arxiv. org/abs/2205.12740
  32. ZHU L. BiFormer: Vision transformer with bi-level routing attention. 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Vancouver, BC, Canada, 2023, pp. 10323-10333.
    [Google Scholar]
  33. Yang L. SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Networks. Proceedings of the 38th International Conference on Machine Learning, PMLR 139, 2021, pp. 11863-11874.
    [Google Scholar]
  34. Liu Yichao Global attention mechanism: Retain information to enhance channel-spatial interactions. ArXiv 2021
    [Google Scholar]
  35. Ren S. He K. Girshick R. Sun J. Faster R-CNN: Towards real-time object detectionwith region proposal networks. IEEE Trans. Pattern Anal. Mach. Intell. 2017 39 6 1137 1149 10.1109/TPAMI.2016.2577031 27295650
    [Google Scholar]
  36. Redmon J. Farhadi A. YOLOv3: An incremental improvement. arXiv.1804.02767 2018
    [Google Scholar]
  37. Wang C-Y. Bochkovskiy A. Liao H-Y.M. YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors. 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) , Vancouver, BC, Canada, 2023, pp. 7464-7475. 10.1109/CVPR52729.2023.00721
    [Google Scholar]
/content/journals/rascs/10.2174/0126662558315127241210053411
Loading
Lightweight Research on Fatigue Driving Face Detection Based on YOLOv8
Bentham Science Publishers ; https://doi.org/10.2174/0126662558315127241210053411
/content/journals/rascs/10.2174/0126662558315127241210053411
/content/journals/rascs/10.2174/0126662558315127241210053411
Loading

Data & Media loading...


  • Article Type:     Research Article
Keywords: Target detection ; attention mechanism ; lightweight improvement ; fatigue driving

Most Cited Most Cited RSS feed

This is a required field
Please enter a valid email address
Approval was a Success
Invalid data
An Error Occurred
Approval was partially successful, following selected items could not be processed due to error
Please enter a valid_number test