[发明专利]一种图元识别方法、装置、设备及介质

权利要求书

1.一种图元识别方法，其特征在于，所述方法包括：

通过裁剪窗口对目标电力系统的电子图纸进行裁剪，得到多个裁剪子图；

针对每个裁剪子图，将该裁剪子图输入图元检测模型，得到针对该裁剪子图的检测结果，所述检测结果包括该裁剪子图中的图元的图元类型和所述图元在该裁剪子图中的第一图元位置；

根据针对电子图纸的裁剪规则，对针对所有裁剪子图的检测结果进行合并，生成检测图纸，所述检测图纸包括多个图元、每个图元的图元类型和每个图元在检测图纸中的第二图元位置；

针对检测图纸中的每个图元，通过与该图元的图元类型对应的图元旋转角度检测模型对该图元进行检测，得到该图元的旋转角度；

输出每个图元的识别信息，所述识别信息包括图元的图元类型、图元在检测图纸中的第二图元位置和图元的旋转角度。

2.根据权利要求1所述的方法，其特征在于，通过裁剪窗口对目标电力系统的电子图纸进行裁剪，得到多个裁剪子图的步骤包括：

确定所述裁剪窗口的移动步长；

按照所确定的移动步长移动所述裁剪窗口，以在所述电子图纸中进行滑动裁剪，来得到多个裁剪子图；

其中，所述移动步长小于裁剪窗口的边长。

3.根据权利要求1所述的方法，其特征在于，通过以下方式对所述图元检测模型进行训练：

获取多张训练电子图纸；

针对每张训练电子图纸，通过裁剪窗口对该训练电子图纸进行裁剪，得到多个训练裁剪子图，每个训练裁剪子图的尺寸大于目标图元的尺寸；

针对每个训练裁剪子图，将该训练裁剪子图按照预设角度值进行多次旋转，获得在不同旋转角度下的训练裁剪子图；

设置初始图元检测模型的网络输入尺寸，所述网络输入尺寸不小于每个训练裁剪子图的尺寸；

基于在不同旋转角度下的所有训练裁剪子图对所述初始图元检测模型进行训练，得到所述图元检测模型。

4.根据权利要求3所述的方法，其特征在于，基于在不同旋转角度下的所有训练裁剪子图对所述初始图元检测模型进行训练，得到所述图元检测模型的步骤包括：

基于在不同旋转角度下的所有训练裁剪子图对初始图元检测模型进行训练，得到训练损失函数；

确定训练损失函数中的最小值，将所述最小值所对应的初始图元检测模型确定为图元检测模型。

5.根据权利要求1所述的方法，其特征在于，根据针对电子图纸的裁剪规则，对针对所有裁剪子图的检测结果进行合并，生成检测图纸的步骤包括：

根据针对电子图纸的裁剪规则，确定每个裁剪子图在电子图纸中的子图位置；

针对每个裁剪子图，根据该裁剪子图中的图元在该裁剪子图中的第一图元位置以及该裁剪子图在电子图纸中的子图位置，确定图元在电子图纸中的第二图元位置；

根据图元的所述图元类型和图元在电子图纸中的第二图元位置，生成检测图纸。

6.根据权利要求5所述的方法，其特征在于，根据图元的所述图元类型和图元在电子图纸中的第二图元位置，生成检测图纸的步骤包括：

获取电子图纸中检测到的全部图元的图元类型以及对应的第二图元位置；

将全部图元的图元类型绘制在电子图纸中的第二图元位置处；

对电子图纸进行图元位置过滤，生成检测图纸。

7.根据权利要求6所述的方法，其特征在于，对电子图纸进行图元位置过滤，生成检测图纸的步骤包括：

针对电子图纸中的每个第二图元位置执行以下处理：

确定该第二图元位置处存在的图元的数量；

若存在的图元的数量为一，则将所存在的图元确定为在检测图纸中位于该第二图元位置处的图元；

若存在的图元的数量不为一，则判断位于该第二图元位置处于的各图元的图元类型是否相同；

若各图元的图元类型相同，则从所存在的多个图元中任意选择一个图元，并将所选择的图元确定为在检测图纸中位于该第二图元位置处的图元。