一种基于Vision Transformer的高分遥感影像水体提取方法及设备

本发明公开了一种基于Vision Transformer的高分遥感影像水体提取方法及设备，首先获取多光谱遥感影像并进行预处理；在对预处理后的影像进行目视水体解译，制作出影像对应的水体标签真值，制作用于深度学习的数据集；再通过设计空间细节信息保留模块、多尺度线性多头自注意力机制、全局多层感知器，结合Vision Transformer模型，构建MSMViT水体提取方法；通过水体样本数据集对MSMViT模型进行训练和调整，并在测试数据集上检查该水体提取方法的效果，完成多光谱遥感影像的水体提取。能够兼顾高分遥感影像的全局特征和细节特征，取得了更好的水体提取效果。