12 April 2023 Single shot multibox detector object detection based on attention mechanism and feature fusion
Xiaoqiang Wang, Kecen Li, Bao Shi, Leixiao Li, Hao Lin, Xinpeng Wang, Jinfan Yang
Author Affiliations +
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.

© 2023 SPIE and IS&T
Xiaoqiang Wang, Kecen Li, Bao Shi, Leixiao Li, Hao Lin, Xinpeng Wang, and Jinfan Yang "Single shot multibox detector object detection based on attention mechanism and feature fusion," Journal of Electronic Imaging 32(2), 023032 (12 April 2023). https://doi.org/10.1117/1.JEI.32.2.023032
Received: 22 December 2022; Accepted: 20 March 2023; Published: 12 April 2023
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Object detection

Feature fusion

Detection and tracking algorithms

Education and training

Data modeling

Target detection

Convolution

Back to Top