[发明专利]一种图像识别的方法及装置、电子设备和存储介质

说明书

本发明实施例提供一种图像识别的方法及装置、电子设备和存储介质，其中，所述方法包括：对目标文档图像进行预处理，得到预处理后的待分割图像；通过语义分割模型对所述待分割图像进行语义分割，得到多个分割字符图像；基于字符识别模型对所述分割字符图像进行识别，得到对应的识别字符；基于各个识别字符在文档图像中的位置，获得文档识别结果，从而可以自动地识别目标文档图像中的字符，解决了人工录入工作量大、效率低、错误率高的问题。

技术领域

本发明涉及神经网络技术领域，尤其涉及一种图像识别的方法及装置、电子设备和存储介质。

背景技术

近年来，随着铁路信息化建设的逐步深入，系统覆盖面越来越广，积累的数据量越来越大。铁路工务设备检修记录簿的检查数据对于设备养护维修数据分析具有重大意义。

由于人工检查记录簿大多以多种表格形式记录检查内容，其汉字、数字等内容分散于表格的不同区域，部分记录簿年限已久，存在破损与老化等现象，且有些现场检查人员记录字迹潦草、书写不够规范，大大增加了记录簿信息的提取难度。

人工检查记录簿一般使用复杂版式的表格型，针对该类表格的复杂性及难点，尚未有一种可靠的技术能够快速准确地从记录簿中提取出所需的信息。

发明内容

本发明实施例提供一种图像识别的方法及装置、电子设备和存储介质，用以解决现有技术中不能够快速准确地从记录簿中提取出所需的信息的缺陷。

本发明实施例提供一种图像识别的方法，包括：

对目标文档图像进行预处理，得到预处理后的待分割图像；

通过语义分割模型对所述待分割图像进行语义分割，得到多个分割字符图像；

基于字符识别模型对所述分割字符图像进行识别，得到对应的识别字符；

基于各个识别字符在文档图像中的位置，获得文档识别结果。

根据本发明一个实施例的图像识别的方法，对目标文档图像进行预处理，得到预处理后的待分割图像，包括：

修复：提取目标文档图像的图像纹理特征，并通过分类模型对目标文档图像进行缺陷分类，依据分类对目标文档图像进行缺陷修复；

配准：采用配准模型对修复图像和表格模板进行配准，得到配准图像；

增强：提取配准图像的均衡度和对比度特征，并利用增强模型对配准图像进行处理，得到所述待分割图像。

根据本发明一个实施例的图像识别的方法，提取配准图像的均衡度特征和对比度特征，并利用增强模型对配准图像进行处理，包括：

提取所述配准图像的均衡度特征和对比度特征，并在均衡度小于第一阈值或者对比度小于第二阈值的情况下，利用增强模型对配准图像进行处理。

根据本发明一个实施例的图像识别的方法，所述配准模型包括多个B样条基函数；

采用配准模型对修复图像和表格模板进行配准，得到配准图像，包括：

提取所述修复图像的感兴趣特征点；

基于所述感兴趣特征点、所述表格模板以及B样条基函数进行处理，得到所述配准图像。

根据本发明一个实施例的图像识别的方法，通过语义分割模型对所述待分割图像进行语义分割，得到多个分割字符图像，包括：

在所述待分割图像包括表格的情况下，确定表格中各个单元格的位置和排列关系，对所述单元格进行切分；

通过语义分割模型对每个所述单元格进行语义分割，得到多个字符区域图像；

对所述字符区域图像进行轮廓识别，得到每个字符区域图像中的多个分割字符图像。