改进CenterNet的遥感图像目标检测  

Target Detection of Improved CenterNet to Remote Sensing Images

作　　者：魏玮 杨茹 朱叶 WEI Wei;YANG Ru;ZHU Ye(School of Artificial Intelligence and Data Science,Hebei University of Technology,Tianjin 300401,China)

机构地区：[1]河北工业大学人工智能与数据科学学院,天津300401

出　　处：《计算机工程与应用》2021年第6期191-199,共9页Computer Engineering and Applications

基　　金：国家自然科学基金(61806071);中国科学院自动化研究所模式识别国家重点实验室开放课题(201900043);河北省高等学校科学技术研究项目(QN2019207);天津科技发展战略研究计划(18ZLZXZF00660);河北省自然科学基金面上项目(F2019202381,F2019202464);天津市企业科技特派员项目(19JCTPJC55800,19JCTPJC57000);科技部重点专项计划项目(2018YFC0809700)。

摘　　要：基于深度学习的遥感图像检测在地球资源调查、军事侦察、环境监测等领域有着广泛的应用,更精准、高效的目标检测算法是目前遥感图像检测研究的热点和难点。提出一种改进的CenterNet遥感图像检测算法,对遥感图像进行预处理,以适应CenterNet网络,提高网络对遥感图像的检测有效性;对原网络进行改进,将残差模块中的标准卷积替换成深度可分离卷积,有效降低网络计算量,减少冗余;同时加入注意力机制,抑制无用信息,提高网络的检测准确率。针对遥感图像观测面积大而目标相对较小,目标尺寸差异较大且分布不均匀的特点来说,降低了目标的误检率和漏检率。实验结果表明,改进的CenterNet算法相较于原始CenterNet算法的效果有明显提升,证明了改进算法的鲁棒性。Predictive image detection based on deep learning has a wide range of applications in the fields of earth resource investigation,military reconnaissance,and environmental monitoring.More accurate and efficient target detection algorithms are the hotspots and difficulties of existing image detection research.An improved CenterNet algorithm is proposed to remote sensing image detection,which preprocesses the remote sensing image to adapt to the CenterNet network and improves the detection efficiency of the remote sensing image.The original network is improved,the standard convolution in the residual module is replaced by the depthwise separable convolutions,which effectively reduces the amount of network calculations and reduces redundancy.At the same time,an attention mechanism is added to suppress useless information and improves the accuracy of network detection.In view of the large observation area of the remote sensing image,the relatively small target,the large difference in target size and the uneven distribution,the false detection rate and the missed detection rate of the target are reduced.The experimental results show that the improved CenterNet algorithm has a significant improvement over the original CenterNet algorithm,which proves the robustness of the improved algorithm.

关 键 词：目标检测 遥感图像 深度可分离卷积 注意力机制 Anchor-Free机制 CenterNet算法 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]