An Improved Distraction Behavior Detection Algorithm Based on YOLOv5  

作　　者：Keke Zhou Guoqiang Zheng Huihui Zhai Xiangshuai Lv Weizhen Zhang 

机构地区：[1]College of Information Engineering,Henan University of Science and Technology,Luoyang,471023,China

出　　处：《Computers, Materials & Continua》2024年第11期2571-2585,共15页计算机、材料和连续体（英文）

基　　金：supported by the National Natural Science Foundation of China(62072158,U2004163);the Key Research and Development Special Projects of Henan Province(231111221500);Science and Technology Project of Henan Province(232102210158,242102210197).

摘　　要：Distracted driving remains a primary factor in traffic accidents and poses a significant obstacle to advancing driver assistance technologies.Improving the accuracy of distracted driving can greatly reduce the occurrence of traffic accidents,thereby providing a guarantee for the safety of drivers.However,detecting distracted driving behaviors remains challenging in real-world scenarios with complex backgrounds,varying target scales,and different resolutions.Addressing the low detection accuracy of existing vehicle distraction detection algorithms and considering practical application scenarios,this paper proposes an improved vehicle distraction detection algorithm based on YOLOv5.The algorithm integrates Attention-based Intra-scale Feature Interaction(AIFI)into the backbone network,enabling it to focus on enhancing feature interactions within the same scale through the attention mechanism.By emphasizing important features,this approach improves detection accuracy,thereby enhancing performance in complex backgrounds.Additionally,a Triple Feature Encoding(TFE)module has been added to the neck network.This module utilizes multi-scale features,encoding and fusing them to create a more detailed and comprehensive feature representation,enhancing object detection and localization,and enabling the algorithm to fully understand the image.Finally,the shape-IoU(Intersection over Union)loss function is adopted to replace the original IoU for more precise bounding box regression.Comparative evaluation of the improved YOLOv5 distraction detection algorithm against the original YOLOv5 algorithm shows an average accuracy improvement of 1.8%,indicating significant advantages in solving distracted driving problems.

关 键 词：Distracted driving YOLOv5 triple feature encoding shape-IoU 

分 类 号：TP39[自动化与计算机技术—计算机应用技术]