[发明专利]识别红外图像中的断路器的方法、装置及可读存储介质

说明书

本申请公开了一种识别红外图像中的断路器的方法、装置及计算机可读存储介质。其中，方法包括对初始训练样本集进行图像预处理，使得每张红外样本图像上标注的断路器的检测框与预设参考面之间的夹角和参考面与相应断路器所成夹角相同；并进行样本总数扩增处理，最后利用处理得到的训练样本集训练神经网络模型得到目标识别模型。将待处理红外图像输入至目标识别模型中，得到包含标注断路器检测框的红外图像，该检测框与参考面之间的第一夹角与断路器和参考面所成的第二夹角的差值不大于预设角度误差值。本申请无需收集大量训练样本便可训练得到精度高的模型，且可提高断路器在红外图像中检测位置和实际所处位置的贴合度。

技术领域

本申请涉及图像处理技术领域，特别是涉及一种识别红外图像中的断路器的方法、装置及计算机可读存储介质。

背景技术

伴随着红外设备在断路器带电检测中的广泛应用，供电公司积累了大量的运维红外图片数据，目前均由人工对红外图片中是否存在断路器进行筛检，并由人工标注断路器在红外图像中所处的位置，以便用于后期存储和统计等。

但是，人工检测效率低，室外环境复杂，无法实时给出检测结果，此外，人工标注格式不规范，标注后的图片数据对于日后提高检测效率和识别率没有任何帮助，从而使得历史数据的使用分析效率低下，不能对日后巡检识别工作做到有力的帮助。

为了解决人工检测并标注存在的问题，相关技术通过图像识别技术进行筛选，但由于红外图片拍摄角度问题，需要大规模的图片做训练样本集，并且识别结果往往并非按照断路器在图片中的角度进行标注，标注得到的区域存在大部分无用的背景信息。

发明内容

本申请提供了一种识别红外图像中的断路器的方法、装置及计算机可读存储介质，不需要收集大量训练样本集便可训练得到精度高的模型，提高了断路器在红外图像中检测位置和实际所处位置的贴合度，大大减小了断路器区域中包含的无用背景信息。

为解决上述技术问题，本发明实施例提供以下技术方案：

本发明实施例一方面提供了一种识别红外图像中的断路器的方法，包括：

获取包含目标断路器的待处理红外图像；

将所述待处理红外图像输入至预先构建的目标识别模型中，得到包含利用所述数据标注格式标注所述目标断路器检测框的红外图像，所述目标断路器检测框与所述参考面之间的第一夹角与所述目标断路器和所述参考面所成的第二夹角的差值不大于预设角度误差值；

其中，所述目标识别模型为利用经过对初始训练样本集进行图像预处理和样本总数扩增处理后得到的训练样本集训练神经网络模型所得，所述训练样本集中每张红外样本图像上按照预设数据标注格式标注的断路器的检测框与预设参考面之间的夹角，和所述参考面与相应断路器所成夹角相同。

可选的，所述初始训练样本集中每个红外样本图像均包含相应标签，所述标签用于表示图中断路器是否发生倾斜以及倾斜方向；所述初始训练样本集通过旋转图像角度进行样本总数扩增处理，所述初始训练样本集通过旋转图像角度进行样本总数扩增处理过程包括：

按照第一预设角度范围内多个角度值分别对标签为发生倾斜的红外样本图像进行旋转，并将旋转得到的图像作为新增样本图像，同时生成相应标签；

按照第二预设角度范围内多个角度值分别对标签为没有倾斜的红外样本图像进行旋转，并将旋转得到的图像作为新增样本图像，同时生成相应标签；

其中，所述第二预设角度范围中的值小于所述第一预设角度范围的值。

可选的，所述获取包含目标断路器的待处理红外图像之后，还包括：

生成所述待处理红外图像的标识信息，以用于将所述待处理红外图像和所述目标断路器建立对应关系；

其中，所述标识信息为所述目标断路器的编号信息。