系统工程与电子技术

作者：张涛, 杨小冈, 卢瑞涛, 谢学立, 刘闯

单位：火箭军工程大学导弹工程学院, 陕西 西安 710025

关键词：任意方向舰船检测,中心点估计,SimAM注意力,方向不变模型

基金：国家自然科学基金(61806209);陕西省自然科学基金(2020JQ-490);陕西省自然科学基金(2021JQ-373)

针对当前舰船目标检测算法存在锚框遍历计算成本高和特征旋转适应性不足等问题, 提出基于关键点的遥感图像舰船目标检测方法, 通过预估舰船中心点实现目标检测。首先, 引入深度可分离卷积降低参数冗余, 结合SimAM无参注意力机制, 增强对舰船目标的关注度。其次, 引入方向不变模型(orientation-invariant model, OIM)生成方向不变特征图, 增强网络对旋转目标的适应能力。最后, 考虑到遥感图像舰船目标任意方向密集排列, 但舰船目标中心点不变的特点, 采用直接预测目标的中心点, 再回归偏移量、目标尺度和角度的思路, 摆脱锚框遍历机制, 提高检测速度。在HRSC2016和RFUE2021数据集上进行对比实验, 实验结果充分说明了本文方法的有效性和先进性。

来源：2022年第8期

《系统工程与电子技术》期刊编辑部

查看系统工程与电子技术杂志2022年第8期