Infrared small target detection based on coarse-to-fine feature extraction network

Peng Han,Zhengrong Zuo
DOI: https://doi.org/10.1117/12.3004140
2024-03-07
Abstract:Infrared small target detection (IRSTD) is a technique that has been developed to detect small and faint targets in cluttered backgrounds. It shows extensive use in medical diagnosis and aircraft navigation. However, due to the limited feature information and low signal-to-noise ratio, accurately detecting infrared targets and avoiding false alarms is a challenging task. Our study presents a Coarse-to-Fine Feature Extraction Network (CFFENet) for accurately classifying targets and backgrounds in blurred images. This network consists of a Two-stream Region Proposal Network (2s-RPN) and a Region Refinement Attention Network (RRAN), specifically designed to tackle the problem of false alarms. Specifically, 2s-RPN adds edge features to the RPN, in order to enable the network to extract discriminative information between targets and false alarms, so as to provide reliable coarse candidate boxes. In RRAN, firstly, attention-based encoder is used to extract the attention-aware features in candidate boxes, and predicted mask map is output through multi-layer perception (MLP). Subsequently, the refined feature map is further processed by attention encoder, and MLP is utilized to generate the ultimate detection results. The effectiveness of our approach is validated through comparative experiment, which shows its superiority over representative state-of-the-art IRSTD methods.
Computer Science,Engineering
What problem does this paper attempt to address?