Research on small target detection algorithm for UAV inspection scene based on SPD-conv

Ruyi Sun; Hong Fan; Yijun Tang; Zupeng He; Yunlong Xu; Enbao Wu

doi:10.1117/12.3021465

19 February 2024 Research on small target detection algorithm for UAV inspection scene based on SPD-conv

Ruyi Sun, Hong Fan, Yijun Tang, Zupeng He, Yunlong Xu, Enbao Wu

Proceedings Volume 13063, Fourth International Conference on Computer Vision and Data Mining (ICCVDM 2023); 130632T (2024) https://doi.org/10.1117/12.3021465
Event: Fourth International Conference on Computer Vision and Data Mining (ICCVDM 2023), 2023, Changchun, China

Abstract

For the UAV inspection process to detect the target size is small, the distribution of dense and serious obstruction and other reasons leading to the target detection difficult problem, a small target detection algorithm for UAV inspection scene based on SPD-Conv is proposed. SPD-Conv is a convolutional layer of a deep learning model, which contains a space-to-depth layer and a non-step-size convolutional layer, and SPD-Conv is a deep learning model. Conv replaces each step-spanning convolutional and pooling layer to reduce the loss of detail characteristics during downsampling and improve network's performance in recognising small targets. By adding a 160×160 small-target detection head and deleting large-target detection head, SA attention mechanism is added before each detection head to reduce interference and improve network's ability to extract features from small targets. Experimental results on VisDrone2019 dataset show that the improved model has improved misdetection and false detection phenomena. The detection accuracy of improved model is 44%, which is 6.6 percent higher than that of YOLOv5s model, and speed reaches 84.03 FPS. The improved model can fulfil real-time detection requirements and detection accuracy is improved.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Ruyi Sun, Hong Fan, Yijun Tang, Zupeng He, Yunlong Xu, and Enbao Wu "Research on small target detection algorithm for UAV inspection scene based on SPD-conv", Proc. SPIE 13063, Fourth International Conference on Computer Vision and Data Mining (ICCVDM 2023), 130632T (19 February 2024); https://doi.org/10.1117/12.3021465

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