[发明专利]基于生成对抗网络的服饰搭配生成方法

摘要

本发明提供了基于生成对抗网络的服饰搭配生成方法，该方法包括：收集在线用户的服饰物品搭配生成训练集和测试集；采用所述训练集训练生成对抗网络服饰搭配生成模型；采用所述测试集对训练完成的生成对抗网络服饰搭配生成模型进行测试；对测试效果进行验证，并给出了生成对抗网络服饰搭配生成模型的结构和训练方式。本发明提出的基于生成对抗网络的服饰搭配生成方法，在实际应用中，对于较小的数据集，其生成对抗网络服饰搭配生成模型可以解决因缺少足够多的备选数据而导致的推荐不充分的问题；对于较大的数据集，可以在数据集中搜索与生成的服饰图片最类似的服饰，以解决推荐复杂度过高的问题。

主权项

1.一种基于生成对抗网络的服饰搭配生成方法，其特征在于，该方法包括如下步骤：/nS1)收集总共由797名在线用户精心制作的共208814套服饰物品图像搭配作为数据集，对于每个用户，分别选取221和41套服饰物品图像搭配用于训练和测试，其中每套服饰物品图像搭配由两件来自不同类别的服饰单品组成，即上衣和下衣，在797名在线用户中，训练集共包括102217件上衣、76245件下衣以及176137套服饰物品图像搭配；测试集共包括26899件上衣、23642件下衣以及32677套服饰物品图像搭配；/nS2)采用步骤S1)收集的训练集训练生成对抗网络服饰搭配生成模型；/nS3)采用步骤S1)收集的测试集对步骤S2)训练完成的生成对抗网络服饰搭配生成模型进行测试，以完成如下两个任务：一、给定一件上衣作为给定输入查询服饰物品图像，设计一件下衣作为目标服饰物品图像与之搭配；二、给定一件下衣作为给定输入查询服饰物品图像，设计一件上衣作为目标服饰物品图像与之搭配；/nS4)对步骤S3)的测试效果进行验证。/n