arXiv:2411.13260cs.CV2024-11被引 12

用先验知识增强局部对比,提升红外小目标检测精度。

Paying more attention to local contrast: improving infrared small target detection performance via prior knowledge

  • 结合人工先验设计局部对比模块,引导模型关注目标区域
  • 在三个公开数据集上超越现有最优方法,检测速度达70帧/秒
  • 模型轻量(194.5万参数),适合边缘设备部署

基于数据驱动的红外小目标检测方法已取得显著进展。然而,由于红外小目标数据集规模小、目标像素极少,深度学习方法直接从中学习极具挑战。利用人类专家知识辅助深度学习,具有探索价值。为此,本文提出一种融合先验知识与数据驱动的U型神经网络LCAE-Net,包含两个改进模块:局部对比增强(LCE)模块和通道注意力增强(CAE)模块。LCE模块利用手工卷积算子提取局部对比注意力(LCA),实现背景抑制并增强潜在目标区域,引导网络聚焦目标位置信息;为有效融合下采样过程中的特征响应,提出CAE模块完成多通道特征融合。实验表明,LCAE-Net在NUDT-SIRST、NUAA-SIRST和IRSTD-1K三个公开数据集上均优于当前最优方法,检测速度可达70 fps。模型参数量为1.945M,浮点运算量(FLOPs)为4.862G,适合部署于边缘设备。

原文摘要 · Abstract (English)

The data-driven method for infrared small target detection (IRSTD) has achieved promising results. However, due to the small scale of infrared small target datasets and the limited number of pixels occupied by the targets themselves, it is a challenging task for deep learning methods to directly learn from these samples. Utilizing human expert knowledge to assist deep learning methods in better learning is worthy of exploration. To effectively guide the model to focus on targets' spatial features, this paper proposes the Local Contrast Attention Enhanced infrared small target detection Network (LCAE-Net), combining prior knowledge with data-driven deep learning methods. LCAE-Net is a U-shaped neural network model which consists of two developed modules: a Local Contrast Enhancement (LCE) module and a Channel Attention Enhancement (CAE) module. The LCE module takes advantages of prior knowledge, leveraging handcrafted convolution operator to acquire Local Contrast Attention (LCA), which could realize background suppression while enhance the potential target region, thus guiding the neural network to pay more attention to potential infrared small targets' location information. To effectively utilize the response information throughout downsampling progresses, the CAE module is proposed to achieve the information fusion among feature maps' different channels. Experimental results indicate that our LCAE-Net outperforms existing state-of-the-art methods on the three public datasets NUDT-SIRST, NUAA-SIRST, and IRSTD-1K, and its detection speed could reach up to 70 fps. Meanwhile, our model has a parameter count and Floating-Point Operations (FLOPs) of 1.945M and 4.862G respectively, which is suitable for deployment on edge devices.

红外检测小目标注意力机制轻量化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。