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

权利要求书

1.一种识别红外图像中的断路器的方法，其特征在于，包括：

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

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

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

2.根据权利要求1所述的识别红外图像中的断路器的方法，其特征在于，所述初始训练样本集中每个红外样本图像均包含相应标签，所述标签用于表示图中断路器是否发生倾斜以及倾斜方向；所述初始训练样本集通过旋转图像角度进行样本总数扩增处理，所述初始训练样本集通过旋转图像角度进行样本总数扩增处理过程包括：

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

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

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

3.根据权利要求2所述的识别红外图像中的断路器的方法，其特征在于，所述获取包含目标断路器的待处理红外图像之后，还包括：

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

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

4.根据权利要求3所述的识别红外图像中的断路器的方法，其特征在于，所述目标断路器检测框为矩形框，所述将所述待处理红外图像输入至所述目标识别模型中，得到包含利用所述数据标注格式标注所述目标断路器检测框的红外图像之后，还包括：

输出所述目标断路器检测框的四个顶点坐标信息及所述目标断路器的设备参数信息，以作为所述红外图像的属性信息进行存储。

5.根据权利要求1至4任意一项所述的识别红外图像中的断路器的方法，其特征在于，所述预先对初始训练样本集进行图像预处理包括：

按照图中断路器是否发生倾斜将所述初始样本训练集中的红外样本图像分为两类，生成待处理图像组和正目标图像组；所述正目标图像组中的红外样本图像的断路器与图像中轴线平行；

对所述待处理图像组中的每张红外样本图像，计算当前红外样本图像中的当前断路器偏离所述参考面的偏转角度，按照所述偏转角度旋转所述当前红外样本图像直至所述当前断路器与图像中轴线平行，同时利用数据标注方法在所述当前红外样本图像中标注断路器的检测框，在生成所述检测框后将所述当前红外样本图像旋转至所述当前断路器与所述参考面所成角度为所述偏转角度。

6.根据权利要求5所述的识别红外图像中的断路器的方法，其特征在于，所述目标识别模型为基于SSD图像识别方法识别红外图像中的断路器，并采用VOC格式标注断路器检测框。

7.根据权利要求6所述的识别红外图像中的断路器的方法，其特征在于，所述获取包含断路器的待处理红外图像之后，还包括：

对所述待处理红外图像进行灰度处理，得到二值化红外图像，以用于将所述二值化红外图像输入至所述目标识别模型中。