YOLO-Extract: Improved YOLOv5 for Aircraft Object Detection in Remote Sensing Images

Abstract

Compared with natural images, remote sensing targets have small and dense target shapes as well as complex target backgrounds. As a result, insufficient detection accuracy and target location cannot be accurately identified. So, this paper proposes the YOLO-extract algorithm based on the YOLOv5 algorithm. Firstly, The YOLO-extract algorithm optimized the model structure of the YOLOv5 algorithm. The YOLO-extract algorithm not only deleted the feature layer and prediction head with poor feature extraction ability but also a new feature extractor with stronger feature extraction ability was integrated into the network. At the same time, YOLO-extract borrowed the idea of residual network to integrate Coordinate Attention into the network. Secondly, the mixed dilated convolution was combined with the redesigned residual structure to enhance the feature and location information extraction ability of the shallow layer of the model and optimize the feature extraction ability of the model for different scale targets. Finally, drawing on the idea of $\alpha $ -IoU Loss, Focal- $\alpha $ EIoU Loss was designed to replace CIoU Loss, which makes the model bounding box regression faster and the loss lower. The experimental results on the test data set show that compared with the YOLOv5 algorithm, the YOLO-extract algorithm has a faster convergence speed, reduces the calculation amount by 45.3GFLOPs and the number of parameters by 10.526M, but increases the mAP by 8.1% and the detection speed by 3 times.

Keywords

• Remote sensing aircraft target
• YOLOv5
• structure optimization
• dilated convolution
• focal-α IoU loss