[发明专利]基于注意力机制的遥感图像小样本舰船检测方法

摘要

本发明公开了一种基于注意力机制的遥感图像小样本舰船检测方法，并应用于遥感图像小样本舰船检测任务当中。包括：将支持集中的支持图像和查询集中的查询图像均输入至权重共享网络中进行特征提取，分别对应输出目标平均特征图和查询图像特征图；将目标平均特征图进行平均池化后得到目标平均特征图的下采样结果Sk，通过将目标平均特征图的下采样结果作为卷积核，在查询图像特征图做可分离卷积的方式，计算查询集和支持集的相似性，输出注意力特征图；对注意力特征图进行预测，输出检测结果。在提取目标特征阶段，采用多分支并行的网络结构，可以在提取特征的阶段使用标签信息，并添加了注意力模块，提高了网络快速学习新目标特征的能力。