Implicit Learning of Rotational Decoupling Detection Heads Applied to Remote Sensing Object Detection
Chengkun Song, Sheng Ding, Yingying Jennifer Chen · 2023
Remote sensing image features complicated background, big image size, small object pixels, and variable object direction due to the arbitrary shooting angle. Conventional object detector has poor detection effect on remote sensing image, which is prone to missed detection and false detection. In view of this, an improved YOLOv5 object detection algorithm was proposed for remote sensing image. First, the C3 module was improved. The output feature information of each convolutional module was transmitted to the deep network, so as to preserve more fine-grained feature information and enhance the small object detection capability. The improved downsampling mechanism integrates convolution downsampling with maximum pooling downsampling mechanism in downs amp ling to separate foreground and background information while retaining feature information. A lightweight rotatory decoupling detection head with implicit learning was proposed to address the conflict between regression and classification, thus improving the network detection capability for objects in any direction. In the meantime, an implicit learning matrix was added to guarantee the lightweight detection head and improve the network detection accuracy for multiple tasks. According to experimental results, under the same model size, compared with the original YOLOv5 detection method, the detection accuracy and detection speed of the proposed method were improved by 8% and 15 frames, or 14.7%, respectively on DOTA-v1.5 data set, effectively improving the detection accuracy and speed in remote sensing image detection. The detection accuracy was increased by 3.5% on coco data set in natural scenario. The method herein can still effectively improve the detection accuracy in natural scenario, proving effectiveness of the modified YOLOv5 method.