基于改进U-Net的遥感影像建筑物提取方法

专利申请号：CN202211112491.X

公 开 号：CN115393735A

发 明 人：徐雯佳 田力 许志辉 

代 理 人：蔡运红

代理机构：天津翰林知识产权代理事务所(普通合伙)

专利类型：发明专利

申 请 日：20221125

公 开 日：20220914

专利主分类号：G06V20/13

关 键 词：建筑物提取 遥感影像 编码器 遥感影像数据 语义分析模块 注意力 特征信息 特征图 解码器 建筑物特征 高分辨率 连接模块 融合 引入 构建 权重 改进 建筑物 传输 分割 线索 全局 保证 

摘      要：本发明为一种基于改进U‑Net的遥感影像建筑物提取方法，首先获取高分辨率的遥感影像，构建遥感影像数据集；接着，对传统U‑Net进行改进，在编码器中引入全局注意模块和密集连接模块，在编码器与解码器之间引入语义分析模块，得到建筑物提取模型；最后，利用遥感影像数据集对建筑物提取模型进行训练，获得训练后的建筑物提取模型；将训练后的建筑物提取模型用于提取遥感影像中的建筑物。将注意力权重作为密集连接的传输线索，通过对不同注意力模块的注意力特征进行融合，保证建筑物特征提取的完整性；语义分析模块提高了编码器提取的特征图中特征信息的利用率，同时对整个特征图进行融合增强其特征信息，使其获得更好的分割效果。