AMFLW-YOLO: A Lightweight Network for Remote Sensing Image Detection Based on Attention Mechanism and Multiscale Feature Fusion
Guili Peng, Zijian Yang, Shoubin Wang, Zhou Yuan
Abstract
The scale of targets in remote sensing images varies greatly and diversity. It has many small targets which distribute densely, and high complexity of image background. The number of network model parameters and the computation amount of the object detection algorithms based on deep learning is huge. It is difficult to apply them on the platform with fixed performance and limited computing resources. A lightweight remote sensing object detection model is proposed in this paper, which called Attention and Multi-Scale Feature Fusion Lightweight-YOLO (AMFLW-YOLO). The deep separable convolution, inverted residual, and linear bottleneck structure are employed to replace the standard convolution layer to reduce the model parameters in the backbone network of the model. The Coordinate Attention (CA) mechanism is introduced into the feature fusion network to capture the direction-aware and location-aware information across channels at the same time, which improves the accuracy of the network. The Bidirectional Feature Pyramid Network (BiFPN) structure is employed to strengthen feature extraction. The learnable weights are introduced to learn the importance of different input features. The multi-scale feature fusion is applied to improve the detection effect. Experimental results show that the algorithm achieves satisfactory performance in terms of efficiency and accuracy and has advantages in detection accuracy and model lightweight.