FDGformer:基于频域引导Transformer网络的红外小目标检测

FDGformer: Frequency Domain Guided Transformer for Infrared Small Target Detection

  • 摘要: 红外小目标检测旨在从背景复杂的红外图像中检测和识别出尺寸较小的目标,在军事、安防以及无人机等领域有着广泛的应用。该任务的挑战在于红外图像通常分辨率较低、目标对比度低以及纹理模糊,导致小目标很容易被包含噪声和杂波的背景中所淹没。因此,如何准确地检测红外小目标的外形信息仍是目前学术界探索的热点问题。为解决上述问题,提出了一种基于频域信息引导Transformer(FDGformer)网络的红外小目标检测算法。首先采用了流行的U-net架构实现目标掩码的生成,在此基础上基于对红外图像不同层级频率域信息的探索,构建了一种基于Transformer结构的频率信息提取(FIE)模块,能够基于频域计算特征的自注意力,从而对输入特征中的特定频率成分进行增强;接着,将得到的频域增强特征作为引导设计了一种频率信息引导的空间Transformer结构,能够同时整合红外特征的全局依赖关系以及频域显著信息,从而更加准确的识别小目标的外形特征。在公开数据集上的实验结果表明,该算法相比其他先进小目标检测算法有着更高的检测精度,同时参数量更少,有效推动检测任务的实际应用。

     

    Abstract: Infrared small target detection (IRSTD) aims to detect and identify small and dim targets in infrared images with complex backgrounds. It is widely used in security, drones, the military, and other fields. In this task, the challenge is that infrared images usually have low resolution, low target contrast, and blurred textures, causing small targets to be easily lost in backgrounds containing noise and clutter. Therefore, accurate detection of the shape information of small infrared targets is currently an important issue explored by the academic community. To solve these problems, an IRSTD algorithm based on the frequency-domain information-guided transformer (FDGformer) network is proposed. First, the popular U-net architecture is used to generate the target mask. After exploring the frequency domain information at different levels of infrared images, a frequency information extraction (FIE) module based on the Transformer structure is constructed. The self-attention of the features calculated in the frequency domain is used to enhance specific frequency components in the input features. Subsequently, a spatial Transformer structure is designed guided by frequency information and the calculated frequency domain enhanced features, to integrate infrared features. Global dependencies and significant frequency-domain information can accurately identify the appearance characteristics of small targets. Experimental results on public datasets show that this algorithm has higher detection accuracy and fewer parameters than other advanced small-target detection algorithms, effectively promoting the practical application of detection tasks.

     

/

返回文章
返回