[发明专利]一种车牌图像生成方法、装置、电子设备及存储介质

权利要求书

1.一种车牌图像生成方法，其特征在于，所述方法包括：

对待处理车牌图像包含的车牌字符序列中目标序位的字符进行处理，得到参考车牌图像；

基于训练后的车牌图像生成模型，对所述参考车牌图像的车牌颜色进行扩展处理，获得扩展车牌图像，所述车牌图像生成模型通过机器学习方法对生成式对抗网络进行训练获得，所述车牌图像生成模型包括多个颜色子模型，一个颜色子模型对应一种预设颜色的颜色信息，不同的颜色子模型对应的预设颜色不同；

其中，所述基于训练后的车牌图像生成模型，对所述参考车牌图像的车牌颜色进行扩展处理，获得扩展车牌图像，包括：

将所述参考车牌图像输入所述车牌图像生成模型的预设数量的颜色子模型，通过所述预设数量的颜色子模型生成预设数量的颜色信息，并通过所述车牌图像生成模型的生成器将所述参考车牌图像的车牌颜色信息替换为所述预设数量的颜色信息，生成所述预设数量的颜色信息对应的预设数量的RGB图像，得到所述预设数量的扩展车牌图像。

2.如权利要求1所述的方法，其特征在于，所述车牌图像生成模型通过如下方式获得：

获取车牌样本图像；

对所述车牌样本图像进行预处理，得到目标样本图像；所述目标样本图像中每个图像的尺寸相同，所述目标样本图像为二值化图像；

将所述目标样本图像输入所述生成式对抗网络的目标颜色子模型，得到所述目标样本图像的目标颜色信息，所述目标颜色子模型对应的颜色与所述车牌样本图像的颜色相同；

将带有目标颜色信息的目标样本图像输入所述生成式对抗网络的生成器，通过所述生成器生成对应颜色的RGB图像；

通过所述生成式对抗网络的判别器对输入所述判别器的图像进行判别；所述输入所述判别器的图像包括：所述带有目标颜色信息的目标样本图像和所述生成器生成的对应颜色的RGB图像，或者所述带有目标颜色信息的目标样本图像和所述车牌样本图像，在所述输入所述判别器的图像为所述带有目标颜色信息的目标样本图像和所述RGB图像时，判别所述输入图像为假，在所述输入所述判别器的图像为所述带有目标颜色信息的目标样本图像和所述车牌样本图像时，判别所述输入图像为真；

基于所述判别结果，通过生成器的损失函数对所述生成器进行优化以及通过所述判别器的损失函数对所述判别器进行优化，以使所述生成器生成的RGB图像更接近所述车牌样本图像，以及使所述判别器的判别结果更准确；

直到所述判别器难以判别输入的图像为真还是为假时，停止网络训练，得到所述车牌图像生成模型。

3.如权利要求2所述的方法，其特征在于，所述通过生成器的损失函数对所述生成器进行优化，包括：

根据所述判别器的判别结果、所述生成器生成的RGB图像与所述车牌样本图像的欧式距离和所述生成器的损失函数，确定所述生成器的生成损失；

根据所述生成损失，更新所述生成器的参数。

4.如权利要求2所述的方法，其特征在于，所述通过所述判别器的损失函数对所述判别器进行优化，包括：

根据所述判别器的判别结果和所述判别器的损失函数，确定所述判别器的判别损失；

根据所述判别损失，更新所述判别器的参数。

5.如权利要求1所述的方法，其特征在于，所述对待处理车牌图像包含的车牌字符序列中目标序位的字符进行处理，得到参考车牌图像，包括：

对所述待处理车牌图像进行二值化处理，获得二值化车牌图像；

对所述二值化车牌图像中目标序位的字符进行处理，得到参考车牌图像。

6.如权利要求1所述的方法，其特征在于，所述对待处理车牌图像包含的车牌字符序列中目标序位的字符进行处理，得到参考车牌图像，包括：

将待处理车牌图像包含的车牌字符序列中目标序位的字符，替换为预设字符或随机字符，得到所述参考车牌图像。