基于双向交叉注意力增强的遥感图像船舰目标检测方法

本发明提供了一种基于双向交叉注意力增强的遥感图像船舰目标检测方法，包括：基于ViT算法对输入的遥感图像进行分割，获得多个查找token，并从多个查找token中采样压缩token；基于多头部双向交叉注意力机制，计算注意力权重矩阵A，执行多头自注意力运算更新压缩token；经过多层感知机进行压缩特征token之间的信息共享和更新；基于反向的多头部双向交叉注意力机制，通过复用注意力权重矩阵A将信息共享后的压缩token中的信息传递回查找token；对查找token进行解码生成目标特征图，目标特征图通过目标检测头预测得到船舰边界框和类别概率。本发明有效地增强了目标特征提取能力，从而提高了检测性能的同时，有效减少了计算开销。