基于卷积神经网络和Transformer的河湖遥感图像分割方法及系统

摘要

本发明公开一种基于卷积神经网络和Transformer的河湖遥感图像分割方法及系统，包括：获取包含类别标签的河湖遥感图像，构建训练集；对训练集采用卷积神经网络进行多层特征图的提取；对提取的最后一层特征图采用Transformer的编码器提取注意力特征，对注意力特征采用Transformer的解码器得到自注意力特征图；对自注意力特征图与第一层特征图进行拼接后训练图像分割模型；基于训练后的图像分割模型得到待测河湖遥感图像中目标的类别分割结果。将Transformer引入到遥感图像分割领域，用自注意力机制代替卷积操作，扩大运算时的感受野面积，由于没有下采样和上采样操作，不会造成图像尺度的变化，解决目标丢失问题，克服现有深度学习分割方法在遥感图像分割领域的不足。