一种基于双分支注意力机制的红外弱小目标检测方法

本发明属于目标检测领域，尤其为一种基于双分支注意力机制的红外弱小目标检测方法。具体步骤为：步骤1，准备数据集：将第一红外数据集和第二红外数据集进行数据增强用于预训练，第三红外数据集用于模型微调；步骤2，构建网络模型：整个模型由目标解析路径、拼接路径及目标重建路径组成；本发明通过在目标解析路径的编码模块中引入全局语义增强大核注意力模块，从通道、横向、纵向和感受野四个角度实现对全局信息的获取与利用，并在拼接路径中引入双分支单头注意力模块中获取不同尺度之间的关联程度，达到消除目标解析路径与目标重建路径之间的语义差异的目的，扩大弱小目标与背景的差异，提高红外弱小目标的检测精度。