[发明专利]电表读数识别方法、装置、电子设备和存储介质

说明书

本申请提供一种电表读数识别方法、装置、电子设备和存储介质。该方法包括：获取待检测的电表图像；将待检测的电表图像输入训练好的YOLO模型，从待检测的电表图像中确定出读数区域；将待检测的电表图像输入训练好的CTPN模型，从待检测的电表图像中确定出至少一个文本区域；根据读数区域和至少一个文本区域，确定读数区域内的至少一个文本区域；利用投影法，从读数区域内的至少一个文本区域中确定电表读数文本区域；判断电表读数文本区域中是否存在小数；将电表读数文本区域的图像输入训练好的CRNN模型，确定电表读数文本区域中的数字；根据电表读数文本区域中的数字、电表读数文本区域中是否存在小数，输出电表读数识别结果。

技术领域

本申请涉及图像识别技术，尤其涉及一种电表读数识别方法、装置、电子设备和存储介质。

背景技术

当前智能电表应用比较广泛，可以自行上传用电量数据。但仍有一些机械电表还在使用当中。

机械表无法实现用电量等数据的自动上传，仍需要人工抄表。抄表员不但需要定期完成电表数据的采集，还需要手动完成电表数据的填报和审核。在人工抄表过程中，还可能出现数据填报错误等问题，导致对机械电表的用电量数据采集的效率较低。

发明内容

本申请提供一种电表读数识别方法、装置、电子设备和存储介质，实现对机械电表数据的自动识别，避免人工数据填报错误等问题，提高机械电表用电量数据采集的效率。

第一方面，本申请提供一种电表读数识别方法，包括：

获取待检测的电表图像；

将所述待检测的电表图像输入训练好的YOLO模型，从所述待检测的电表图像中确定出读数区域；

将所述待检测的电表图像输入训练好的CTPN模型，从所述待检测的电表图像中确定出至少一个文本区域；

根据所述读数区域和所述至少一个文本区域，确定所述读数区域内的至少一个文本区域；

利用投影法，从所述读数区域内的至少一个文本区域中确定电表读数文本区域；

判断所述电表读数文本区域中是否存在小数；

将所述电表读数文本区域的图像输入训练好的CRNN模型，确定所述电表读数文本区域中的数字；

根据所述电表读数文本区域中的数字、所述电表读数文本区域中是否存在小数，输出电表读数识别结果。

可选的，所述方法还包括：

预先采集多张电表图像样本；

对每张所述电表图像样本的读数区域进行标注，构建YOLO训练数据集；

利用YOLO训练数据集，基于YOLO预训练模型进行训练，得到训练好的YOLO模型。

可选的，所述将所述待检测的电表图像输入训练好的CTPN模型，从所述待检测的电表图像中确定出至少一个文本区域，包括：

将所述待检测的电表图像的像素值缩小到1200*1200以下，得到缩放后的电表图像；

将所述缩放后的电表图像输入训练好的CTPN模型，从所述缩放后的电表图像中确定出至少一个文本区域。

可选的，所述根据所述读数区域和所述至少一个文本区域，确定读数区域内的至少一个文本区域，包括：

根据所述读数区域的坐标分布、所述至少一个文本区域的坐标分布，确定位于所述读数区域内的至少一个文本区域。

可选的，所述利用投影法，从所述读数区域内的至少一个文本区域中确定电表读数文本区域，包括：

对所述读数区域的图像进行灰度化处理，得到所述读数区域的灰度图；