An improved small target detection algorithm for SSD

Han Ai; Haifeng Zhang; Long Ren

doi:10.1117/12.2652290

10 November 2022 An improved small target detection algorithm for SSD

Han Ai, Haifeng Zhang, Long Ren

Proceedings Volume 12331, International Conference on Mechanisms and Robotics (ICMAR 2022); 1233108 (2022) https://doi.org/10.1117/12.2652290
Event: International Conference on Mechanisms and Robotics (ICMAR 2022), 2022, Zhuhai, China

Abstract

In order to improve the accuracy and robustness of the small target detection for SSD (Single Shot Multi-Box Detector) algorithm, this paper proposes an improved SSD algorithm. Firstly, the Mish activation function is introduced in the backbone network to provide more linear relationships. Secondly, a copy-reduce-paste data enhancement method is proposed to ensure a more balanced train. In the multi-scale detection stage, feature enhancement module and feature fusion strategy guided by channel attention mechanism are used to improve the feature information. At the same time, modify the loss function to improve the model training effect. Experiments on the VOC2007+2012 dataset show that the detection performance of the proposed algorithm is better than that of the SSD algorithm.

Citation Download Citation

Han Ai, Haifeng Zhang, and Long Ren "An improved small target detection algorithm for SSD", Proc. SPIE 12331, International Conference on Mechanisms and Robotics (ICMAR 2022), 1233108 (10 November 2022); https://doi.org/10.1117/12.2652290

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