首页 | 本学科首页   官方微博 | 高级检索  
     

基于[KG-*9]Ghost卷积与注意力机制的SAR图像建筑物检测算法
引用本文:严继伟,苏娟,李义红.基于[KG-*9]Ghost卷积与注意力机制的SAR图像建筑物检测算法[J].兵工学报,2022,43(7):1667-1675.
作者姓名:严继伟  苏娟  李义红
作者单位:(火箭军工程大学 核工程学院, 陕西 西安 710025)
摘    要:针对深度卷积神经网络存在模型参数量大、占用内存资源等问题,提出了一种基于轻量化网络的SAR图像建筑物检测算法。首先以旋转目标检测算法R-centernet为基础,将主干网络中的传统卷积替换为Ghost卷积,并构建Ghost-ResNet网络,降低模型参数量;其次提出了融合宽高信息的通道注意力模块,增强网络对图像中显著区域的精确定位能力;使用CARAFE上采样代替网络中的DCN模块,在上采样过程中充分结合特征图信息,提高目标检测能力;最后使用改进的R-centernet算法在旋转标注的SAR图像建筑物数据集上进行训练与测试。实验结果表明,相比于原始R-centernet算法,改进后的算法准确率提高了3.8%,召回率提高了1.2%,检测速度提高了12帧/s。

关 键 词:轻量化网络  SAR图像建筑物检测  旋转目标检测  Ghost卷积  通道注意力  CARAFE上采样  

Building Detection Algorithm in SAR Images Based on Ghost Convolution and Attention Mechanisms
YAN Jiwei,SU Juan,LI Yihong.Building Detection Algorithm in SAR Images Based on Ghost Convolution and Attention Mechanisms[J].Acta Armamentarii,2022,43(7):1667-1675.
Authors:YAN Jiwei  SU Juan  LI Yihong
Affiliation:(College of Nuclear Engineering, Rocket Force University of Engineering, Xi'an 710025, Shaanxi, China)
Abstract:A building detection algorithm in SAR images based on a lightweight network is proposed to address the problems caused by the large quantity of model parameters and memory resources involved in deep convolutional neural networks. First, the algorithm is based on the rotating detector R-centernet, and the traditional convolution in backbone network is replaced by Ghost convolution. Then, the Ghost-ResNet is constructed to reduce the number of model parameters. Second, a channel attention module that fuses width and height information is developed to enhance the network's ability to locate significant regions in the images accurately. An up-sampling method named CARAFE is used to replace the DCN module in the network, and the feature map information is fully combined in the up-sampling process to improve target detection. Finally, the improved R-centernet is used to train and test the SAR rotating building dataset. Based on the experimental results, compared with the R-centernet, the improved algorithm has increased detection accuracy by 3.8%, recall by 1.2%, and detection speed by 12 frames per second.
Keywords:lightweightnetwork  buildingdetectioninSARimage  rotatingtargetdetection  Ghostconvolution  channelattention  CARAFEup-sampling  
点击此处可从《兵工学报》浏览原始摘要信息
点击此处可从《兵工学报》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号