[发明专利]基于生成对抗网络的车牌图像生成方法及系统

权利要求书

1.一种基于生成对抗网络的车牌图像生成方法，其特征在于，包括：

获取目标文本图和测试车牌原图；

将所述测试车牌原图和所述目标文本图输入至文本替换模型中，得到所述文本替换模型输出的车牌预测图；

其中，所述文本替换模型是基于所述训练车牌原图进行文本擦除得到的背景训练图形成的伪车牌图以及所述训练车牌原图训练得到的；

所述文本替换模型用于将预先获取的目标文本图的文本内容替换至所述测试车牌原图，得到车牌预测图。

2.根据权利要求1所述的基于生成对抗网络的车牌图像生成方法，其特征在于，所述文本替换模型，包括：

文本转换层，基于所述测试车牌原图将所述目标文本图的前景风格转换为所述测试车牌原图的前景风格，得到目标文本前景图；

背景修复层，对所述测试车牌原图进行文本擦除和纹理填充，形成车牌背景图；

文本融合层，将所述目标文本前景图和所述车牌背景图进行文本融合，得到车牌预测图。

3.根据权利要求2所述的基于生成对抗网络的车牌图像生成方法，其特征在于，训练所述文本替换模型，包括：

对训练车牌原图进行文本擦除，得到背景图；

基于预先获取的前景文本图和所述背景图合成，得到伪车牌图；

将所述目标文本图、预先获取的车牌背景图、预先获取的文本骨架图、所述前景文本图和所述伪车牌图输入至所述文本替换模型中，得到所述文本替换模型输出的车牌训练图；

基于所述前景文本图、所述背景图、所述车牌训练图、所述文本骨架图、所述目标文本图和所述伪车牌图构建损失函数，并基于所述损失函数收敛以结束训练。

4.根据权利要求3所述的基于生成对抗网络的车牌图像生成方法，其特征在于，所述将所述目标文本图、预先获取的车牌背景图、预先获取的文本骨架图、所述前景文本图和所述伪车牌图输入至所述文本替换模型中，得到所述文本替换模型输出的车牌训练图，包括：

将所述目标文本图、预先获取的车牌背景图、预先获取的文本骨架图、所述前景文本图和所述伪车牌图输入至所述文本转换层，得到所述文本转换层输出的前景训练图；

将所述训练车牌原图输入至所述背景修复层，得到所述背景修复层输出的背景训练图；

将所述前景训练图和所述背景训练图输入至所述文本融合层，得到所述文本融合层输出的车牌训练图。

5.根据权利要求4所述的基于生成对抗网络的车牌图像生成方法，其特征在于，所述基于所述前景文本图、所述背景图、所述车牌训练图、所述文本骨架图、所述目标文本图和所述伪车牌图构建损失函数，包括：

基于所述伪车牌图、所述目标文本图、所述文本骨架图、所述前景文本图和所述前景训练图构建文本转换损失函数；

基于所述伪车牌、所述背景图和所述背景训练图构建背景修复损失函数；

基于所述伪车牌图和所述车牌训练图构建文本融合损失函数；

基于所述文本转换损失函数、所述背景修复损失函数和所述文本融合损失函数得到损失函数。

6.根据权利要求3所述的基于生成对抗网络的车牌图像生成方法，其特征在于，所述对训练车牌原图进行文本擦除，得到背景图，包括：

基于所述训练车牌原图及其对应的预先获取的背景原图进行卷积操作，以获得特征图；

基于所述特征图进行反卷积操作，生成背景图；

基于卷积的中间层特征进行逐层横向连接，以使生成的所述背景图保留所述训练车牌原图的纹理特征。

7.根据权利要求3所述的基于生成对抗网络的车牌图像生成方法，其特征在于，在所述基于预先制作的标准字体图、预先获取的前景文本图和所述背景图合成，得到伪车牌图之前，包括：

制作具有所述训练车牌原图相同文本内容的标准字体图；

基于所述训练车牌原图以及所述背景图进行差值计算，获得所述差值图；

利用骨架提取算法对所述差值图进行提取，获得所述文本骨架图；

将所述差值图进行二值化，获得背景为黒、文本为白色的文本掩码图；

利用所述文本掩码图和所述训练车牌原图进行取交集的图像处理，获得所述前景文本图。