A mobilenet-SSD target detection algorithm based on attention mechanism

Yilin Li

doi:10.1117/12.2624745

13 December 2021 A mobilenet-SSD target detection algorithm based on attention mechanism

Yilin Li

Proceedings Volume 12087, International Conference on Electronic Information Engineering and Computer Technology (EIECT 2021); 120871K (2021) https://doi.org/10.1117/12.2624745
Event: International Conference on Electronic Information Engineering and Computer Technology (EIECT 2021), 2021, Kunming, China

Abstract

Object detection based on convolutional neural network has important applications in mobile devices, intelligent robots and other fields. However, due to the diversity of traditional models, large number of parameters and slow computing speed, it is difficult to meet the real-time detection requirements of embedded systems. To solve this problem, a mobilenet-SSD model integrating attention mechanism was proposed. Based on the lightweight network model Mobilenet-SSD, the convolution block attention module was integrated into the model, which greatly improved the accuracy of the detection model at the cost of a small increase in computation. Meanwhile, CIOU loss function is used as position loss function to optimize learning strategy and improve training effect. In addition, the processing mode of pre-selection box is improved to Soft-NMS to optimize the detection performance of multi-target overlapping. Experimental results on Pascal VOC2007 and VOC2012 data sets show that the proposed model achieves good performance in recognition accuracy and detection speed, with detection speed of 60.9 FPS and mAP accuracy of 73.6% on GTX1080.The model in this paper has the advantages of small number of parameters, fast computing speed and high identification accuracy, which can better meet the real-time detection requirements of mobile terminals and embedded systems, and provide a good target detection scheme for industrial systems and portable systems.

Citation Download Citation

Yilin Li "A mobilenet-SSD target detection algorithm based on attention mechanism", Proc. SPIE 12087, International Conference on Electronic Information Engineering and Computer Technology (EIECT 2021), 120871K (13 December 2021); https://doi.org/10.1117/12.2624745

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
9 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Target detection

Convolution

Data modeling

Detection and tracking algorithms

Instrument modeling

Performance modeling

Systems modeling

Show All Keywords

Keywords/Phrases

Search In:

Publication Years