[发明专利]空心村庄中建筑的识别方法、装置、电子设备及存储介质

说明书

本发明涉及图像处理技术领域，提供一种空心村庄中建筑的识别方法、装置、电子设备及存储介质，包括：确定目标村庄所在的识别区域的图像；将所述图像输入到建筑分类识别模型中，得到由所述建筑分类识别模型输出的建筑分类信息，所述建筑分类信息用于表征所述目标村庄中各建筑的类别；所述建筑分类识别模型包括语义分割层和特征提取层。通过建筑分类识别模型对图像中的建筑进行分类，该建筑分类识别模型包括分割建筑区域的语义分割层和基于注意力机制对各建筑区域进行特征提取的特征提取层，通过对区域的分割和特征提取的运算过程，能够提取较为准确的图像特征，提供图像中各建筑分类的准确性。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种空心村庄中建筑的识别方法、装置、电子设备及存储介质。

背景技术

空心村整治是盘活空废土地、优化土地利用、改善乡村居住环境的重要战略，为此，要通过技术手段对空心村中的各类建筑进行识别检测。

传统检测手段对操作人员专业性依赖大，费时费力，无法实现大规模自动测量；其次测量数据类型单一，无法快速获取整个区域地貌特征；

高分卫星影像的清晰度对空心村建筑的特征识别造成了一定局限。高分卫星影像中建筑细部特征(损毁建筑的屋面植物、屋面破损等)较小、不够明显，导致识别难度较大，误判率较高，并在模型训练中，产生深层网络结构会出现特征消失等问题。

传统的机器学习提取方法多依靠遥感影像的浅层特征，浅层特征包含光谱、形态学指数等，虽然实现了建筑物自动提取，但是精度不高、受环境影响大、易出现多尺度漏检等问题。

由此可知，目前对空心村中建筑的识别不够准确。

发明内容

针对现有技术存在的问题，本发明提供一种空心村庄中建筑的识别方法、装置、电子设备及存储介质。

第一方面，本发明提供一种空心村庄中建筑的识别方法，包括：

确定目标村庄所在的识别区域的图像；

将所述图像输入到建筑分类识别模型中，得到由所述建筑分类识别模型输出的建筑分类信息，所述建筑分类信息用于表征所述目标村庄中各建筑的类别；

其中，所述建筑分类识别模型为将根据样本图像中各建筑的图像特征和所述样本图像中各建筑的类别标识作为输入，通过机器学习训练得到的，用于对图像中各建筑进行分类的模型；

所述建筑分类识别模型包括语义分割层和特征提取层，所述语义分割层用于对所述图像中各建筑区域进行分割；所述特征提取层，用于基于注意力机制对各建筑区域进行特征提取。

在一个实施例中，所述将所述图像输入到建筑分类识别模型中，得到由所述建筑分类识别模型输出的建筑分类信息，包括：

将所述图像输入到语义分割层，由所述语义分割层输出各建筑区域；

将各建筑区域输入到特征提取层，由所述特征提取层基于注意力机制增加提取深度，根据所述提取深度对各建筑区域进行特征提取，输出各建筑区域的图像特征；

将各建筑区域的图像特征输入到所述建筑分类识别模型的分类层，由所述分类层输出各建筑区域的建筑分类信息。

在一个实施例中，所述特征提取层包括第一运算单元和第二运算单元，相应地，将各建筑区域输入到特征提取层，由所述特征提取层基于注意力机制增加提取深度，根据所述提取深度对各建筑区域进行特征提取，输出各建筑区域的图像特征，包括：

将各建筑区域输入到所述第一运算单元，由所述第一运算单元基于所述提取深度，根据各建筑区域的像素值生成权重矩阵；

将各建筑区域输入到所述第二运算单元，由所述第二运算单元根据各建筑区域的像素值确定初始图像特征，并根据所述初始图像特征和所述权重矩阵确定图像特征。