(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202211020560.4 (22)申请日 2022.08.24 (71)申请人 北京建筑大学 地址 100044 北京市西城区展览馆路1号 (72)发明人 田董炜　王文科　钟华　方超　 (74)专利代理 机构 北京路浩知识产权代理有限 公司 11002 专利代理师 聂俊伟 (51)Int.Cl. G06V 20/10(2022.01) G06V 10/764(2022.01) G06V 10/82(2022.01) G06V 10/40(2022.01) G06V 10/26(2022.01) G06N 20/00(2019.01) (54)发明名称 空心村庄中建筑的识别方法、 装置、 电子设 备及存储介质 (57)摘要 本发明涉及图像处理技术领域， 提供一种空 心村庄中建筑的识别方法、 装置、 电子设备及存 储介质， 包括： 确定目标村庄所在的识别 区域的 图像； 将所述图像输入到建筑分类识别模型中， 得到由所述建筑分类识别模型输出的建筑分类 信息， 所述 建筑分类信息用于表征所述目标村庄 中各建筑的类别； 所述建筑分类识别模型包括语 义分割层和特征提取层。 通过建筑分类识别模型 对图像中的建筑进行分类， 该建筑分类识别模型 包括分割建筑区域的语义分割层和基于注意力 机制对各建筑区域进行特征提取的特征提取层， 通过对区域的分割和特征提取的运算过程， 能够 提取较为准确的图像特征， 提供图像中各建筑分 类的准确性。 权利要求书2页 说明书8页 附图2页 CN 115457384 A 2022.12.09 CN 115457384 A 1.一种空心村庄中建筑的识别方法， 其特 征在于， 包括： 确定目标村庄所在的识别区域的图像； 将所述图像输入到建筑分类识别模型中， 得到由所述建筑分类识别模型输出的建筑分 类信息， 所述建筑分类信息用于表征 所述目标村庄中各建筑的类别； 其中， 所述建筑分类识别模型为将根据样本图像中各建筑的图像特征和所述样本图像 中各建筑的类别标识作为输入， 通过机器学习训练得到的， 用于对图像中各建筑进行分类 的模型； 所述建筑分类识别模型包括语义分割层和特征提取层， 所述语义分割层用于对所述图 像中各建筑区域进行分割； 所述特征提取层， 用于基于注意力机制对各建筑区域进行特征 提取。 2.根据权利要求1所述的空心村庄中建筑的识别方法， 其特征在于， 所述将所述图像输 入到建筑分类识别模型中， 得到由所述建筑分类识别模型输出的建筑分类信息， 包括： 将所述图像输入到语义分割层， 由所述语义分割层输出 各建筑区域； 将各建筑区域输入到特征提取层， 由所述特征提取层基于注意力机制增加提取深度， 根据所述 提取深度对各建筑区域进行 特征提取， 输出 各建筑区域的图像特 征； 将各建筑区域的图像特征输入到所述建筑分类识别模型的分类层， 由所述分类层输出 各建筑区域的建筑分类信息 。 3.根据权利要求2所述的空心村庄中建筑的识别方法， 其特征在于， 所述特征提取层包 括第一运算单元和 第二运算单元， 相应地， 将各建筑区域输入到特征提取层， 由所述特征提 取层基于注意力机制增加 提取深度， 根据所述提取深度对各建筑区域进行特征提取， 输出 各建筑区域的图像特 征， 包括： 将各建筑区域输入到所述第一运算单元， 由所述第一运算单元基于所述提取深度， 根 据各建筑区域的像素值 生成权重矩阵； 将各建筑区域输入到所述第 二运算单元， 由所述第 二运算单元根据各建筑区域的像素 值确定初始图像特 征， 并根据所述初始图像特 征和所述权 重矩阵确定图像特 征。 4.根据权利要求1所述的空心村庄中建筑的识别方法， 其特征在于， 所述确定目标村庄 所在的识别区域的图像， 包括： 采集目标村庄所在的识别区域的原始图像， 对所述原始图像进行正射处理， 确定所述 图像。 5.一种空心村庄中建筑的识别装置， 其特 征在于， 包括： 确定模块， 用于确定目标村庄所在的识别区域的图像； 识别模块， 用于将所述图像输入到建筑分类识别模型中， 得到由所述建筑分类识别模 型输出的建筑分类信息， 所述建筑分类信息用于表征 所述目标村庄中各建筑的类别； 其中， 所述建筑分类识别模型为将根据样本图像中各建筑的图像特征和所述样本图像 中各建筑的类别标识作为输入， 通过机器学习训练得到的， 用于对图像中各建筑进行分类 的模型； 所述建筑分类识别模型包括语义分割层和特征提取层， 所述语义分割层用于对所述图 像中各建筑区域进行分割； 所述特征提取层， 用于基于注意力机制对各建筑区域进行特征 提取。权　利　要　求　书 1/2 页 2 CN 115457384 A 26.根据权利要求5所述的空心村庄中建筑的识别装置， 其特征在于， 所述识别模块具体 用于： 将所述图像输入到语义分割层， 由所述语义分割层输出 各建筑区域； 将各建筑区域输入到特征提取层， 由所述特征提取层基于注意力机制增加提取深度， 根据所述 提取深度对各建筑区域进行 特征提取， 输出 各建筑区域的图像特 征； 将各建筑区域的图像特征输入到所述建筑分类识别模型的分类层， 由所述分类层输出 各建筑区域的建筑分类信息 。 7.根据权利要求6所述的空心村庄中建筑的识别装置， 其特征在于， 所述特征提取层包 括第一运算单元和第二运算单元， 相应地， 所述识别模块在将各建筑区域输入到特征提取 层， 由所述特征提取层基于注意力机制增加 提取深度， 根据所述提取深度对各建筑区域进 行特征提取， 输出 各建筑区域的图像特 征的处理过程中， 具体用于： 将各建筑区域输入到所述第一运算单元， 由所述第一运算单元基于所述提取深度， 根 据各建筑区域的像素值 生成权重矩阵； 将各建筑区域输入到所述第二运算单元， 由所述第二运算单元基于所述提取深度， 根 据各建筑区域的像素值确定初始图像特征， 并根据所述初始图像特征和所述权重矩阵确定 图像特征。 8.根据权利要求6所述的空心村庄中建筑的识别装置， 其特征在于， 所述确定模块具体 用于： 采集目标村庄所在的识别区域的原始图像， 根据所述原始图像进行正射处理， 确定所 述图像。 9.一种电子设备， 包括处理器和存储有计算机程序的存储器， 其特征在于， 所述处理器 执行所述计算机程序时实现权利要求1至4任一项所述空心村庄中建筑的识别方法的步骤。 10.一种处理器可读存储介质， 其特征在于， 所述处理器可读存储介质存储有计算机程 序， 所述计算机程序用于使所述处理器执行权利要求 1至4任一项 所述空心村庄中建筑的识 别方法的步骤。权　利　要　求　书 2/2 页 3 CN 115457384 A 3