[发明专利]基于信源概率加权的多源遥感图像水体提取方法及装置

权利要求书

1.一种基于信源概率加权的多源遥感图像水体提取方法，其特征在于，包括如下步骤：

1)目标图像重叠分块步骤：将目标图像分为多个设定规格的图像块，图像块之间部分重叠；

2)图像块语义分割处理步骤：对每个图像块进行语义分割处理，确定每个图像块中每个像素对应类别标签的概率，所述类别标签至少分为水体标签和背景标签；

3)图像块的信源分类步骤：对每个图像块进行信源分类，得到每个图像块对应的信源概率分布；

4)信源概率加权平均步骤：对目标图像中的每个像素，搜索该像素在其所处的若干个图像块中对应类别标签的概率，并结合信源概率分布进行加权平均计算，得到每个像素的类别标签，从而得到所有像素的类别标签以完成水体提取。

2.根据权利要求1所述的基于信源概率加权的多源遥感图像水体提取方法，其特征在于，所述步骤2)中利用U-Net模型对每个图像块进行语义分割处理，确定每个像素对应类别标签的概率。

3.根据权利要求1所述的基于信源概率加权的多源遥感图像水体提取方法，其特征在于，所述步骤3)中利用AlexNet模型对图像块进行信源分类，得到每个图像块对应的信源概率分布。

4.根据权利要求2所述的基于信源概率加权的多源遥感图像水体提取方法，其特征在于，对U-Net模型进行训练，模型训练过程中采取多次平移、旋转以及变形处理进行样本标签增强，变形处理采取线性移动最小二乘算法。

5.根据权利要求3所述的基于信源概率加权的多源遥感图像水体提取方法，其特征在于，对AlexNet模型进行训练，模型训练过程中采取多次平移、旋转以及变形处理进行样本标签增强，变形处理采取线性移动最小二乘算法。

6.根据权利要求1所述的基于信源概率加权的多源遥感图像水体提取方法，其特征在于，所述步骤1)中将目标图像分为64*64的图像块，分块时横向与纵向均以32像素为步长产生50％的图像重叠。

7.根据权利要求1-6任意一项所述的基于信源概率加权的多源遥感图像水体提取方法，其特征在于，所述步骤4)中加权平均计算过程如下：

其中，N是像素x所在图像块总数；M为信源数量；表示从第j信源得到像素x在第i块中对应类别标签的概率；w_j为第j信源的概率；是最终计算的像素x类别标签。

8.一种基于信源概率加权的多源遥感图像水体提取装置，其特征在于，包括处理器和存储器，以及存储在存储器上并在所述处理器上运行的计算机程序，处理器执行所述计算机程序时实现如权利要求1-7任意一项所述的基于信源概率加权的多源遥感图像水体提取方法。