[发明专利]刻蚀字符识别网络训练样本增广方法

权利要求书

1.一种刻蚀字符识别网络训练样本增广方法，其特征在于，所述方法包括以下步骤：

步骤1，采集场景中的刻蚀字符图像；

步骤2，根据刻蚀字符图像生成内容图像和风格图像；

步骤3，构建双向生成对抗网络；

步骤4，训练所述双向生成对抗网络；

步骤5，将内容图像和风格图像输入至训练好的双向生成对抗网络，生成刻蚀字符图像。

2.根据权利要求1所述的刻蚀字符识别网络训练样本增广方法，其特征在于，步骤2所述根据刻蚀字符图像生成内容图像，具体包括：

步骤2-1，标注刻蚀字符图像的文本信息；

步骤2-2，根据标注的刻蚀字符图像的真实标签统计字符信息；

步骤2-3，根据字符信息生成多种字体的内容图像。

3.根据权利要求1或2所述的刻蚀字符识别网络训练样本增广方法，其特征在于，步骤2所述根据刻蚀字符图像生成风格图像，具体为：根据刻蚀字符图像特征生成风格图像。

4.根据权利要求3所述的刻蚀字符识别网络训练样本增广方法，其特征在于，步骤2所述根据刻蚀字符图像生成风格图像，具体包括：

从采集的刻蚀字符图像中选取分辨率符合第一预设条件和或清晰度符合第二预设条件和或特征显著度符合第三预设条件的图像，作为风格图像。

5.根据权利要求4所述的刻蚀字符识别网络训练样本增广方法，其特征在于，步骤3中所述构建双向生成对抗网络，具体包括：

步骤3-1，构建风格化生成对抗网络；

步骤3-2，构建去风格化生成对抗网络；

步骤3-3，构建损失函数。

6.根据权利要求5所述的刻蚀字符识别网络训练样本增广方法，其特征在于，步骤3-1中所述风格化生成对抗网络包括：风格化生成网络和风格化判别网络；

所述风格化生成网络，其输入为内容图像和风格图像，输出为风格化字符图像；风格化生成网络包括：内容编码器E_x1，其输入为内容图像，输出为内容特征向量；风格编码器E_x2，其输入为风格图像，输出为风格特征向量；生成器G_x，其输入为所述内容特征向量和风格特征向量，输出为风格化字符图像；

所述风格化判别网络，其输入为所述风格化字符图像或真实的刻蚀字符图像，输出为一个0到1之间的数，用于表示输入的图像为真实图像的概率。

7.根据权利要求6所述的刻蚀字符识别网络训练样本增广方法，其特征在于，步骤3-2中所述去风格化生成对抗网络包括：去风格化生成网络和去风格化判别网络；

所述去风格化生成网络，其输入为所述风格化字符图像，输出为去风格化字符图像；该去风格化生成网络包括：第一编码器E_y，其输入为风格化字符图像，输出为特征向量；第二生成器G_y，其输入为第一编码器E_y输出的特征向量，输出为去风格化字符图像；

所述去风格化判别网络，其输入为所述去风格化字符图像或者真实的内容图像，输出为一个0到1之间的数，用于表示输入的图像是真实图像的概率。