Sensors (Dec 2024)
Edge-Guided Feature Pyramid Networks: An Edge-Guided Model for Enhanced Small Target Detection
Abstract
Infrared small target detection technology has been widely applied in the defense sector, including applications such as precision targeting, alert systems, and naval monitoring. However, due to the small size of their targets and the extended imaging distance, accurately detecting drone targets in complex infrared environments remains a considerable challenge. Detecting drone targets accurately in complex infrared environments poses a substantial challenge. This paper introduces a novel model that integrates edge characteristics with multi-scale feature fusion, named Edge-Guided Feature Pyramid Networks (EG-FPNs). This model aims to capture deep image features while simultaneously emphasizing edge characteristics. The goal is to resolve the problem of missing target information that occurs when Feature Pyramid Networks (FPNs) perform continuous down-sampling to obtain deeper semantic features. Firstly, an improved residual block structure is proposed, integrating multi-scale convolutional feature extraction and inter-channel attention mechanisms, with significant features being emphasized through channel recalibration. Then, a layered feature fusion module is introduced to strengthen the shallow details in images while fusing multi-scale image features, thereby strengthening the shallow edge features. Finally, an edge self-fusion module is proposed to enhance the model’s depiction of image features by extracting edge information and integrating it with multi-scale features. We conducted comparative experiments on multiple datasets using the proposed algorithm and existing advanced methods. The results show improvements in the IoU, nIoU, and F1 metrics, while also showcasing the lightweight nature of EG-FPNs, confirming that they are more suitable for drone detection in resource-constrained infrared scenarios.
Keywords