基于双向注意力聚合机制的红外小目标检测方法

摘要

本发明公开了一种基于双向注意力聚合机制的红外小目标检测方法，主要解决现有方法检测准确率较低的问题，实现步骤为：构建训练样本集和测试样本集；构建基于双向注意力聚合机制的目标检测网络模型：包括并行排布的第一分支和第二分支；第一分支包括顺次连接的Stem块、第一和第二残差块、第一转置卷积层、第一双向注意力聚合块、第二转置卷积层、第二双向注意力聚合块、head网络；第二分支包括顺次连接的三个泰勒中心差分残差块；双向注意力聚合块包括多个瓶颈结构、多个可变性卷积层和多个非线性激活层；泰勒中心差分残差块包括多个卷积层和多个非线性激活层；对目标检测网络模型进行迭代训练。本发明能够提高红外小目标检测的准确率。

关键词