[发明专利]一种SAR影像水体提取方法、装置、电子设备及存储介质

权利要求书

1.一种SAR影像水体提取方法，其特征在于，包括：

获取待检测SAR影像；

对所述待检测SAR影像进行预处理；

通过水体分割模型，对预处理后的待检测SAR影像进行处理，得到水体分割结果；

其中，所述水体分割模型为增加双注意力机制的DeepLabv3+语义分割模型，并通过包含SAR影像样本和DEM数据的样本数据集训练得到；

所述方法还包括：对水体分割模型进行训练的步骤，具体包括：

步骤S1：建立训练样本数据集；所述训练样本数据集包括SAR影像和外部DEM数据；

步骤S2：采用迁移学习对一个ResNet-50网络进行预训练，利用训练好的ResNet-50网络初始化水体分割模型的模型参数；

步骤S3：将训练样本数据集数据分成多个batch；

步骤S4：将一个batch的数据输入水体分割模型，得到预测结果，利用预测结果和样本标签结果计算损失函数：

其中，为预测为真实标签的概率，γ为参数；

步骤S5：利用损失函数更新水体分割模型的参数；

迭代步骤S3至步骤S5，直至训练样本数据集全部训练完毕；

所述建立训练样本数据集，包括：

采集多幅星载SAR影像；

以已有光学影像作为地理参考基准，通过多模态匹配技术，对每幅星载SAR影像进行控制点匹配，直接获取匹配点地理坐标或投影坐标，并将获取到的匹配点作为区域网平差控制点、连接点，生成具有地理参考的SAR影像；

根据像素值范围确定水体区域的阈值分割方法，对预处理后的SAR影像进行自动水体初步标注；对比同期的光学遥感影像，结合专家解译知识，对错误标记的区域进行人工纠正，得到正样本数据；

使用外部DEM数据模拟SAR影像，将模拟SAR影像中的山体阴影区域作为负样本数据；

将道路数据作为负样本数据；

对正样本数据和负样本数据均进行切片，获取到训练样本数据集；

所述水体分割模型包括：编码器和解码器；所述编码器至少包括：特征提取模块、空间金字塔池化模块、双注意力机制模块和融合模块；

通过水体分割模型，对所述预处理后的待检测SAR影像进行处理，得到水体分割结果；包括：

通过所述特征提取模块，提取预处理后的待检测SAR影像的语义特征图F1和低级语义特征图F2，将语义特征图F1分别输入空间金字塔池化模块和双注意力机制模块，将低级语义特征图F2输入解码器；

通过所述空间金字塔池化模块，对语义特征图F1进行处理，得到不同尺度的特征图，将不同尺度的特征图融合后得到语义特征F3；

通过所述双注意力机制模块，对语义特征图F1进行位置注意力机制和通道注意力机制处理，得到语义特征F4；

通过所述融合模块，对语义特征F3和语义特征F4进行融合，得到高级语义特征F5，将高级语义特征F5输出至解码器；

通过所述解码器，对低级语义特征F2和高级语义特征F5进行融合，得到水体分割结果；

所述特征提取模块采用带空洞的ResNet-50，ResNet-50共包含5个卷积块；第一卷积块为单一卷积层；第二卷积块包括3个顺序连接的残差模块；第三卷积块包括4个顺序连接的残差模块；第四卷积块包括6个顺序连接的残差模块；第五卷积块包括3个顺序连接的残差模块，其中，第五卷积块的卷积方式为空洞卷积；所述残差模块包括：三个顺序连接的卷积核1×1，3×3，1×1和第一加法单元，三个卷积核对输入特征进行处理，输出输入特征的残差，第一加法单元将输入特征的残差和输入特征进行映射相加，输出处理后的输入特征；

通过所述特征提取模块，提取预处理后的待检测SAR影像的语义特征图F1和低级语义特征图F2；包括：

将经过第一卷积块和第二卷积块处理得到特征图作为低层语义特征图F2；

将经过第一卷积块、第二卷积块、第三卷积块、第四卷积块和第二卷积块处理得到特征图作为语义特征图F1。