【技术实现步骤摘要】
本申请涉及计算机视觉，具体而言，涉及一种证件图像生成方法、装置和电子设备。
技术介绍
1、在传统的人像证件照制作过程中，用户通常需要在专业的摄影环境中进行拍摄照片，摄像师通过照片编辑软件进行调整，以满足特定的证件照标准，耗时且成本高，因此基于图像生成算法制作证件照非常必要。
2、目前，采用图像生成算法进行人像证件照制作的过程中，通常需要用户提供高质量的照片，且照片数量至少需要10张以上，以生成人像证件照。
3、然而，若存在不符合要求的照片，会降低证件照生成的效果，从而大大提高了证件照的制作难度。
技术实现思路
1、有鉴于此，本申请实施例提供了一种证件图像生成方法、装置和电子设备，以降低证件照制作难度。
2、第一方面，本申请实施例提供了一种证件图像生成方法，包括：
3、获取目标人物的原始图像；
4、根据所述原始图像进行服饰分割，得到所述目标人物的服饰掩膜；
5、根据所述原始图像、所述服饰掩膜以及多种预设证件服饰提示信息，获取针对所述目标人物的训练图像集；
6、根据所述训练图像集对预设扩散模型进行模型训练，得到所述目标人物的证件图像生成模型；
7、获取针对所述目标人物的目标证件服饰提示信息；
8、采用所述证件图像生成模型，根据所述目标证件服饰提示信息，生成具有所述目标证件服饰提示信息对应的服饰的所述目标人物的目标证件图像。
9、在一可选的实施方式中，所述根据所述原始图
10、根据所述原始图像、所述服饰掩膜以及所述多种预设证件服饰提示信息，分别生成所述目标人物的多个目标图像，一个目标图像为具有一种预设证件服饰提示信息对应的服饰的所述目标人物的证件图像；
11、根据所述原始图像和所述多个目标图像，生成所述训练图像集。
12、在一可选的实施方式中，所述方法还包括：
13、对所述原始图像进行抠像处理，得到所述目标人物的人像掩膜；
14、对预设背景图像和所述人像掩膜进行图像融合，得到融合图像；
15、所述根据所述原始图像进行服饰分割，得到所述目标人物的服饰掩膜，包括：
16、对所述融合图像进行服饰分割，得到所述目标人物的服饰掩膜。
17、在一可选的实施方式中，所述方法还包括：
18、获取预设标准证件图像的标准人脸位置信息，以及所述原始图像中所述目标人物的人脸位置信息；
19、根据所述标准人脸位置信息和所述目标人物的人脸位置信息，对所述原始图像进行对齐处理，得到对齐图像；
20、所述根据所述原始图像进行服饰分割，得到所述目标人物的服饰掩膜，包括：
21、对所述对齐图像进行服饰分割，得到所述目标人物的服饰掩膜。
22、在一可选的实施方式中，所述方法还包括：
23、分别对所述原始图像以及所述多个目标图像进行镜像处理，得到第一镜像图像以及多个第二镜像图像；
24、所述根据所述原始图像和所述多个目标图像，生成训练图像集，包括：
25、根据所述原始图像、所述多个目标图像、所述第一镜像图像以及所述多个第二镜像图像，生成所述训练图像集。
26、在一可选的实施方式中，所述采用所述证件图像生成模型，根据所述目标证件服饰提示信息，生成具有所述目标证件服饰提示信息对应的服饰的所述目标人物的目标证件图像，包括：
27、获取所述目标人物的性别信息以及针对所述目标证件图像的背景提示信息；
28、采用所述证件图像生成模型，根据所述目标证件服饰提示信息、所述目标人物的性别信息以及所述背景提示信息，生成所述目标证件图像，所述目标证件图像的图像背景为所述背景提示信息对应的背景。
29、在一可选的实施方式中，所述方法还包括：
30、分别获取所述目标证件图像中所述目标人物的第一人脸特征，以及所述原始图像中所述目标人物的第二人脸特征；
31、根据所述第一人脸特征和所述第二人脸特征，计算人脸相似度；
32、若所述人脸相似度小于预设相似度阈值，则重新生成所述目标证件图像，直至所述人脸相似度大于或等于所述预设相似度阈值。
33、在一可选的实施方式中，所述方法还包括：
34、对所述目标证件图像进行人脸数量检测，得到人脸数量检测结果；
35、若所述人脸数量检测结果指示所述目标证件图像中存在至少两个人脸，则重新生成所述目标证件图像。
36、第二方面，本申请实施例还提供了一种证件图像生成装置，包括：
37、获取模块，用于获取目标人物的原始图像；
38、处理模块，用于根据所述原始图像进行服饰分割，得到所述目标人物的服饰掩膜；
39、所述获取模块，还用于根据所述原始图像、所述服饰掩膜以及多种预设证件服饰提示信息，获取针对所述目标人物的训练图像集；
40、训练模块，用于根据所述训练图像集对预设扩散模型进行模型训练，得到所述目标人物的证件图像生成模型；
41、所述获取模块，还用于获取针对所述目标人物的目标证件服饰提示信息；
42、生成模块，用于采用所述证件图像生成模型，根据所述目标证件服饰提示信息，生成具有所述目标证件服饰提示信息对应的服饰的所述目标人物的目标证件图像。
43、在一可选的实施方式中，所述获取模块，具体用于：
44、根据所述原始图像、所述服饰掩膜以及所述多种预设证件服饰提示信息，分别生成所述目标人物的多个目标图像，一个目标图像为具有一种预设证件服饰提示信息对应的服饰的所述目标人物的证件图像；
45、根据所述原始图像和所述多个目标图像，生成所述训练图像集。
46、在一可选的实施方式中，所述处理模块，还用于：
47、对所述原始图像进行抠像处理，得到所述目标人物的人像掩膜；
48、对预设背景图像和所述人像掩膜进行图像融合，得到融合图像；
49、所述处理模块，具体用于：
50、对所述融合图像进行服饰分割，得到所述目标人物的服饰掩膜。
51、在一可选的实施方式中，所述获取模块，还用于：
52、获取预设标准证件图像的标准人脸位置信息，以及所述原始图像中所述目标人物的人脸位置信息；
53、所述处理模块，还用于根据所述标准人脸位置信息和所述目标人物的人脸位置信息，对所述原始图像进行对齐处理，得到对齐图像；
54、所述处理模块，具体用于：
55、对所述对齐图像进行服饰分割，得到所述目标人物的服饰掩膜。
56、在一可选的实施方式中，所述处理模块，还用于：
57、分别对所述原始图像以及所述多个目标图像进行镜像处理，得到第一镜像图本文档来自技高网...
【技术保护点】
1.一种证件图像生成方法，其特征在于，包括：
2.根据权利要求1所述的方法，其特征在于，所述根据所述原始图像、所述服饰掩膜以及多种预设证件服饰提示信息，获取针对所述目标人物的训练图像集，包括：
3.根据权利要求1所述的方法，其特征在于，所述方法还包括：
4.根据权利要求1所述的方法，其特征在于，所述方法还包括：
5.根据权利要求2所述的方法，其特征在于，所述方法还包括：
6.根据权利要求1所述的方法，其特征在于，所述采用所述证件图像生成模型，根据所述目标证件服饰提示信息，生成具有所述目标证件服饰提示信息对应的服饰的所述目标人物的目标证件图像，包括：
7.根据权利要求5所述的方法，其特征在于，所述方法还包括：
8.根据权利要求5所述的方法，其特征在于，所述方法还包括：
9.一种证件图像生成装置，其特征在于，包括：
10.一种电子设备，其特征在于，包括：处理器、存储器和总线，所述存储器存储有所述处理器可执行的机器可读指令，当电子设备运行时，所述处理器与所述存储器之间通过总线通信
...【技术特征摘要】
1.一种证件图像生成方法，其特征在于，包括：
2.根据权利要求1所述的方法，其特征在于，所述根据所述原始图像、所述服饰掩膜以及多种预设证件服饰提示信息，获取针对所述目标人物的训练图像集，包括：
3.根据权利要求1所述的方法，其特征在于，所述方法还包括：
4.根据权利要求1所述的方法，其特征在于，所述方法还包括：
5.根据权利要求2所述的方法，其特征在于，所述方法还包括：
6.根据权利要求1所述的方法，其特征在于，所述采用所述证件图像生成模型，根据所述目标证件服饰提示信息...
【专利技术属性】
技术研发人员：张明琦，吴茗，韩晟，
申请(专利权)人：杭州小影创新科技股份有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。