[发明专利]一种基于权重学习的多尺度图像分割方法

说明书

本发明公开了一种基于权重学习的多尺度图像分割方法。本发明设计的神经网络在不同尺度上充分提取了目标特征，充分保证了目标的边缘和形态特征，用可学习的权重来保留有用的特征剔除有噪声的特征。该网络模型主要包括编码和解码两部分，在编码阶段，网络提取不同尺度下的特征，在解码阶段融合多层特征空间的特征，每个增强的特征可以获得类别概率分布图，并且通过可学习的自适应权重对所获得增强特征进行加权得到最终特征。实验表明，应用本发明的方法在道路提取上具有更高的准确率和召回率，在道路外形上更贴近真实道路的轮廓。

技术领域

本发明属于遥感图像处理技术领域，是一种基于权重学习的多尺度图像卷积层特征学习方法，可应用于图像目标识别、图像分割等图像特征提取阶段。

背景技术

特征提取是目标识别和图像分割的重要步骤。目标特征主要有目标的颜色特征、纹理特征、形状特征和空间关系特征。颜色特征是一种全局特征，描述了图像或图像区域所对应的景物的表面性质；纹理特征也是一种全局特征，它也描述了图像或图像区域所对应景物的表面性质；形状特征有两类表示方法，一类是轮廓特征，另一类是区域特征，图像的轮廓特征主要针对物体的外边界，而图像的区域特征则关系到整个形状区域；空间关系特征，是指图像中分割出来的多个目标之间的相互的空间位置或相对方向关系，这些关系也可分为连接/邻接关系、交叠/重叠关系和包含/包容关系等。良好的特征表达，对最终算法的准确性起了非常关键的作用，而且系统主要的计算和测试工作都耗在这一大部分。但实际中一般都是人工完成的，然而，手工地选取特征是一件非常费力、启发式的方法，能不能选取好很大程度上靠经验和运气，而且它的调节需要大量的时间。

深度学习可以自动地学习一些特征，不需要人参与特征的选取过程，它是机器学习中的一个分支，是当今AI领域最热门最前沿的研究内容。它考虑的是用非线性多处理层来学习数据的多尺度特征的抽象表征用反向传播算法对计算模型训练，通过监督或若监督的特征学习和分层特征提取代替手工获取特征。随着网络越深，模型学习到的特征语义信息越高，空间分辨率越低，丢失了浅层的位置特征和形状特征。基于深度学习模型的图像特征提取是目标识别、图像分割等应用的重要步骤，良好的图像特征可极大提示目标识别及图像分割的准确率。然而浅层特征不仅包含物体的位置特征和形状特征，还包含大量的噪声，有效地利用浅层的特征，剔除噪声可以提高目标识别与图像分割的准确率，本发明针对上述现有技术中的不足，对现有的深度学习模型进行了改进。

发明内容

本发明设计了一种基于权重学习的多尺度图像分割方法，用于改善常规卷积特征质量，提升目标识别与图像分割的准确率，增强对不同尺度图像的特征表达能力。该模型在常规的编码-解码结构网络架构后端增加了尺度权重学习特征融合模块，以获得不同级别和不同尺度的特征。解码器用于根据特征级别融合不同级别和不同级别的特征。每个增强特征可以获得类别概率分布图，并且通过可学习的自适应权重对所获得的不同级别的类概率分布图进行加权来获得加权类别概率加权分布图，最后，通过分类器获得最终的分割结果。

本发明实现的具体步骤如下：

1)将样本输入到所构建的模型中进行训练，获得训练好的权重，所述模型包括编码器和解码器；

2)将测试区遥感影像作为输入源输入到训练好的模型中；

3)利用编码器对测试区影像进行特征编码，得到n个不同尺度的池化特征；所述编码器包括五个降采样模块，前两个降采样模块分别包含两个卷积模块和一个池化层，后三个降采样模块包含三个卷积模块和一个池化层，其中卷积模块包含一个BN层，一个3*3卷积核的卷积层和一个ReLu的激活层；

4)将第n层池化特征输入解码器中的尺度模块的第一个尺度子模块中进行解码，得到解码特征，将得到的解码特征与第n-1层池化特征进行融合，得到尺度增强特征一，其中尺度子模块的个数为n，每个尺度子模块包含一个上采样层、三个卷积模块，每个卷积模块包含一个BN层和一个3*3卷积层；