[发明专利]构件识别方法、装置、设备及存储介质

说明书

本申请涉及一种构件识别方法、装置、设备及存储介质，该方法包括获取与建筑图纸对应的图纸图像；识别图纸图像中的各建筑空间，并从图纸图像中获取各建筑空间各自的空间数据；利用空间数据和图纸图像，得到各建筑空间各自的构件分类结果。本实施例中由于先对图纸图像进行建筑空间的识别，从而实现可以以建筑空间为单位对图纸图像进行构件的识别，而由于具有相似图像特征的构件通常位于不同的建筑空间内，因此当以建筑空间为单位进行构件的识别时，不容易发生由于图像特征相似导致的构件误识别的问题。

技术领域

本申请涉及图像识别领域，尤其涉及一种构件识别方法、装置、设备及存储介质。

背景技术

建筑图纸的构件识别是对建筑图纸中构成建筑的基本组合物体进行检测和标定的过程，它是机器理解建筑图纸的具体含义，是对建筑图纸进行语义理解的重要基础，在建筑图纸智能审查中发挥着重要作用。

相关技术中，对建筑图纸中的构件识别时，对建筑图纸中的图层进行筛选，获取筛选出的图层里面的各个构件各自的图片，采用预先训练的构件分类模型对各个构件各自的图片进行识别，从而得到各个构件各自的分类结果。

然而，由于属于不同类别的构件可能具有相似的图像特征，因此在采用构件分类模型识别具有相似的图像特征的两类构件时，容易造成误判。以信报箱和空调外机为例，从模型识别的角度，信报箱和空调外机的图像特征类似，因此在采用构件分类模型识别信报箱(空调外机)的图片时，容易误将图片中的信报箱(空调外机)识别成空调外机(信报箱)。

发明内容

本申请提供了一种构件识别方法、装置、设备及存储介质，用以解决相关技术中存在的构件误识别的问题。

第一方面.提供一种构件识别方法，包括：

获取与建筑图纸对应的图纸图像；

识别所述图纸图像中的各建筑空间，并从所述图纸图像中获取所述各建筑空间各自的空间数据，所述空间数据包括空间类别和轮廓信息；

利用所述空间数据和所述图纸图像，得到所述各建筑空间各自的构件分类结果。

可选地，识别所述图纸图像中的各建筑空间，并从所述图纸图像中获取所述各建筑空间各自的空间数据，包括：

将所述图纸图像输入预先训练的空间分类模型中，得到所述各建筑空间各自的空间数据。

可选地，利用所述空间数据和所述图纸图像，得到所述各建筑空间各自的构件分类结果，包括：

根据所述轮廓信息，从所述图纸图像中截取所述各建筑空间各自对应的建筑子图像；

按照空间类别分别识别所述各建筑子图像，得到所述各建筑子图像各自对应的建筑空间的构件分类结果。

可选地，按照空间类别分别识别所述各建筑子图像，得到所述各建筑子图像各自对应的建筑空间的构件分类结果，包括：

对于所述各建筑子图像中的任一建筑子图像，获取所述建筑子图像对应的目标空间类别；

基于所述目标空间类别，确定与所述建筑子图像匹配的构件分类模型；

采用所述构件分类模型识别所述建筑子图像，得到所述建筑子图像对应的建筑空间的构件分类结果。

可选地，基于所述目标空间类别，确定与所述建筑子图像匹配的构件分类模型，包括：

分别获取预先训练的至少一个构件分类模型各自的模型标签，所述模型标签用于指示构件分类模型的分类对象；

从所述模型标签中确定与所述目标空间类别匹配的目标模型标签；

将所述目标模型标签对应的构件分类模型确定为与所述建筑子图像匹配的构件分类模型。