郑州大学学报（工学版）2024，Vol.45Issue(3)：38-45,95,9.DOI:10.13705/j.issn.1671-6833.2023.06.009
基于注意力与多级特征融合的YOLOv5算法
Object Detection and Recognition Algorithm Based on YOLOv5 and the Fusion of Attention and Multistage Features
摘要
Abstract
To tackle the problem of low accuracy of detection and recognition for object in complex scenes,YOLOv5 object detection and recognition algorithm based on attention and multistage feature fusion(AMFF)was proposed in this study.The main ideas included adding the proposed dual space directions pyramid split attention(DSD-PSA)mechanism to the backbone network of the traditional YOLOv5s model to enhance the learning of the feature map space and channel information,adopting multistage feature fusion(MFF)structure in the bottleneck network to fuse the features of different branches,increasing richness of the feature and improving the ability to cope with complex scenes.In addition,C3Ghost module and depthwise separable convolution were used to replace C3 module and common convolution to reduce the number of parameters and the complexity of network.Compared with the tradi-tional YOLOv5s algorithm,the mean average accuracy of the proposed algorithm in the VOC2007+2012 data set reached 85%,and the mean average accuracy of the smart retail cabinet commodity identification data set reached 97.2%,which verified the effectiveness and feasibility of the proposed algorithm.关键词
深度学习/YOLOv5s/目标检测/多级特征融合/注意力机制Key words
deep learning/YOLOv5s/object detection/multistage feature fusion/attention mechanism分类
信息技术与安全科学引用本文复制引用
王瑜,毕玉,石健彤,肖洪兵,孙梅..基于注意力与多级特征融合的YOLOv5算法[J].郑州大学学报（工学版）,2024,45(3):38-45,95,9.基金项目
北京市教委-市自然科学基金联合资助项目(KZ202110011015) （KZ202110011015）
