摘要
本发明公开了一种基于注意力增强的任意方向密集舰船目标检测方法,涉及图像处理技术领域,包括:获取原始图像;基于主干特征提取模块,提取原始图像的多尺度特征,得到多个特征图;基于双分支注意力增强模块,学习至少部分特征图中各通道和各空间的重要程度、学习至少部分特征图中各位置信息的重要程度,得到基于双分支注意力增强模块的输出特征图;基于加权双向特征融合网络,通过跨尺度连接操作,将输出特征图的信息进行筛选和融合,得到增强特征图;基于检测器中的分类结构和边界框回归结构,对增强特征图进行检测。本发明能够提升网络对重要特征的提取能力,实现对目标的精确提取和方向估计。
- 单位