Ship target detection algorithm of optical remote sensing image based on YOLOv5

CHENG Qian,LI Jia,DU Juan
DOI: https://doi.org/10.12305/j.issn.1001-506x.2023.05.02
2023-01-01
Abstract:When applied to the ship target detection task of optical remote sensing images, you only look once(YOLO)v5 algorithm suffered false positive and false negative a lot on small targets. An improved method based on YOLOv5 is proposed to deal with this situation. Firstly, a module named semantic information enhancement module aiming at extracting shallow features with more semantic information is designed and added into the path aggregation network, so as to enhance the expression ability of small target features. Then, the Swish function is used as the activation function to improve the network’s ability to characterize the nonlinear characteristics of the data and accelerate the convergence speed of the model. Finally, according to the size characteristics of the ship, the large target detection head is removed to reduce the amount of inference calculation and optimize the effect of the detection end. The experiment on the test set shows that compared with the method before the improvement, the proposed method improves the detection accuracy by 5.2% and the inference time is reduced. At the same time of ensuring the real-time detection, the small target discrimination ability of the model is increased.
What problem does this paper attempt to address?