一种基于注意力机制的遥感图像多尺度目标检测方法

摘要

本发明公开一种基于注意力机制的遥感图像多尺度目标检测方法，针对遥感图像背景复杂、小目标众多，目标尺寸多样化的特点，采用特征金字塔网络进行不同尺度目标的预测。为了更加准确地关注到目标所在区域，在网络中引入注意力机制，有效地提高目标检测精度。其中，包括全局空间注意力模块和像素特征注意力模块。全局空间注意力模块通过从浅层特征中提取空间位置相关信息，并与深层特征融合，强化深层特征的位置表达能力。像素特征注意力模块以多尺度卷积核产生与输入同尺寸的特征图，并以通道注意力的方式为每层特征图赋予权值，得到细节信息表现良好的像素级注意力图在不增加计算量的同时增加了感受野。