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

说明书

本申请提供了一种图元识别方法、装置、设备及介质，其中，该方法包括：通过裁剪窗口对目标电力系统的电子图纸进行裁剪，得到多个裁剪子图；针对每个裁剪子图，将该裁剪子图输入图元检测模型，得到针对该裁剪子图的检测结果；根据针对电子图纸的裁剪规则，对针对所有裁剪子图的检测结果进行合并，生成检测图纸；针对检测图纸中的每个图元，通过与该图元的图元类型对应的图元旋转角度检测模型对该图元进行检测，得到该图元的旋转角度；输出每个图元的识别信息。本申请通过对电子图纸进行图元类型、图元位置的检测生成检测图纸，再对检测图纸进行旋转角度检测确定图元的旋转角度，提升了对电子图纸中图元检测的准确性和速度。

技术领域

本申请涉及电力系统技术领域，具体而言，涉及一种图元识别方法、装置、设备及介质。

背景技术

针对电力调度控制系统一般采用通用信息模型(Common Information Modle,CIM)格式文件作为描述电力系统电气设备、场站图形等设备模型的画面，调度运维人员需要采用人工绘制、录入的方式，参考厂站电子图纸的设计原图绘制CIM画面，并开展电气设备建模工作，由于图形样式复杂，设备类型众多，导致维护工作繁琐，同时极易出现属性缺失、关联错误、图形样式输入错误等问题。

为了解决人工绘制图纸过于繁琐的问题，会预先对电子图纸中的图元进行识别。现有技术中的图元识别过程为：针对每个类型、每种角度的图元都预先设置了一个模版，然后在电子图纸中基于模板匹配的方式来搜索是否存在图元以及图元的角度。当上述图元识别方式存在以下缺陷：

(1)执行速度极慢，针对一张电子图纸，通常按照滑动窗口的方式逐一的采用模板匹配的方式来对比，时间成本非常高；

(2)准确率低，上述图元识别方式抗干扰性弱，只要与标准模板稍有不同，就无法识别；

(3)适应性差，无法适应不同尺寸的输入图像。

发明内容

有鉴于此，本申请的目的在于提供一种图元识别方法、装置、设备及介质，以克服上述至少一个缺陷。

第一方面，本申请实施例提供了一种图元识别方法，所述方法包括：通过裁剪窗口对目标电力系统的电子图纸进行裁剪，得到多个裁剪子图；针对每个裁剪子图，将该裁剪子图输入图元检测模型，得到针对该裁剪子图的检测结果，所述检测结果包括该裁剪子图中的图元的图元类型和所述图元在该裁剪子图中的第一图元位置；根据针对电子图纸的裁剪规则，对针对所有裁剪子图的检测结果进行合并，生成检测图纸，所述检测图纸包括多个图元、每个图元的图元类型和每个图元在检测图纸中的第二图元位置；针对检测图纸中的每个图元，通过与该图元的图元类型对应的图元旋转角度检测模型对该图元进行检测，得到该图元的旋转角度；输出每个图元的识别信息，所述识别信息包括图元的图元类型、图元在检测图纸中的第二图元位置和图元的旋转角度。

可选地，通过裁剪窗口对目标电力系统的电子图纸进行裁剪，得到多个裁剪子图的步骤包括：确定所述裁剪窗口的移动步长；按照所确定的移动步长移动所述裁剪窗口，以在所述电子图纸中进行滑动裁剪，来得到多个裁剪子图；其中，所述移动步长小于裁剪窗口的边长。

可选地，通过以下方式对所述图元检测模型进行训练：获取多张训练电子图纸；针对每张训练电子图纸，通过裁剪窗口对该训练电子图纸进行裁剪，得到多个训练裁剪子图，每个训练裁剪子图的尺寸大于目标图元的尺寸；针对每个训练裁剪子图，将该训练裁剪子图按照预设角度值进行多次旋转，获得在不同旋转角度下的训练裁剪子图；设置初始图元检测模型的网络输入尺寸，所述网络输入尺寸不小于每个训练裁剪子图的尺寸；基于在不同旋转角度下的所有训练裁剪子图对所述初始图元检测模型进行训练，得到所述图元检测模型。

可选地，基于在不同旋转角度下的所有训练裁剪子图对所述初始图元检测模型进行训练，得到所述图元检测模型的步骤包括：基于在不同旋转角度下的所有训练裁剪子图对初始图元检测模型进行训练，得到训练损失函数；确定训练损失函数中的最小值，将所述最小值所对应的初始图元检测模型确定为图元检测模型。