一种改进Swin Transformer网络的资源三号水体提取方法

针对传统的水体指数法和监督分类法在对卫星影像进行水体提取时经常出现错分的问题,提出一种改进Swin Transformer网络的资源三号影像水体提取方法,并对其特征金字塔的构建进行优化。其中,结合PANet算法原理快速融合特征金字塔顶层信息和底层信息,通过融合不同尺度的信息提高对水体特征提取的效果;利用Swin Transformer网络包含的滑动窗口分割和循环位移操作来减少训练时的计算复杂度。通过混淆矩阵计算得到该方法对城区、郊区和山区水体提取结果的总体精度、Kappa系数、F1-score指数,分别在90.14%～96.52%之间、0.80～0.88之间、0.81～0.87之间,证明该方法对资源三号卫星影像进行水体提取的有效性与可靠性。