logo
基于特征金字塔和距离约束FCN的遥感船舶检测方法
  【资源类型】专利   【发明人】 张向荣 焦李成 侯伟宁 唐旭 朱鹏 周挥宇 马文萍    【公开(公告)日】2019.05.03   【法律状态】实质审查的生效  
【摘要】 本发明提出一种基于特征金字塔结构全卷积神经网络的遥感船舶检测方法,主要解决现有技术存在水平轮廓包围框互相重叠、覆盖及检出率低的问题。其实现方案为:1.在现有遥感数据集中选取样本图片并裁剪,划分为训练样本和测试样本;2.使用由带像素距离边框最小值约束的类别损失函数和形状损失函数构成的整体损失函数,训练全卷积神经网络;3.将测试样本输入训练好的全卷积网络,输出为各像素点的特征矩阵,并进行坐标变换获得轮廓包围框的坐标;4.对各像素轮廓包围框进行筛选、合并,得到检测结果图及坐标文件。本发明能对遥感船舶图像生成带有角度的轮廓边界框,检出率高,边界准确,可用于从光学传感器拍摄的遥感图像中提取船舶目标。

【权利要求】1.一种基于特征金字塔结构全卷积网络的遥感船舶检测方法,其特征在于,包括如下:(1)从现有遥感数据库中获取图像及类标文件,并对其进行划分构建训练样本集和测试样本集;(2)将用于特征提取的残差网络1至5级残差块的输出层,按照特征金字塔结构的连接方式,与包含有上采样层、特征拼接层、卷积层的四个全卷积模块分别连接,构成全卷积网络;(3)设定全卷积神经网络的损失函数L<sub>loss</sub>:(3a)定义船舶类别判断损失函数L<sub>s</sub>:<img file="FDA0001903993860000011.GIF" wi="700" he="70"/>其中,Y代表预测结果的集合,Y代表真实结果的集合,Y<sub>m</sub>、Y<sub>m</sub>代表集合中单个像素预测结果和其对应的真实结果,n代表Y中元素的个数,d<sub>min</sub>为像素点距离上、下、左、右边框距离中的最小值,β为正负样本的平衡系数:<img file="FDA0001903993860000012.GIF" wi="284" he="159"/>(3b)定义几何形状损失函数L<sub>g</sub>:<img file="FDA0001903993860000013.GIF" wi="700" he="141"/>其中,R=(d<sub>3</sub>+d<sub>4</sub>)×(d<sub>1</sub>+d<sub>2</sub>)为预测面积,d<sub>1</sub>、d<sub>2</sub>、d<sub>3</sub>、d<sub>4</sub>分别为像素点距离上、下、左、右四边的预测距离;R=(d<sub>3</sub>+d<sub>4</sub>)×(d<sub>1</sub>+d<sub>2</sub>)为真实面积,d<sub>1</sub>、d<sub>2</sub>、d<sub>3</sub>、d<sub>4</sub>为像素点距离上、下、左、右四边的真实距离;w=min(d<sub>3</sub>,d<sub>3</sub>)+min(d<sub>4</sub>,d<sub>4</sub>)为R和R重叠部分的宽;h=min(d<sub>1</sub>,d<sub>1</sub>)+min(d<sub>2</sub>,d<sub>2</sub>)为R和R重叠部分的高;|R∩R|=w×h,|R∪R|=R+R‑|R∩R|,λ<sub>θ</sub>为实验选取的权重系数,θ为预测角度,θ为实际角度;(3c)根据(2a)和(2b)得到全卷积神经网络的整体损失函数L<sub>loss</sub>:L<sub>loss</sub>=L<sub>s</sub>+λ<sub>g</sub>L<sub>g</sub>其中,λ<sub>g</sub>代表多次实验后测得的权重系数;(4)通过梯度下降法对损失函数L<sub>loss</sub>进行优化,得到训练好的全卷积神经网络参数;(5)将测试样本集中的测试样本图片依次输入到训练好的全卷积神经网络中,得到对应的多通道特征矩阵:I<sub>f</sub>=[S,AABB,∠],该特征矩阵分别包括单通道像素类别判断信息S,四通道像素距离边框四边的距离信息AABB以及单通道角度信息∠;(6)设置像素类别判断信息的过滤阈值,对特征矩阵进行挑选,当特征点像素类别判断信息超过该阈值时,保留该点的特征矩阵,该特征矩阵含有距离信息和角度信息;当像素类别判断信息低于该阈值时,则舍弃;(7)将通过(5)保留的像素类别判断信息S及每像素点的距离信息和角度信息,通过数值计算的方法,获得对应像素点的轮廓包围框,表示为:W=[x<sub>1</sub>,y<sub>1</sub>,x<sub>2</sub>,y<sub>2</sub>,∠]其中x<sub>1</sub>、y<sub>1</sub>为轮廓包围框左上点坐标,x<sub>2</sub>、y<sub>2</sub>为轮廓包围框右下点坐标,∠为角度信息。(8)采用改进的添加合并权重的非极大值抑制方法,对(5)中保留的特征矩阵中像素获得的轮廓包围框进行两两合并,得到船舶检测目标。
【申请人】 西安电子科技大学
【申请号】CN201811524239.3
【申请日】2018.12.13
【分类号】G06K9/00(2006.01)I;G06K9/46(2006.01)I;G06K9/62(2006.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I
【主分类号】G06K9/00(2006.01)I
【公开(公告)号】CN109711288A
【专利代理机构】 陕西电子工业专利中心 61205
【专利代理人】 王品华
【地址】 陕西省西安市雁塔区太白南路2号 (邮编:710071)
【国省】陕西
【说明书】 1 2 3 4 5 6 7 8 9 10 11 12 13
关闭     意见反馈