基于稀疏Transformer的遥感旋转目标检测

作     者：何林远 白俊强 贺旭 王晨 刘旭伦 He Linyuan;Bai Junqiang;He Xu;Wang Chen;Liu Xulun

作者机构：西北工业大学无人系统技术研究院陕西西安710072 空军工程大学航空工程学院陕西西安710038 

出 版 物：《激光与光电子学进展》 (Laser & Optoelectronics Progress)

年 卷 期：2022年第59卷第18期

页      面：45-53页

核心收录：

学科分类：07[理学] 08[工学] 082501[工学-飞行器设计] 070206[理学-声学] 0802[工学-机械工程] 0825[工学-航空宇航科学与技术] 0702[理学-物理学] 

基　　金：国家自然科学基金(61701524,62006245) 中国博士后基金(2019M653742) 

主　　题：图像处理 遥感图像 旋转目标检测 稀疏Transformer Kmeans 

摘      要：针对遥感图像目标广邻域稀疏、多邻域聚集、方向多样等特性导致检测难度大的问题,提出了一种基于稀疏Transformer的遥感旋转目标检测方法。首先,所提方法在典型端到端Transformer网络的基础上,根据遥感图像的特性,利用Kmeans算法实现多域聚集,从而更好提取稀疏域下的目标特征;其次,为适配旋转目标的基本属性,在边框生成阶段,利用目标包围框的中心点及边框特征学习的策略高效获取目标回归斜边框;最后,为提升网络对遥感目标的检测率,对网络的损失函数进行了优化。在DOTA和UCASAOD遥感数据集上的实验结果表明,所提方法的平均精度分别为72.87%和90.4%,能很好地适应遥感图像中各类旋转目标的形状与分布特性。