一种基于Transformer架构的红外图像目标检测方法

本发明涉及红外图像目标检测领域，提供一种基于Transformer架构的红外图像目标检测方法，该方法包括：通过主干网络提取红外图像中的特征信息，采用逐层卷积的方式，依次获得第一特征、第二特征和第三特征；通过AIFI模块对第三特征进行尺度交互，得到第四特征，将第四特征和第一特征、第二特征分别作为三个输入特征传递到特征融合模块进行融合，获得特征融合结果；通过感知查询模块对特征融合结果进行查询，获得候选框与目标图像间的交互信息；通过解码器对交互信息进行解码，获得候选框；通过检测头对候选框进行分类和定位，获得目标检测结果。本发明实现了具有更小算法规模和更高检测精度的红外图像目标检测方法。