基于U型注意力控制网络的高分辨率遥感影像建筑物提取方法及装置

摘要

本发明公开了一种基于U型注意力控制网络的高分辨率遥感影像建筑物提取方法，包括：输入遥感影像至编码器；编码器提取不同网格维度层级的特征并生成编码器特征图并输出至转换器；在编码器和解码器的卷积块之间的跳跃连接中串入注意力阀门；转换器提取抽象特征图并输出至解码器；解码器对转换器输入的抽象特征图进行逐级的上采样处理，生成解码器特征图并输出至所述分割单元；分割单元调整输入的解码器特征图的通道数，获取遥感影像中的建筑物分割结果。此外，本发明还公开了一种基于U型注意力控制网络的高分辨率遥感影像建筑物提取装置。采用本发明的U型注意力控制网络在预测性能和结果精度上实现了高分遥感影像建筑物提取质量的有效提升。