一种基于注意力机制的小目标检测算法

摘要

本发明提出一种基于注意力机制的小目标检测算法，在检测过程中加入了注意力机制，从而将注意力机制应用到不同通道的特征图上，增强对于识别有利的通道，抑制对于识别无用的通道。在检测时，先将图像输入RPN中，得到预选区域，而后经过FPN网络提取各个尺度的特征，将其输入注意力模块，得到新的特征图，最后对这些特征进行分类与边框回归，得到检测结果。