一种基于改进注意力机制的目标检测方法及装置

摘要

本申请实施例提供了一种基于改进注意力机制的目标检测方法及装置。用于提升目标检测的准确性。所述方法包括：获取第一通道特征信息和第二通道特征信息；其中，第一通道特征信息为进行全局平均池化处理得到的特征信息，第二通道特征信息为进行全局最大池化处理，并经过第一激活函数进行归一化处理之后得到的特征信息；通过多层感知机基于第一通道特征信息和第二通道特征信息生成输出特征信息；将输出特征信息输入第一激活函数得到通道注意力信息；将通道注意力信息与原始特征图合并得到通道特征图；基于空间注意力机制对通道特征图进行处理得到空间注意力信息；将空间注意力信息与通道特征图合并得到目标特征图；基于目标特征图进行目标检测。