训练图像的扩充方法、训练方法及相关装置与流程

文档序号:19894473发布日期:2020-02-11 13:05阅读:来源:国知局

技术特征:

1.一种训练图像的扩充方法,其特征在于,所述方法包括:

获取行人的多张待转换图像;所述多张待转换图像包括所述行人的至少两种行人颜色数据;

根据所述多张待转换图像,获取颜色数据生成模型;所述颜色数据生成模型为所述至少两种行人颜色数据在生成对抗网络下训练得到的模型;

将所述多张待转换图像输入至所述颜色数据生成模型,获取多张目标颜色图像;

将所述待转换图像和所述多张目标颜色图像作为颜色训练集。

2.根据权利要求1所述的方法,其特征在于,所述获取行人的多张待转换图像,包括:

获取所述行人的至少一张第一图像和至少一张第二图像;所述至少一张第一图像为所述行人的着装颜色为目标颜色之外的行人图像,所述至少一张第二图像为所述行人的着装颜色为所述目标颜色的行人图像;

将所述至少一张第一图像和所述至少一张第二图像作为所述多张待转换图像。

3.根据权利要求2所述的方法,其特征在于,所述根据所述多张待转换图像,获取颜色数据生成模型,包括:

获取所述至少一张第一图像的纹理信息;所述纹理信息表征所述行人的图像切分信息;

根据所述纹理信息和所述至少一张第一图像,生成多张待确认图像;所述多张待确认图像为所述行人的着装颜色为所述目标颜色的行人图像;

将所述至少一张第一图像作为第一训练集,将所述多张待确认图像和所述至少一张第二图像作为第二训练集;

根据所述第一训练集和所述第二训练集,获取所述颜色数据生成模型。

4.根据权利要求3所述的方法,其特征在于,所述生成对抗网络包括第一生成器、第二生成器和判别器,所述根据所述第一训练集和所述第二训练集,获取所述颜色数据生成模型,包括:

将所述第一训练集通过所述第一生成器,生成与所述目标颜色一致的第一图形数据,以使所述第一图形数据与所述第二训练集符合判别损失函数的收敛条件;将所述第二训练集通过所述判别器,以使所述第二训练集与所述判别损失函数的发散条件匹配;

将所述第一图形数据通过所述第二生成器,生成与所述第一训练集的颜色一致的第二图形数据,以使所述第二图形数据与所述第一图形数据符合生成损失函数的收敛条件;

获取符合模型收敛条件的所述颜色数据生成模型。

5.一种颜色识别模型的训练方法,其特征在于,所述方法包括:

获取行人的颜色训练集;所述颜色训练集包括多张待转换图像和多张目标颜色图像;所述多张待转换图像包括至少所述行人的至少两种行人颜色数据,所述多张目标颜色图像为所述多张待转换图像输入至颜色数据生成模型获取的行人图像,所述颜色数据生成模型为所述至少两种行人颜色数据在生成对抗网络下训练得到的模型;

根据所述颜色训练集,获取多个教师模型;

根据所述多个教师模型和所述颜色训练集,获取颜色识别模型。

6.根据权利要求5所述的方法,其特征在于,所述根据所述多个教师模型和所述颜色训练集,获取颜色识别模型,包括:

对所述多个教师模型进行特征适配,以使所述多个教师模型与所述颜色识别模型的模型训练信息匹配;所述模型训练信息包括特征维度和特征值大小。

7.根据权利要求5所述的方法,其特征在于,所述根据所述多个教师模型和所述颜色训练集,获取颜色识别模型,还包括:

将所述颜色训练集输入至第一模型,获取第一输出特征;所述第一模型为所述多个教师模型的任意一个教师模型;

将所述颜色训练集输入至待确认识别模型,获取第二输出特征;

将所述第一输出特征与所述第二输出特征匹配,以使相似度损失函数收敛;所述相似度损失函数用于使所述待确认识别模型与所述第一模型匹配;

当所述相似度损失函数收敛时,将所述待确认识别模型作为所述颜色识别模型。

8.一种训练图像的扩充装置,其特征在于,包括:第一获取模块和第一处理模块;

所述第一获取模块,用于获取行人的多张待转换图像;所述多张待转换图像包括所述行人的至少两种行人颜色数据;

所述第一处理模块,用于根据所述多张待转换图像,获取颜色数据生成模型;所述颜色数据生成模型为所述至少两种行人颜色数据在生成对抗网络下训练得到的模型;

所述第一获取模块还用于将所述多张待转换图像输入至所述颜色数据生成模型,获取多张目标颜色图像;

所述第一处理模块还用于将所述待转换图像和所述多张目标颜色图像作为颜色训练集。

9.一种颜色识别模型的训练装置,其特征在于,包括:第二获取模块和第二处理模块;

所述第二获取模块,用于获取行人的颜色训练集;所述颜色训练集包括多张待转换图像和多张目标颜色图像;所述多张待转换图像包括至少所述行人的至少两种行人颜色数据,所述多张目标颜色图像为所述多张待转换图像输入至颜色数据生成模型获取的行人图像,所述颜色数据生成模型为所述至少两种行人颜色数据在生成对抗网络下训练得到的模型;

所述第二处理模块,用于根据所述颜色训练集,获取多个教师模型;

所述第二处理模块还用于根据所述多个教师模型和所述颜色训练集,获取颜色识别模型。

10.一种电子设备,其特征在于,包括处理器和存储器,所述存储器存储有能够被所述处理器执行的机器可执行指令,所述处理器可执行所述机器可执行指令以实现权利要求1-4任一项所述的扩充方法或权利要求5-7任一项所述的训练方法。

11.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1-4任一项所述的扩充方法或权利要求5-7任一项所述的训练方法。

当前第2页1 2 3 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1