[发明专利]OCR训练数据生成方法、装置、计算机设备和存储介质

权利要求书

1.一种OCR训练数据生成方法，其特征在于，所述方法包括：

获取训练语料库；

获取图像参数信息，根据所述图像参数信息建立空白图像；

提取所述训练语料库中预设字符串，按照预设的生成字符参数信息将提取的字符串中的字符写入所述空白图像中，生成初始图像；

对所述初始图像进行图像增强处理，得到训练图像；

根据所述训练图像生成训练数据。

2.根据权利要求1所述的OCR训练数据生成方法，其特征在于，所述按照预设的生成字符参数信息将提取的字符串中的字符写入所述空白图像中，生成初始图像，包括：

根据所述图像参数信息获取所述生成字符参数信息；

从所述训练语料库的字符串中获取生成字符；

根据所述生成字符参数信息将所述生成字符写入所述空白图像，得到初始图像。

3.根据权利要求2所述的OCR训练数据生成方法，其特征在于，所述图像参数信息包括图像宽度和图像高度；所述根据所述生成字符参数信息将所述字符写入所述空白图像，得到初始图像，包括：

从所述空白图像的字符起始水平位置开始，根据所述生成字符参数信息将所述生成字符写入所述空白图像中，直到当前生成字符的字符起始水平位置超过所述图像宽度，得到初始图像。

4.根据权利要求1所述的OCR训练数据生成方法，其特征在于，所述对所述初始图像进行图像增强处理，得到训练图像，包括：

对所述初始图像进行随机二值化；

对二值化后的所述初始图像中的黑色像素进行随机映射至预设灰度值范围；

对所述初始训练图像进行抖动处理生成点阵二值图；

对所述点阵二值图进行横向或纵向的随机缩放，得到训练图像。

5.根据权利要求1所述的OCR训练数据生成方法，其特征在于，所述对所述初始图像进行图像增强处理，得到训练图像，包括：

对所述初始图像进行随机二值化；

根据二值化后的所述初始图像获取黑色像素微量形态学膨胀的图像；

对所述初始图像进行拷贝，得到拷贝图像；

对所述初始图像进行纵向腐蚀运算；

根据所述图像参数信息生成所述初始图像的映射矩阵；

对所述初始图像和所述拷贝图像，根据所述映射矩阵对腐蚀运算处理掉的所述初始图像中的横向笔画进行连接；

对所述初始图像的横向或纵向进行随机的双线性插值或立方插值；

对插值后的所述初始图像进行随机二值化，得到训练图像。

6.根据权利要求5所述OCR训练数据生成方法，其特征在于，对所述初始图像和所述拷贝图像，根据所述映射矩阵对腐蚀运算处理掉的所述初始图像中的横向笔画进行连接，包括：

若所述映射矩阵在某个像素点的值为第一点值，对所述初始图像和所述拷贝图像对应的像素点值进行与运算；

若所述映射矩阵在某个像素点的值为第二点值，则删除该像素点在所述初始图像中对应的像素值。

7.根据权利要求1所述OCR训练数据生成方法，其特征在于，所述根据所述训练图像生成训练数据，包括：

保存所述训练图像到预设路径中；

保存所述训练图像的预设路径；

保存所述训练图像对应的字符串文本；

整合所述训练图像、所述预设路径和所述字符串文本，得到训练数据。

8.一种OCR训练数据生成装置，其特征在于，所述装置包括：

语料获取模块，用于获取训练语料库；

图像建立模块，用于获取图像参数信息，根据所述图像参数信息建立空白图像；

字符提取模块，用于提取所述训练语料库中预设字符串，按照预设的生成字符参数信息将提取的字符串中的字符写入所述空白图像中，生成初始图像；

图像增强模块，对所述初始图像进行图像增强处理，得到训练图像；

数据生成模块，用于根据所述训练图像生成训练数据。