Single shot multibox detector object detection based on attention mechanism and feature fusion

Abstract

The single shot multibox detector (SSD) is one of the most important algorithms in single-stage target detection, having the characteristics of multiscale detection and rapid detection speed. However, the effective SSD feature layers are independent of one another, which can lead to object detection difficulties. To address this problem, we proposed an improved SSD object detection algorithm. First, the global attention mechanism (GAM)—which can enhance spatial and channel information—was introduced into the multiscale feature layer. The channel attention module of the GAM was improved. Second, a feature fusion module was introduced to strengthen the relationship between feature layers. Finally, the cross stage partial structure was introduced into the feature fusion module, and used to improve the model’s learning ability. For model training and detection based on the PASCAL VOC dataset, the mean average precision and frames per second obtained by the improved SSD algorithm were 84.67% and 18.67, respectively, which could effectively detect difficult targets.