2 September 2021 Approach for improving YOLOv5 network with application to remote sensing target detection
Shilei Tan, Jun Yan, Ziqiang Jiang, Li Huang
Author Affiliations +
Abstract

Remote sensing images have the characteristics of a multi-scale, high resolution, and complex background, making the accuracy of small target detection too low. We propose an improved method based on the “you-only-look-once” v5 network model to make it more suitable for small target detection. First, we design a four-scale detection layer that improves the depth of the network and the accuracy of small target detection. Then, we introduce complete intersection over union-non-maximum suppression (NMS) to replace the commonly used NMS as a suppression of the prediction bounding box, and it also considers the overlap area, center distance, and aspect ratio to improve the accuracy of predicting bounding boxes. Furthermore, the remote sensing image is segmented into several fixed-size smaller images, making its size suitable for the input of the model to avoid target loss caused by directly scaling the size. The experimental results show that, compared to the four available detection methods, the improved algorithm significantly improves the detection accuracy of targets from small aircraft and ships in complex environments. The miss detection rate is also greatly reduced, especially for small targets, which verifies the effectiveness and robustness of the algorithm. Moreover, the detection rate has been greatly improved, which can meet the demand for real-time detection.

© 2021 Society of Photo-Optical Instrumentation Engineers (SPIE) 1931-3195/2021/$28.00 © 2021 SPIE
Shilei Tan, Jun Yan, Ziqiang Jiang, and Li Huang "Approach for improving YOLOv5 network with application to remote sensing target detection," Journal of Applied Remote Sensing 15(3), 036512 (2 September 2021). https://doi.org/10.1117/1.JRS.15.036512
Received: 21 April 2021; Accepted: 19 August 2021; Published: 2 September 2021
Lens.org Logo
CITATIONS
Cited by 8 scholarly publications.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Target detection

Remote sensing

Detection and tracking algorithms

Image segmentation

Data modeling

Environmental sensing

Evolutionary algorithms

Back to Top