本发明涉及图形处理领域,具体涉及一种图层分类方法、装置、设备及存储介质。
背景技术:
1、在工程制图领域中,当通过cad进行制图时,通常包含多个图层,在cad中,图层就像是含有文字或图形等元素的胶片,一张张按顺序叠放在一起,组合起来形成页面的最终效果。
2、在ai识别cad图纸过程中,需要将这些图层进行分类,如果不进行分类,将对算法带来大量干扰,影响识别精度。现有技术中,通常是人工在构建cad图纸时,按照较为规范的命名规则对图层进行预先命名,此时计算机设备在ai识别cad图纸过程中,可以直接基于图层命名规则对图层进行分类。
3、上述方案只能处理名称相对规范的场景,cad图纸的命名规则并没有统一的标准,并且cad图纸的类别较多,通过图层命名规则对图层进行分类的分类准确率较低。
技术实现思路
1、本申请提供了一种图层分类方法、装置、设备及存储介质,提高了图层分类的准确性,该技术方案如下。
2、一方面,提供了一种图层分类方法,所述方法包括:
3、获取目标图纸数据;
4、通过目标检测模型对所述目标图纸数据进行数据处理,确定目标类型的图元所在的目标检测区域;
5、按照目标图纸数据中各个图元与目标检测区域的相交情况,判断所述目标图纸数据中各个图元是否为所述目标类型;
6、当检测到目标图纸数据中的目标图层中,所述目标类型的图元的数量占比大于指定阈值时,将所述目标图层确定为目标类型的图层。
7、又一方面,提供了一种图层分类装置,所述装置包括:
8、图纸获取模块,用于获取目标图纸数据;
9、检测区域获取模块,用于通过目标检测模型对所述目标图纸数据进行数据处理,确定目标类型的图元所在的目标检测区域;
10、目标类型判断模块,用于按照目标图纸数据中各个图元与目标检测区域的相交情况,判断所述目标图纸数据中各个图元是否为所述目标类型;
11、图层类型确定模块,用于当检测到目标图纸数据中的目标图层中,所述目标类型的图元的数量占比大于指定阈值时,将所述目标图层确定为目标类型的图层。
12、在一种可能的实现方式中,所述检测区域获取模块,包括:
13、图像矩阵获取单元,用于根据所述目标图纸数据的图元分布,获取目标图像矩阵;
14、矩阵区域确定单元,用于通过目标检测模型对所述目标图像矩阵进行数据处理,确定所述目标图像矩阵所对应的矩阵检测区域;
15、检测区域获取单元,用于将所述矩阵检测区域转换至所述目标图纸数据对应的坐标系,以获得所述目标检测区域。
16、在一种可能的实现方式中,所述图像矩阵获取单元,还用于,
17、将所述目标图纸数据中的各个像素点作为矩阵元素,构建所述目标图像矩阵。
18、在一种可能的实现方式中,所述图像矩阵获取单元,还用于,
19、将所述目标图纸数据按照指定大小进行划分为各个目标栅格;
20、根据各个所述目标栅格包含的图元几何关系,确定各个所述目标栅格中的值,以构成所述目标图像矩阵;所述图元几何关系包括相交、平行、垂直中的至少一者。
21、在一种可能的实现方式中,所述检测区域获取模块,还包括:
22、图元属性检测单元,用于检测所述目标图纸数据中各个图元的类型属性,并在目标图纸数据中将目标类型属性的图元删除;
23、其中,所述目标类型属性包括文本标注与尺寸标注中的至少一者。
24、在一种可能的实现方式中,矩阵区域确定单元,还用于,
25、通过目标检测模型对所述目标图像矩阵进行数据处理,确定所述目标图像矩阵所对应的候选检测区域;
26、将所述候选检测区域中,置信度大于置信度阈值的确定为所述矩阵检测区域。
27、在一种可能的实现方式中,所述目标类型判断模块,包括:
28、获取目标图纸数据中,与目标检测区域相交的相交图元;
29、当所述相交图元的长度在指定区间内,获取所述相交图元与所述目标检测区域的第一交点与第二交点;
30、计算第一交点对应的第一距离,以及第二交点对应的第二距离;所述第一距离为所述第一交点与所述目标检测区域的各个端点的距离之和;所述第二距离为所述第二交点与所述目标检测区域的各个端点的距离之和;
31、当所述第一距离与所述第二距离的差值小于距离阈值时,将所述相交图元确定为目标类型。
32、再一方面,提供了一种计算机设备,所述计算机设备中包含处理器和存储器,所述存储器中存储有至少一条指令、至少一段程序、代码集或指令集,所述至少一条指令、至少一段程序、代码集或指令集由处理器加载并执行以实现上述图层分类方法。
33、又一方面,提供了一种计算机可读存储介质,所述存储介质中存储有至少一条指令,所述至少一条指令由处理器加载并执行以实现上述的图层分类方法。
34、再一方面,提供了一种计算机程序产品还提供了一种计算机程序产品或计算机程序,该计算机程序产品或计算机程序包括计算机指令,该计算机指令存储在计算机可读存储介质中。计算机设备的处理器从计算机可读存储介质读取该计算机指令,处理器执行该计算机指令,使得该计算机设备执行上述的图层分类方法。
35、本申请提供的技术方案可以包括以下有益效果:
36、当需要对图层进行分类时,计算机设备可以获取到整体的目标图纸数据,并通过目标检测模型对目标图纸数据进行目标检测,获取到目标类型所对应的图元所在的目标检测区域,计算机设备再在目标图纸数据上获取与目标检测区域相交的图元,并以此判断在目标图纸数据上的各个图元是否为目标类型,当完成了各个图元的判断后,且检测到目标图纸数据的目标图层中,目标类型的图元数量占比大于指定阈值时,则将目标图层确定为目标类型的图层,上述方案在通过图层中包含的图元类型判断图层类型,并且图元分类过程考虑到了整体的图纸数据,提高了对图层的分类准确性。
1.一种图层分类方法,其特征在于,所述方法包括:
2.根据权利要求1所述的方法,其特征在于,所述通过目标检测模型对所述目标图纸数据进行数据处理,确定目标类型的图元所在的目标检测区域,包括:
3.根据权利要求2所述的方法,其特征在于,根据所述目标图纸数据的图元分布,获取目标图像矩阵,包括:
4.根据权利要求2所述的方法,其特征在于,根据所述目标图纸数据的图元分布,获取目标图像矩阵,包括:
5.根据权利要求2所述的方法,其特征在于,所述根据所述目标图纸数据的图元分布,获取目标图像矩阵之前,还包括:
6.根据权利要求2所述的方法,其特征在于,所述通过目标检测模型对所述目标图像矩阵进行数据处理,确定所述目标图像矩阵所对应的矩阵检测区域,包括:
7.根据权利要求1至6任一所述的方法,其特征在于,按照目标图纸数据中各个图元与目标检测区域的相交情况,判断所述目标图纸数据中各个图元是否为所述目标类型,包括:
8.一种图层分类装置,其特征在于,所述装置包括:
9.一种计算机设备,其特征在于,所述计算机设备中包含处理器和存储器,所述存储器中存储有至少一条指令、至少一段程序、代码集或指令集,所述至少一条指令、至少一段程序、代码集或指令集由处理器加载并执行以实现如权利要求1至7任一所述的图层分类方法。
10.一种计算机可读存储介质,所述存储介质中存储有至少一条指令,所述至少一条指令由处理器加载并执行以实现如权利要求1至7任一所述的图层分类方法。