[发明专利]一种用于电力设备的文本识别方法及设备

权利要求书

1.一种用于电力设备的文本识别方法，其特征在于，所述方法包括：

获取设备监拍图像；

通过预设的文本检测模型，确定所述设备监拍图像的文本区域；其中，所述文本检测模型通过若干电力设备样本图训练得到；所述电力设备样本图基于预先确定的电力设备标注图以及电力领域文本集生成；所述电力设备标注图为标注电力设备表面的文本信息的图像；

将所述文本区域相应的区域图像，输入预设的文本识别模型，以确定所述区域图像中的文本信息；其中，所述文本识别模型通过预设处理后的若干文本区域图像训练得到；所述预设处理为所述文本识别模型将各所述文本区域图像遮挡处理；所述文本区域图像根据所述电力设备样本图得到；

根据所述文本信息，生成所述设备监拍图像相应的标注图像；其中，所述标注图像包括标注的所述文本信息。

2.根据权利要求1所述方法，其特征在于，通过预设的文本检测模型，确定所述设备监拍图像的文本区域之前，所述方法还包括：

通过互联网爬虫软件，获取若干电力领域词汇；其中，所述电力领域词汇为爬取若干电力专业词汇得到的；

通过预设的开源字体库，对各所述电力领域词汇进行字体处理；所述字体处理为将所述电力领域词汇的字体属性进行随机设置；所述字体属性至少包括以下一项或多项：字体大小、笔画宽度、字体间距；

根据字体处理后的各所述电力领域词汇与数码管字体集，生成所述电力领域文本集。

3.根据权利要求1所述方法，其特征在于，通过预设的文本检测模型，确定所述设备监拍图像的文本区域之前，所述方法还包括：

确定各所述电力设备标注图中，标注的所述文本区域图像的前景图像的图像特征，为第一特征；所述前景图像为所述文本区域图像中的文本；

确定所述电力设备标注图中背景图像的图像特征，为第二特征；

将所述电力领域文本集中各电力文本，与各所述第一特征进行随机匹配，得到随机文本图像集合；

将所述随机文本图像集合中的各文本图像与各所述第二特征相应的背景图像进行随机组合，生成所述电力设备样本图。

4.根据权利要求1所述方法，其特征在于，通过预设的文本检测模型，确定所述设备监拍图像的文本区域，具体包括：

将所述设备监拍图像，进行双边滤波处理，得到边缘保存图像；

通过颜色空间转换算法，将所述边缘保存图像转换至HSV颜色空间；其中，所述颜色空间转换算法用于将RGB色彩模式与所述HSV颜色空间进行转换；

通过限制对比度自适应直方图均衡CLAHE算法，将转换至所述HSV颜色空间的所述边缘保存图像的亮度分量，进行对比度处理，得到预处理图像；其中，所述对比度处理用于调整所述边缘保存图像的背景与文本的对比度；

通过所述颜色空间转换算法，将所述预处理图像转换至所述RGB色彩模式，以将所述RGB色彩模式的所述预处理图像输入所述文本检测模型，确定所述设备监拍图像的文本区域。

5.根据权利要求1所述方法，其特征在于，所述文本识别模型将各所述文本区域图像遮挡处理，具体包括：

根据所述文本区域图像，确定所述文本区域图像中文本序列；所述文本序列根据所述文本区域图像中的文本字符数量生成；

确定所述文本序列中的各待遮挡文本；其中，所述待遮挡文本至少包括所述文本序列相应的一个文本字符；

将所述文本区域图像中，各所述待遮挡文本相应的区域依次进行遮挡，得到遮挡文本区域图像，直至所述文本区域图像的各区域完成遮挡处理。

6.根据权利要求1所述方法，其特征在于，将所述文本区域相应的区域图像，输入预设的文本识别模型，以确定所述区域图像中的文本信息，具体包括：

通过所述文本识别模型，对所述区域图像进行卷积处理，得到所述区域图像相应的特征向量；

根据注意力机制，将所述区域图像输入所述文本识别模型的注意力层，得到所述区域图像的若干注意力图；

将各所述注意力图相应向量与所述特征向量进行点乘运算，并将所述点乘运算的运算结果输入所述文本识别模型的全连接层，得到所述区域图像相应的所述文本信息。