基于多尺度特征增强卷积神经网络遥感目标检测算法

周秦汉; 王振

doi:doi:10.3969/j.issn.1671-637x.2022.11.013

电光与控制, 2022, 29 (11): 74, 网络出版: 2023-02-10

基于多尺度特征增强卷积神经网络遥感目标检测算法

A Remote Sensing Target Detection Algorithm Based on Multi-Scale Feature Enhancement CNNs

论文大纲

周秦汉王振

作者单位

空军工程大学信息与导航学院,西安 710000

遥感图像处理目标检测卷积神经网络多尺度特征增强 remote sensing image processing target detection convolutional neural network multi-scale feature enhancement

摘要

随着技术的不断发展, 遥感技术被广泛应用于地图绘制、资源勘探以及灾害预警等领域。遥感目标检测是进行遥感图像解译的关键步骤。传统的目标检测算法在对遥感目标进行检测的过程中存在目标漏检、检测精度低以及无法解决小目标检测等问题。提出一种基于多尺度特征增强卷积神经网络 (MSFE-CNNs) 的遥感目标检测算法, 通过对不同卷积层特征进行增强和融合, 使得模型具有更快的训练速度和更高的检测精度。所提算法结合特征提取模块、特征增强模块、自注意力机制和金字塔特征注意力机制。特征提取模块对输入的海量遥感数据进行特征提取, 获取不同类别目标的多尺度特征; 特征增强模块用于增强不同卷积层特征相关性, 强化模型的学习能力和特征之间的非线性关系; 自注意力机制和金字塔特征注意力机制主要解决传统卷积神经网络无法获取小尺度目标特征的问题。为了验证所提算法的有效性, 在DOTA数据集上进行不同方法对比, 实验结果表明所提算法在检测精度和训练速度上均优于现有基于深度学习的目标检测算法。

Abstract

With the continuous development of remote sensing technology,it is widely used in the fields of map drawing,resource exploration and disaster early-warning.Remote sensing target detection is the key step of remote sensing image interpretation.In the process of detecting remote sensing targets,the traditional detection algorithm has some deficiencies,such as target missing,low detection accuracy and inability to detect small target.A remote sensing target detection algorithm based on Multi-Scale Feature Enhancement Convolution Neural Networks (MSFE-CNNs) is proposed.By enhancing and fusing the features of different convolution layers,the model has faster training speed and higher detection accuracy.The proposed algorithm combines feature extraction module,feature enhancement module,self-attention mechanism and pyramid feature attention mechanism.The feature extraction module extracts features from the input of massive remote sensing data to obtain multi-scale features of different types of targets.The feature enhancement module is used for enhancing the correlation of features of different convolution layers and strengthening the learning ability of the model and the nonlinear relationship between features.Self-attention mechanism and pyramid feature attention mechanism mainly solve the problem that traditional convolutional neural network can not obtain the features of small-scale targets.To verify the effectiveness of the proposed algorithm,different algorithms are compared on DOTA data sets.Experimental results show that the proposed algorithm is superior to the existing target detection algorithms based on deep learning in both detection accuracy and training speed.

参考文献

[1] 李圣琀.基于深度学习的遥感图像中停车场的检测研究［D］.青岛: 青岛大学,2020.

[2] 高立宁,毕福昆,龙腾,等.一种光学遥感图像海面舰船检测算法［J］.清华大学学报(自然科学版),2011,51(1):105-110.

[3] 曹玉社.高分辨率遥感图像中的飞行器目标检测技术研究［D］.长沙: 国防科学技术大学,2016.

[4] BURGER W,BURGE M J.Scale-invariant feature transform (SIFT)［M］//BURGER W,BURGE M J.Digital image processing.London:Springer,2016.

[5] GUEDIRA M R,QADI A E,LRIT M R,et al.A novel method for image categorization based on histogram oriented gradient and support vector machine［C］//International Conference on Electrical and Information Technologies (ICEIT).Rabat:IEEE,2017:1-5.

[6] FELZENSZWALB P F,MCALLESTER D A,RAMANAN D.A discriminatively trained,multiscale,deformable part model［C］//IEEE Conference on Computer Vision and Pattern Recognition.Anchorage:IEEE,2008:1-8.

[7] GUO Y,JIA X,PAULL D.A domain-transfer support vector machine for multi-temporal remote sensing imagery classification［C］//IEEE International Geoscience and Remote Sensing Symposium.Fort Worth:IEEE,2017:2215-2218.

[8] SHEYKHMOUSA M,MAHDIANPARI M,GHANBARI H,et al.Support vector machine versus random forest for remote sensing image classification:a meta-analysis and systematic review［J］.IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing,2020,13:6308-6325.

[9] BARALDI A.Fuzzification of a crisp near-real-time operational automatic spectral-rule-based decision tree preliminary classifier of multisource multispectral remotely sensed images［J］.IEEE Transactions on Geoscience and Remote Sensing,2011,49(6):2113-2134.

[10] YANG J,YE Z W,ZHANG X,et al.Attribute weighted Naive Bayes for remote sensing image classification based on cuckoo search algorithm［C］//International Conference on Security,Pattern Analysis,and Cybernetics (SPAC).Shenzhen:IEEE,2017:169-174.

[11] 李湘眷,王彩玲,李宇,等.窗口融合特征对比度的光学遥感目标检测［J］.光学精密工程, 2016,24(8):2067-2077.

[12] 毕福昆,高立宁,龙腾,等.结合视觉显著性引导与分类器融合的遥感目标检测［J］.红外与激光工程, 2011,40(10):2058-2064.

[13] LIU W,ANGUELOV D,ERHAN D,et al.SSD:single shot multi-box detector［R］.Los Alamos: arXiv Preprint,2015: arXiv:1512.02325.

[14] REDMON J,DIVVALA S,GIRSHICK R,et al.You only look once:unified,real-time object detection［R］.Los Alamos: arXiv Preprint,2016: arXiv:1612.03144.

[15] CAI Z W,VASCONCELOS N.Cascade R-CNN:delving into high quality object detection［R］.Los Alamos: arXiv Preprint,2017: arXiv:1712.00726.

[16] LIN T T,DOLLAR P,GIRSHICK R,et al.Feature pyramid networks for object detection［R］.Los Alamos: arXiv Preprint,2016: arXiv:1612.03144.

[17] SZEGEDY C,VANHOUCKE V,IOFFE S,et al.Rethinking the inception architecture for computer vision［R］.Los Alamos: arXiv Preprint,2015: arXiv:1512.00567.

[18] HE K M,ZHANG X Y,REN S Q,et al.Deep residual learning for image recognition［C］//IEEE Conference on Computer Vision and Pattern Recognition(CVPR).Las Vegas:IEEE,2016:770-778.

[19] CHEN Y P,LI J N,XIAOH X,et al.Dual path networks［R］.Los Alamos: arXiv Preprint,2017:arXiv:1707.01629v2.

[20] HUANG G,LIU Z,LAURENS V,et al.Densely connected convolutional networks［C］//IEEE Conference on Computer Vision and Pattern Recognition (CVPR).Honolulu:IEEE,2017:2261-2269.

[21] LIN T Y,GOYAL P,GIRSHICK R,et al.Focal loss for dense object detection［J］.IEEE Transactions on Pattern Analysis & Machine Intelligence,2020,42(2): 318-327.

[22] REN S Q,HE K M,GIRSHICK R, et al.Faster R-CNN:towards real-time object detection with region proposal networks［J］.IEEE Transactions on Pattern Analysis and Machine Intelligence,2017,39(6):1137-1149.

周秦汉, 王振. 基于多尺度特征增强卷积神经网络遥感目标检测算法[J]. 电光与控制, 2022, 29(11): 74. ZHOU Qinhan, WANG Zhen. A Remote Sensing Target Detection Algorithm Based on Multi-Scale Feature Enhancement CNNs[J]. Electronics Optics & Control, 2022, 29(11): 74.

基于多尺度特征增强卷积神经网络遥感目标检测算法

关于本站 Cookie 的使用提示

全站搜索

基于多尺度特征增强卷积神经网络遥感目标检测算法

相关论文

相关资讯

关于本站 Cookie 的使用提示

全站搜索