Optical Remote Sensing Image Object Detection Model Based On Improved YOLOv5s

Abstract

Aiming at the problem of error and omission caused by the complex image background and the high similarity between the remote sensing objectt and the background in remote sensing image detection, an improved model of remote sensing image detection based on YOLOv5s is proposed. The dual attention mechanism of channel and space is added to the convolutional layer of the trunk feature extraction network and the feature fusion network, which enhances the features with high correlation of the objectt to be detected and suppresses the features with low correlation, and improves the ability of the model to extract the features of remote sensing objectts in a complex background, so as to improve the detection accuracy of the algorithm. In this study, a comparative experiment was conducted on the NWPU VHR-10 dataset, and the results showed that the average accuracy of the proposed model was 94% when the dataset was turned over and the ratio was 0.5, which was 2.3% higher than that of the original YOLov5s, which verified the effectiveness of the proposed method.