[发明专利]图像数据处理方法、装置、计算机设备及存储介质

说明书

本发明公开一种图像数据处理方法、装置、计算机设备及存储介质，应用在图像识别技术领域。该方法包括：采用爬虫工具爬取原始图像，每一原始图像对应一图像类型；采用文本定位算法对原始图像进行文本定位，获取至少一个文本行区域，对每一文本行区域进行截图，获取区域块图像；基于图像类型和区域块图像的位置信息，获取目标OCR识别模型；采用目标OCR识别模型对区域块图像进行识别，获取目标识别结果，目标识别结果包括至少两个识别文字和对应的识别概率；基于至少两个识别文字和对应的识别概率，获取目标文字，将目标文字确定为标注文本，基于区域块图像和标注文本，获取目标图像样本。该方法可有效提高目标图像样本的获取效率并降低获取成本。

技术领域

本发明涉及图像识别技术领域，尤其涉及一种图像数据处理方法、装置、计算机设备及存储介质。

背景技术

在图像识别技术领域，需将携带标注信息的图像样本输入到基于CNN、LSTM或者其他网络模型中进行训练，以获取图像识别模型，使得该图像识别模型可对图像中的文字进行识别。当前图像样本主要是通过爬虫工具爬取大量的原始图像，对每一原始图像中的文字内容进行人工标注，以形成可用于进行模型训练的图像样本。这种人工标注图像样本的方式存在效率低、耗时长且人工成本高等问题。

发明内容

本发明实施例提供一种图像数据处理方法、装置、计算机设备及存储介质，以解决当前人工标注图像样本过程中存在效率低、耗时长且成本高的问题。

一种图像数据处理方法，包括：

采用爬虫工具爬取原始图像，每一所述原始图像对应一图像类型；

采用文本定位算法对所述原始图像进行文本定位，获取至少一个文本行区域，并对每一所述文本行区域进行截图，获取对应的区域块图像，每一所述区域块图像对应一位置信息；

基于所述图像类型和所述区域块图像的位置信息，获取目标OCR识别模型；

采用所述目标OCR识别模型对所述区域块图像进行识别，获取与所述区域块图像相对应的目标识别结果，所述目标识别结果包括至少两个识别文字和与每一所述识别文字相对应的识别概率；

基于至少两个所述识别文字和与每一所述识别文字相对应的识别概率，获取与所述区域块图像相对应的目标文字，将所述目标文字确定为标注文本，基于所述区域块图像和所述标注文本，获取目标图像样本。

一种图像数据处理装置，包括：

原始图像获取模块，用于采用爬虫工具爬取原始图像，每一所述原始图像对应一图像类型；

区域块图像获取模块，用于采用文本定位算法对所述原始图像进行文本定位，获取至少一个文本行区域，并对每一所述文本行区域进行截图，获取对应的区域块图像，每一所述区域块图像对应一位置信息；

识别模型获取模块，用于基于所述图像类型和所述区域块图像的位置信息，获取目标OCR识别模型；

识别结果获取模块，用于采用所述目标OCR识别模型对所述区域块图像进行识别，获取与所述区域块图像相对应的目标识别结果，所述目标识别结果包括至少两个识别文字和与每一所述识别文字相对应的识别概率；

图像样本获取模块，用于基于至少两个所述识别文字和与每一所述识别文字相对应的识别概率，获取与所述区域块图像相对应的目标文字，将所述目标文字确定为标注文本，基于所述区域块图像和所述标注文本，获取目标图像样本。

一种计算机设备，包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现上述图像数据处理方法的步骤。

一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序被处理器执行时实现上述图像数据处理方法的步骤。