一种基于改进注意力的Water-Res-Swin遥感图像水体分割方法

一种基于改进注意力的Water‑Res‑Swin遥感图像水体分割方法，包括以下步骤：1.预处理自制数据集，得到水体分割数据集；2.搭建Water‑Res‑Swin模型：分别使用ResNet50网络和Swin Transformer网络进行特征提取，得到多尺度特征图，将相同维度的特征图融合，提高模型的特征表现能力。同时，在模型中嵌入优化代理注意力(OAG)模块，以降低计算复杂度并生成增强的特征表示；3.采用自制水体分割数据集中的训练集及验证集对Water‑Res‑Swin网络模型进行训练与微调，生成分割模型Y；4.将水体分割数据集中的测试集输入到Y中，得到初始分割结果，对初始分割结果进行后处理，修正错误或不确定性，得到最终的水体分割结果。本发明提出的基于改进注意力的Water‑Res‑Swin模型，在减少计算复杂度的同时提高了模型分割精度。