基于生成对抗网络的多标签汽车造型生成方法技术

本发明专利技术提供了一种基于生成对抗技术来根据指定标签生成汽车图片的方法，步骤为：1)获取真实汽车图片，并打上标签Fi；2)建立汽车图片对抗生成网络。3)得到汽车生成图片。4)计算汽车生成图片和真实图片中Fi标签数据组合损失；5)更新汽车图片判别器Di权重；6)更新汽车图片生成器G的权重；7)完成汽车图片对抗生成网络的训练；8)在汽车图片对抗生成网络中输入指定的h种标签，输出对应的汽车生成图片。本发明专利技术能够根据指定的条件标签，生成对应的汽车图片，丰富了生成图片的多方约束，使得结果更加精准、满足用户需求。满足用户需求。

【技术实现步骤摘要】
基于生成对抗网络的多标签汽车造型生成方法


[0001]本专利技术属于生成模型以及汽车图片领域，具体是涉及一种基于生成对抗网络的多标签汽车造型生成方法。

技术介绍

[0002]汽车已经是当代人群不可或缺的生活用品，汽车行业在汽车造型的设计中，由于设计师短期内能够设计的汽车造型有限，且从设计草图到比较完善的效果展示图需要较长的设计周期，同时设计期间会有各种不同需求从而产生不同修改，这都使得汽车的设计过程繁琐冗长。
[0003]除此之外，购车用户面对众多琳琅满目的汽车商品，也难以快速描述出自己想要的汽车样品。用户的目标车辆样式无法得到直观表达，经销商也因此不能明确用户需求，这都使得用户购车过程增加了复杂度。
[0004]而现有的汽车图片以及对应的标签数据，已经大量地散布在网络上，通过人工智能技术，利用其对大规模数据的处理、训练、记忆能力，可以使模型在多种内容范围内快速生成出大量的目标数据，这恰恰满足了汽车设计师以及购车用户对迅速得到目标车辆样式图片这一要求。GAN(Generative Adversarial Networks，生成对抗网络)是2014年提出的新兴概念，本身只是一个框架，可以灵活搭配各种损失函数以及生成器判别器进行设计。作为一个生成模型，可以用于对真实数据分布的建模和生成，最典型的就是图像领域。由于其内部训练对抗的机制，可以解决一些传统的机器学习中面临的数据不足的问题，也可以拓展在半监督学习、无监督学习、多任务学习等等领域中。
[0005]目前常用的条件对抗生成网络及其相关变种往往只能针对于某一维度上的条件，限定某一特征来生成对应图像，在汽车生成方面表现出来的结果，体现在生成出的图像内容单一，且约束针对性不足，不能满足人们在对多维度共同限定条件下得到目标汽车图像的要求。

技术实现思路

[0006]本专利技术的目的是解决现有技术中存在的问题。
[0007]本专利技术提出基于生成对抗网络的多标签汽车造型生成方法，主要包括以下步骤：
[0008]1)获取真实汽车图片，并打上标签Fi。i＝1，2，
…
，n。n为标签类别总数。
[0009]所述标签类别包括汽车颜色F1、汽车车头方向F2、汽车品牌F3、汽车车型F4、汽车型号F5、汽车占真实汽车图片的比例F6、车身材质F7。
[0010]2)建立汽车图片对抗生成网络，包括汽车图片生成器G和汽车图片判别器D＝[D1,D2,
…
，Dn]。
[0011]所述汽车图片生成器G包括噪声条件合成模块和生成模块。生成模块包括一个批标准化层、若干上采样块和一个输出块。每个上采样块包括上采样层、3*3卷积层、批标准化层、ReLU激活函数层。输出块包括卷积层和Tanh激活函数层。所述生成模块输出汽车生成图
片。
[0012]所述噪声条件合成模块将高斯噪声与指定的所有i种标签数据进行点乘，从而得到融合条件信息的点乘结果。所述噪声条件合成模块将点乘结果输入至生成模块。
[0013]汽车图片判别器D包括卷积层、LeakyReLU激活函数层、判别块。所述判别块包括卷积层、批标准化层，LeakyReLU激活函数层、输出块。所述输出块包括卷积层、ReLU激活函数层、全连接层、输出层。其中，ReLU激活函数层的输出传输至全连接层，得到输出值r1。ReLU激活函数层的输出与Fi标签的one
‑
hot编码进行点乘，，得到点乘结果r2。输出层输出结果R＝r2+r1。
[0014]汽车图片判别器Di的Fi标签组合损失Li如下所示：
[0015][0016]式中，R
i
是汽车图片判别器Di中输出块全连接层输出结果。c为Fi的种类数目。y
j
为Fi标签的one
‑
hot编码形式。p
j
为上一层网络的输出概率分布。
[0017]汽车图片判别器D的总损失函数L
D
如下所示：
[0018]L
D
＝λ1L1+λ2L2+...+λ
i
L
i
。
ꢀꢀꢀꢀꢀꢀ
(2)
[0019]式中，λ1、λ2、...、λ
i
为用于平衡Fi标签组合损失的超参数。
[0020]3)将高斯噪声和所有i种标签数据输入到汽车图片生成器G中，得到汽车生成图片。
[0021]4)分别利用汽车图片判别器Di和Fi标签，计算汽车生成图片和真实图片中Fi标签数据组合损失。i初始值为1。
[0022]5)将汽车生成图片和真实图片中Fi标签数据组合损失反馈给汽车图片判别器Di，更新汽车图片判别器Di权重。
[0023]6)判断i≤n是否成立，若是，则进入步骤6)，否则，令i＝i+1，并返回步骤4)，
[0024]7)将汽车生成图片所有标签数据组合损失反馈给汽车图片生成器G，更新汽车图片生成器G的权重。
[0025]8)重复步骤3)至步骤7)，直至汽车图片判别器D的总损失函数L
D
低于设定值。
[0026]9)在汽车图片对抗生成网络中输入指定的h种标签，输出对应的汽车生成图片。h≤n。
[0027]本专利技术的有益成果是：通过提出一种汽车图片生成对抗网络，能够根据指定的多个维度的条件标签，生成对应的汽车图片，丰富了生成图片的多方约束，使得结果更加精准、满足用户需求。具体来说，设计了多个判别器+单个生成器的网络结构，生成器根据给定的标签条件和噪声，生成指定图片，两个判别器分别在不同标签的角度判定当前图片的组合损失。这样一来，训练的最终结果，判别器将不仅能判断生成图片的真假，还能在多个条件标签维度对其进行判断，这使得生成器能够得到更多的约束条件，生成的条件更加细化具体，在汽车设计以及汽车描述筛选方面具有巨大的潜力。本专利技术提供的用于产生汽车图片的生成对抗网络，可以为汽车设计师，汽车购买人员快速生成符合指定多种标签的汽车图片，为下一步行动提供参考依据，在汽车设计，汽车购买领域有着巨大的应用空间。
附图说明
[0028]图1是本专利技术的生成对抗网络模型框架图；
具体实施方式
[0029]下面结合实施例对本专利技术作进一步说明，但不应该理解为本专利技术上述主题范围仅限于下述实施例。在不脱离本专利技术上述技术思想的情况下，根据本领域普通技术知识和惯用手段，做出各种替换和变更，均应包括在本专利技术的保护范围内。
[0030]实施例1：
[0031]参见图1，基于生成对抗网络的多标签汽车造型生成方法，主要包括以下步骤：
[0032]1)获取真实汽车图片，并打上标签Fi。i＝1，2，
…
，n。n为标签类别总数。
[0033]所述标签类别包括汽车颜色F1、汽车车头方向F2、汽车品牌F3、汽车车型F4、汽车型号F5、汽车占真实汽车图片的比例F6、车身材质F7、风格F8。车身材质F7根据汽车品牌F3、汽车车型F4、汽车型号F5本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.基于生成对抗网络的多标签汽车造型生成方法，其特征在于，主要包括以下步骤：1)获取所述真实汽车图片，并打上标签Fi；i＝1，2，
…
，n；n为标签类别总数。2)建立汽车图片对抗生成网络，包括汽车图片生成器G和汽车图片判别器D＝[D1,D2,
…
，Dn]；3)将高斯噪声和所有i种标签数据输入到汽车图片生成器G中，得到汽车生成图片。4)分别利用汽车图片判别器Di和Fi标签，计算汽车生成图片和真实图片中Fi标签数据组合损失；i初始值为1；5)将汽车生成图片和真实图片中Fi标签数据组合损失反馈给汽车图片判别器Di，更新汽车图片判别器Di权重；6)判断i≤n是否成立，若是，则进入步骤6)，否则，令i＝i+1，并返回步骤4)，7)将汽车生成图片所有标签数据组合损失反馈给汽车图片生成器G，更新汽车图片生成器G的权重；8)重复步骤3)至步骤7)，直至汽车图片判别器D的总损失函数L
D
低于设定值；9)在汽车图片对抗生成网络中输入指定的h种标签，输出对应的汽车生成图片；h≤n。2.根据权利要求1所述的基于生成对抗网络的多标签汽车造型生成方法，其特征在于：所述标签类别包括汽车颜色F1、汽车车头方向F2、汽车品牌F3、汽车车型F4、汽车型号F5、汽车占真实汽车图片的比例F6、车身材质F7。3.根据权利要求1所述的基于生成对抗网络的多标签汽车造型生成方法，其特征在于：所述汽车图片生成器G包括噪声条件合成模块和生成模块；生成模块包括一个批标准化层、若干上采样块和一个输出块；每个上采样块包括上采样层、3*3卷积层、批标准化层、ReLU激活函数层；输出块包括卷积层和Tanh激活函数层；所述生成模块输出汽车生成图片。4.根据权利要...

【专利技术属性】
技术研发人员：刘礼，张文彬，廖军，罗皓，钱爽，雍滋蕊，
申请(专利权)人：重庆大学，
类型：发明
国别省市：