Remote Sensing Image Object Detection Based on Improved YOLOv5

Abstract

Aiming at the problems of complex background, high difficulty of small target detection and high miss detection rate in target detection of satellite remote sensing images, this paper proposes a multiscale target detection model based on YOLOv5 network with attention mechanism. —The feature extraction capability of the backbone network is enhanced by fusing efficient channel attention modules in the backbone network, and the detection head is decoupled and parallel convolution is used to perform classification and regression tasks separately to alleviate the conflict between classification and regression tasks. After experimental validation, the algorithm achieves 74.2% mAP and 64 FPS detection speed on Dior remote sensing dataset. experimental results show that the improved detection algorithm can effectively improve the detection capability of YOLOv5 for small and medium targets in remote sensing images and meet the real-time performance of detection.