In order to solve the problem that YOLOv5 algorithm is not accurate in locating the target object and local features,a YOLOv5-ABN target detection network based on YOLOv5 is proposed.Firstly,in order to get accurate positioning information,this paper combines spatial attention mechanism and coordinate attention module in convolution attention mechanism.A spatial and coordinated attention module is proposed,which integrates spatial feature information and coordinate information.The spatial and coordinated attention module is added to the Backbone network of YOLOv5.The weighted bidirectional feature pyramid network is introduced into the Neck detection layer,and the staggered box parameters are optimized,which improves the detection accuracy and positioning ability of the network for sample models and local features.The customized data set and Cornell data set are trained with the improved algorithms respectively.The experiment results show that the improved algorithm is better than YOLOv5 algorithm,and the target object is more accurate and more robust without reducing the recognition accuracy.
Key words
YOLOv5 algorithm /
target detection /
attention mechanism /
weighted bidirectional pyramid network
{{custom_sec.title}}
{{custom_sec.title}}
{{custom_sec.content}}
References
[1] 蒋超,张豪章,恩泽惠,等.基于改进YOLOv5s的行人车辆目标检测算法[J].扬州大学学报:自然科学版,2022,25(6):45-49.
[2] 赵元章,耿生玲.基于改进YOLOv5的人脸遮挡物目标检测算法[J].信息通信,2021,34(11):32-35.
[3] 张德春,李海涛,李勋,等.基于CBAM和BiFPN改进YoloV5的渔船目标检测[J].渔业现代化,2022(3):49.
[4] 孙丰刚,王云露,兰鹏,等.基于改进YOLOv5s和迁移学习的苹果果实病害识别方法[J].农业工程学报,2022(11):38.
[5] 吕晓玲,杨胜月,张明路,等.改进YOLOv5网络的鱼眼图像目标检测算法[J].计算机工程与应用,2023,59(6):10.
[6] 向浩,田地.微型计算机与分析仪器的通信技术研究[J].现代科学仪器,2020(6):51-55.
[7] Xia X,Zhai J.Study on Small Target Detection of Construction Site Helmet Based on YOLOv5[C]//Chinese Intelligent Systems Conference.Springer,Singapore,2022.