Abstract:In view of the small size and dense distribution of remote sensing image targets, this paper adds a detection head P2 specifically for small-scale targets on the basis of the three detection layers of the original YOLOv5 model, and involves the shallow high-resolution feature map in the subsequent multi-scale feature fusion module. The problem of losing the key feature information of the small-scale target in the process of multiple downsampling is effectively avoided. Firstly, an enhanced multi-scale feature fusion pyramid network DSI-FPN is designed. The FPN+PAN network is optimized by using DepthwiseSparable Convolution and Involution operators with fewer parameters and computations, as well as a spatial attention mechanism to generate feature graphs with richer information for network detection tasks. Secondly, we propose an adaptive channel spatial attention mechanism SCBAM, which introduces a self-attention mechanism into CBAM module to add non-local information to the interaction that originally had only local information, breaks the convolution kernel limit, expands the model receptive field, and improves the feature expression ability of the model. Thirdly, in order to solve the problem of insufficient computing power when deploying the target detector for equipment, we propose a network knowledge distillation framework for joint teachers based on the feature layer. The distillation loss of teacher is designed, and the trend of student online learning is adjusted dynamically by balancing the contributions of teacher network and truth value. The detection accuracy of the student network is obviously improved, and the parameters and model size of the network are effectively reduced. Finally, Comparing with other remote sensing image object detection methods, the experimental results show that the approach presented has better detection effect for small-scale targets of remote sensing images under different lighting conditions. The detection accuracy reached 43.9%, and 7.4% higher than that of the original model. After knowledge distillation, the model parameters are reduced to 1/3 of the original, and the detection accuracy is 40.2%.

Target detection based on multi-scale feature fusion and cross-channel interactive attention mechanism

Aerial images object detection method based on cross-scale multi-feature fusion

Small target detection algorithm for UAV aerial photography based on attention mechanism

An Efficient UAV Image Object Detection Algorithm Based on Global Attention and Multi-Scale Feature Fusion

FE-YOLOv5: Improved YOLOv5 Network for Multi-scale Drone-Captured Scene Detection

Target detection for remote sensing based on the enhanced YOLOv4 with improved BiFPN

Object Detection of Remote Sensing Image Based on Multi-Scale Feature Fusion and Attention Mechanism

Improvement of Yolov5 Target Detection Algorithm Combined with Multi-Scale Feature Fusion

An Aerial Image Detection Algorithm Based on Improved YOLOv5

Multi-scale object detection in UAV images based on adaptive feature fusion

MCF-YOLOv5: A Small Target Detection Algorithm Based on Multi-Scale Feature Fusion Improved YOLOv5

Small Target-YOLOv5: Enhancing the Algorithm for Small Object Detection in Drone Aerial Imagery Based on YOLOv5

Adaptive Feature Fusion and Improved Attention Mechanism-Based Small Object Detection for UAV Target Tracking

An Improved YOLOv5 Model Based on Feature Fusion and Attention Mechanism for Multi-Scale Satellite Recognition

Target Detection of Remote Sensing Image Based on an Improved YOLOv5

A small target detection algorithm based on improved YOLOv5 in aerial image

A Multiscale Information Fusion Network Based on PixelShuffle Integrated With YOLO for Aerial Remote Sensing Object Detection

Small Target Detection Algorithm for UAV Based on Improved YOLOv5

An improved UAV target detection algorithm based on ASFF-YOLOv5s.

An improved YOLOv5 method for large objects detection with multi-scale feature cross-layer fusion network

Small Target Detection for Unmanned Aerial Vehicle Images Based on YOLOv5l