基于注意力与多级特征融合的YOLOv5算法OA北大核心CSTPCD

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.