基于Swin Transformer和特征融合的高寒山区建筑物提取方法

本发明公开了一种基于Swin Transformer和特征融合的高寒山区建筑物提取方法，该方法包括以下步骤：获取高分辨率三波段遥感影像，构建误差范围为两个像元的数据标签；对源影像和标签进行影像裁剪并通过旋转、镜像两种方式进行数据增强；针对影像中建筑存在浓重的阴影遮挡建筑物周围的上下文信息的问题，融合彩色空间与归一化彩色空间的差异特征、色彩空间亮度特征、聚类特征和植被阴影特征，对数据集影像阴影进行处理；划分训练集和验证集，构建高精度高寒山区建筑物影像数据集；将模型损失函数更换为二元交叉熵损失函数和平方骰子损失的组合损失函数指导训练，有效提高模型特征提取的能力。