基于特征金字塔和距离约束FCN的遥感船舶检测方法技术

本发明专利技术提出一种基于特征金字塔结构全卷积神经网络的遥感船舶检测方法，主要解决现有技术存在水平轮廓包围框互相重叠、覆盖及检出率低的问题。其实现方案为：1.在现有遥感数据集中选取样本图片并裁剪，划分为训练样本和测试样本；2.使用由带像素距离边框最小值约束的类别损失函数和形状损失函数构成的整体损失函数，训练全卷积神经网络；3.将测试样本输入训练好的全卷积网络，输出为各像素点的特征矩阵，并进行坐标变换获得轮廓包围框的坐标；4.对各像素轮廓包围框进行筛选、合并，得到检测结果图及坐标文件。本发明专利技术能对遥感船舶图像生成带有角度的轮廓边界框，检出率高，边界准确，可用于从光学传感器拍摄的遥感图像中提取船舶目标。

【技术实现步骤摘要】
基于特征金字塔和距离约束FCN的遥感船舶检测方法
本专利技术属于图像处理
，涉及一种遥感船舶检测方法，可用于从光学传感器拍摄的遥感图像中提取船舶目标。
技术介绍
遥感图像中的船舶检测，旨在取代繁琐的人工作业，利用深度学习方法，得到相较于传统平行轮廓包围框更为精确的船舶检测结果图。目前的船舶提取方法大致可以分为三类。第一类是基于知识的船舶提取方法，例如：阈值法、模版匹配法；第二类是基于传统机器学习的方法，例如：聚类、支撑矢量机、贝叶斯分类等；这两大类方法船舶提取效果不够理想，总体精度较低。目前更多研究人员致力于基于深度学习学习的船舶检测方法，例如：FasterR-CNN、YOLO等，这些方法利用一些有标记的训练样本，能够在平行轮廓包围框上取得很好的结果，但是对于密集场景，如对船舶的遥感图像来说，平行轮廓包围框的提取会造成较多区域覆盖、重叠，需要更精确的轮廓包围框进行遥感图像中的船舶目标。特征金字塔结构，是利用卷积神经网络各卷积层特征信息的结构，因其包含有自上而下阶段和自下而上阶段两个部分，相较于一般卷积神经网络只对同一个尺寸的图片进行训练，特征金字塔结构对卷积神经网络各层级获得的本文档来自技高网...

【技术保护点】
1.一种基于特征金字塔结构全卷积网络的遥感船舶检测方法，其特征在于，包括如下：(1)从现有遥感数据库中获取图像及类标文件，并对其进行划分构建训练样本集和测试样本集；(2)将用于特征提取的残差网络1至5级残差块的输出层，按照特征金字塔结构的连接方式，与包含有上采样层、特征拼接层、卷积层的四个全卷积模块分别连接，构成全卷积网络；(3)设定全卷积神经网络的损失函数Lloss：(3a)定义船舶类别判断损失函数Ls：

【技术特征摘要】
1.一种基于特征金字塔结构全卷积网络的遥感船舶检测方法，其特征在于，包括如下：(1)从现有遥感数据库中获取图像及类标文件，并对其进行划分构建训练样本集和测试样本集；(2)将用于特征提取的残差网络1至5级残差块的输出层，按照特征金字塔结构的连接方式，与包含有上采样层、特征拼接层、卷积层的四个全卷积模块分别连接，构成全卷积网络；(3)设定全卷积神经网络的损失函数Lloss：(3a)定义船舶类别判断损失函数Ls：其中，Y代表预测结果的集合，Y代表真实结果的集合，Ym、Ym代表集合中单个像素预测结果和其对应的真实结果，n代表Y中元素的个数，dmin为像素点距离上、下、左、右边框距离中的最小值，β为正负样本的平衡系数：(3b)定义几何形状损失函数Lg:其中，R＝(d3+d4)×(d1+d2)为预测面积，d1、d2、d3、d4分别为像素点距离上、下、左、右四边的预测距离；R＝(d3+d4)×(d1+d2)为真实面积，d1、d2、d3、d4为像素点距离上、下、左、右四边的真实距离；w＝min(d3,d3)+min(d4,d4)为R和R重叠部分的宽；h＝min(d1,d1)+min(d2,d2)为R和R重叠部分的高；|R∩R|＝w×h，|R∪R|＝R+R-|R∩R|，λθ为实验选取的权重系数，θ为预测角度，θ为实际角度；(3c)根据(2a)和(2b)得到全卷积神经网络的整体损失函数Lloss：Lloss＝Ls+λgLg其中，λg代表多次实验后测得的权重系数；(4)通过梯度下降法对损失函数Lloss进行优化，得到训练好的全卷积神经网络参数；(5)将测试样本集中的测试样本图片依次输入到训练好的全卷积神经网络中，得到对应的多通道特征矩阵：If＝[S,AABB,∠]，该特征矩阵分别包括单通道像素类别判断信息S，四通道像素距离边框四边的距离信息AABB以及单通道角度信息∠；(6)设置像素类别判断信息的过滤阈值，对特征矩阵进行挑选，当特征点像素类别判断信息超过该阈值时，保留该点的特征矩阵，该特征矩阵含有距离信息和角度信息；当像素类别判断信息低于该阈值时，则舍弃；(7)将通过(5)保留的像素类别判断信息S及每像素点的距离信息和角度信息，通过数值计算的方法，获得对应...

【专利技术属性】
技术研发人员：张向荣，焦李成，侯伟宁，唐旭，朱鹏，周挥宇，马文萍，
申请(专利权)人：西安电子科技大学，
类型：发明
国别省市：陕西,61