[发明专利]一种绝缘子的识别方法及装置

说明书

本申请提供了一种绝缘子的识别方法及装置，本申请提供的方法包括：获取待识别图片；待识别图片包括含有绝缘子的图片、含有破损绝缘子的图片以及不含绝缘子的图片。将待识别图片输入绝缘子识别模型中，得到绝缘子识别结果；绝缘子识别模型包括待识别图片与绝缘子识别结果的对应关系。绝缘子识别模型包括注意力层；注意力层用于确定绝缘子在待识别图片中的位置数据。本申请通过注意力层将绝缘子在复杂的背景中予以突出显示，使得绝缘子识别模型可以准确在复杂的背景中精准定位绝缘子的区域，从而提高对绝缘子识别的准确率。

技术领域

本申请涉及电力技术领域，特别涉及一种绝缘子的识别方法及装置。

背景技术

输配电系统中，绝缘子的使用是最频繁的，因此对绝缘子的检测是对输配电系统日常检测的一项重要内容。传统的人工检测方式，是由工作人员到绝缘子所在的各个地点，肉眼判断绝缘子是否存在磨损的情况，如果存在，则对磨损的绝缘子进行进一步的检测。人工检测的方法，不仅浪费了大量的人力资源，而且在复杂的输配电系统中，存在遗漏检测绝缘子的可能性。

目前，对绝缘子进行智能检测成为一种主流工作方式。无人机拍摄照片后，检测设备自动识别照片中是否存在绝缘子，并判断绝缘子是否有缺陷。由于绝缘子处于各种不同的环境中，例如深色山丘、树林以及其他深色环境中。如果绝缘子处于复杂的环境中，则检测设备难以从复杂的环境中识别出绝缘子。

基于此，目前亟需一种绝缘子的识别方法，用于解决现有技术中难以从复杂的环境中识别出绝缘子的问题。

发明内容

本申请提供了一种绝缘子的识别方法及装置，可用于解决在现有技术中难以从复杂的环境中识别出绝缘子的问题。

第一方面，本申请提供了一种绝缘子的识别方法，所述方法包括：

获取待识别图片；所述待识别图片包括含有绝缘子的图片、含有破损绝缘子的图片以及不含绝缘子的图片；

将所述待识别图片输入绝缘子识别模型中，得到绝缘子识别结果；所述绝缘子识别模型包括待识别图片与绝缘子识别结果的对应关系；所述绝缘子识别模型包括注意力层；所述注意力层用于确定所述绝缘子在所述待识别图片中的位置数据。

结合第一方面，在第一方面的一种可实现方式中，所述绝缘子识别模型通过以下方法训练：

对样本图片进行人工处理，得到样本绝缘子识别结果；

将所述样本图片划分为训练样本图片以及测试样本图片；

根据训练样本图片以及训练样本绝缘子识别结果，对初始绝缘子识别模型进行训练，得到训练后的绝缘子识别模型；所述训练样本绝缘子识别结果为所述训练样本图片对应的样本绝缘子识别结果；

将所述测试样本图片输入所述训练后的绝缘子识别模型中，得到预测绝缘子识别结果；

判断所述预测绝缘子识别结果与测试样本绝缘子识别结果的差值是否在预设范围内；

如果所述预测绝缘子识别结果与所述测试样本绝缘子识别结果的差值在所述预设范围内，则将所述训练后的绝缘子识别模型确定为所述绝缘子识别模型。

结合第一方面，在第一方面的一种可实现方式中，所述绝缘子识别模型包括以下工作步骤：

提取所述待识别图片的多个特征值，并根据所述多个特征值，确定特征图；

利用所述注意力层确定所述特征图的区域建议窗口；所述区域建议窗口包括待识别的绝缘子；

对所述区域建议窗口进行归一化处理，得到归一化后的区域建议窗口；

对所述归一化后的区域建议窗口进行分类，得到所述绝缘子识别结果。