基于改进注意力机制的红外小目标检测

Infrared Small Target Detection Based on Improving Attention Mechanism

  • 摘要: 针对红外小目标图像特征提取较难、对比度低等问题,提出基于改进注意力机制的红外目标检测方法。首先在反注意力机制的基础上设计并联双通道的反向注意力机制,一个分支按空间注意力、通道注意力顺序操作进行,另一个分支按通道注意力、空间注意力顺序操作进行,这两个独立的分支并联使用,将它们的输出合并在一起。然后并联双通道反向注意力机制引入到Res2Net,并且在Res2Net增设改进区域强度水平模块。最后损失函数考虑全局约束损失函数和局部约束损失函数。仿真结果表明,本文算法视觉效果较好,精确率、ROC检测指标性能优于其他算法。

     

    Abstract: To address the challenges of difficult feature extraction and low contrast in infrared small-target images, an improved attention mechanism is proposed. First, a parallel dual-channel reverse attention mechanism is designed. Based on the reverse attention concept, one branch processes features in the order of spatial attention followed by channel attention, while the other branch follows the reverse order—channel attention followed by spatial attention. These two independent branches are used together to merge their outputs. Second, a parallel dual-channel reverse attention mechanism is introduced into the Res2Net structure, and an improved region strength level module is added to Res2Net. Third, the loss function considers the global and local constraint loss functions. The simulation results show that improving the attention mechanism has good visual effects, and the accuracy and ROC detection performance are better than those of other algorithms.

     

/

返回文章
返回