Dense RFB和LSTM遥感图像舰船目标检测  

Ship detection in remote sensing image based on dense RFB and LSTM

作　　者：张涛 杨小冈 卢孝强[2] 卢瑞涛 张胜修 ZHANG Tao;YANG Xiaogang;LU Xiaoqiang;LU Ruitao;ZHANG Shengxiu(College of Missile Engineering,Rocket Force University of Engineer,Xi’an 710025,China;Xi’an Institute of Optics and Precision Mechanics,Chinese Academy of Sciences,Xi’an 710068,China)

机构地区：[1]火箭军工程大学导弹工程学院,西安710025 [2]中国科学院西安光学精密机械研究所,西安710068

出　　处：《遥感学报》2022年第9期1859-1871,共13页NATIONAL REMOTE SENSING BULLETIN

基　　金：国家自然科学基金(编号:61806209);航空科学基金(编号:201851U8012);陕西省自然科学基础研究计划(编号:2021JQ-373)。

摘　　要：针对当前遥感图像舰船目标检测精度不佳问题,本文构建舰船目标数据集STAR,提出基于Dense RFB和LSTM多尺度舰船目标检测算法。该算法首先在SSD网络基础上设计了浅层特征增强模块,基于人眼视点图采用Dense RFB特征复用和膨胀卷积增大感受野的尺度和种类,增强浅层网络对细节特征的提取能力;其次设计了深层多尺度特征金字塔融合模块,采用FPN和LSTM思想,基于反卷积和残差网络对深层不同尺度特征进行融合,增强网络结构非线性和特征层的表征能力;最后加入聚焦分类损失函数进行联合训练,有效避免了正负样本失衡问题。在遥感图像舰船数据集上实验,本文所提舰船目标检测算法精度均值达到81.98%,检测速度达到29.6帧/s。此外,遥感图像中成像模糊、被遮挡、部分被裁剪等舰船目标的检测效果也优于原有经典算法,实验结果表明该算法对遥感图像舰船目标检测的泛化能力较强,有效地提高了遥感图像舰船目标检测的精度。Ship detection plays a crucial role in various applications and has drawn increasing attention in recent years.Deep learning methods based on CNNs,particularly SSD,have greatly improved detection performance due to their highly efficient feature extraction capability.However,SSD still has two problems.For instance,the detection network of arbitrarily arranged ship targets lacks a connection between high and low-level features and ignores contextual semantic information.Another problem is that natural factors such as light and clouds affect remote sensing images,thus ship detection may cause an imbalance of positive and negative samples.Aiming at solving the above issues,this paper proposes to achieve ship detection in remote sensing images by using a method based on Dense RFB and LSTM.This proposed method includes three elements.First,to enhance the detail feature extraction capability,this proposed method introduces a shallow feature enhancement module.This module draws on the idea of the human viewpoint,which uses Dense RFB feature reuse and expansion convolution to increase the receptive field.Second,to effectively extract deep semantic information and enhance the expressive ability of the network feature layer,a deep multi-scale feature pyramid fusion module(MFPF) is designed,as this proposed method draws on FPN and LSTM deconvolution and residual structure fuse deep multi-scale features.Finally,to solve the imbalance of positive and negative samples,the focal classification loss function is added,improving the accuracy of ship detection during training.The experiments were carried out on an optical remote sensing image dataset,in which only the ship dataset was used for training,validation,and testing.Results indicate that the proposed algorithm achieved an Average Precision(AP) of 81.98% and the detection speed reached 29.6 fps for ship targets,in which most ships were detected successfully.Moreover,for blurred,occluded,and partially-cropped ship targets,the algorithm’s detection effect is better than the

关 键 词：舰船目标检测 Dense RFB 特征金字塔 LSTM 多尺度特征 

分 类 号：TP751[自动化与计算机技术—检测技术与自动化装置] U675.79[自动化与计算机技术—控制科学与工程] E91[交通运输工程—船舶及航道工程]