[发明专利]基于CAD图纸的图元识别方法和识别装置

说明书

本发明提供一种基于CAD图纸的图元识别方法和图元识别装置，所述方法包括：将CAD图纸中包含的所有图元划分为多个图组，其中每个图组中包含同一类型的图元；计算所述图组的多个特征变量，所述特征变量用于表征所述图组中包含的图元的形状特征；将所述多个特征变量输入识别模型，以确定所述图组中包含的图元类型。本发明首先基于CAD图纸的图层和颜色对所有图元划分为多个图组，分别计算每个图组中的多个特征变量，最后通过将特征变量输入识别模型，以确定每个图组中的图元对应的类型。特别的，本发明中的特征变量基于图元的数量、长度、角度、连接关系等泛化特征，无需涉及业务特征，可以有效减少人力成本和时间成本，提高图源识别效率和准确度。

技术领域

本发明涉及图元识别技术领域，特别涉及一种基于CAD图纸的图元识别方法、装置、计算机设备和计算机可读存储介质。

背景技术

随着互联网技术的发展，为了更好地快速建立准确的建筑模型信息，智能识别技术已经开始引入到工程建模过程中。通过智能识别CAD图纸，能够更快速更准确的获取用户所需信息，配合具体的业务需求，得到用户需要的数据，快速应用到招投标阶段。因此，如何快速获取准确的CAD图纸信息，是相关设计软件需要解决的一个问题。

现有技术中一般基于业务特征训练机器学习模型进行图元识别。这种情况下，需要程序员对不同业务都有一定程度的了解，根据业务的不同及图形的样式，进行特征的选取及训练。现有技术需要对程序员进行大量的前期业务培训，短时间内难以见到成效，并且非常耗费人力成本和时间成本。另外，基于业务特征的机器识别容易与CAD图纸中形状相似的图形产生拟合，容易出现识别结果不准确的情况。

发明内容

本发明的目的是提供一种简单通用的图元识别技术，从而使操作人员无需掌握大量业务特征就可以轻松处理，以解决现有技术中存在的上述问题。

为实现上述目的，本发明提供一种基于CAD图纸的图元识别方法，包括：

将CAD图纸中包含的所有图元划分为多个图组，其中每个图组中包含同一类型的图元；

计算所述图组的多个特征变量，所述特征变量用于表征所述图组中包含的图元的形状特征；

将所述多个特征变量输入识别模型，以确定所述图组中包含的图元类型。

示例性地，所述将CAD图纸中包含的所有图元划分为多个图组的步骤包括：

获取所述CAD图纸的任一个图层中包含的第一图元；

从所述第一图元中获取具有相同颜色的第二图元；

将所述第二图元划分为同一个图组。

示例性地，所述计算所述图组的多个特征变量包括：

确定所述图组中包含的联通组，获取联通组信息；所述联通组是指彼此相交的图元组成的集合；

获取所述图组中的基本信息、长度信息、图元数量信息、角度信息、连接信息、弧线信息；

根据所述联通组信息、所述基本信息、所述长度信息、所述图元数量信息、所述角度信息、所述连接信息和所述弧线信息计算所述多个特征变量。

示例性地，所述特征变量包括以下一种或多种：

颜色特征，表征所述图组中图元的颜色；

线型特征，表征所述图组中图元的线性；

图组个数特征，表征所述CAD图纸中包含的图组的个数；

长度比值，表征所述图组中所有线段的总长度与所述CAD图纸中所有线段的总长度的比值；

图元数量比值，表征所述图组中的图元数量与所述CAD图纸中的图源数量的比值；