[发明专利]一种构造训练数据集的方法和装置

权利要求书

1.一种构造训练数据集的方法，其特征在于，包括：

从待标注数据集中随机选取两张待标注图像作为第一待标注图像和第二待标注图像，并确定第一待标注图像的第一基础美学标签和第二待标注图像的第二基础美学标签；

将所述第一待标注图像和第二待标注图像作为输入数据，输入预训练的图像美学比较模型，得到第一待标注图像的第一评价因子和第二待标注图像的第二评价因子；

根据所述第一评价因子和所述第二评价因子，更新所述第一基础美学标签和所述第二基础美学标签；

重复上述步骤，直至重复次数达到阈值，将最终得到的基础美学标签作为待标注图像的最终美学标签，以得到训练数据集。

2.根据权利要求1所述的方法，其特征在于，确定第一待标注图像的第一基础美学标签和第二待标注图像的第二基础美学标签包括：

将第一待标注图像和第二待标注图像分别作为输入数据，输入预训练的基础美学评价模型，得到第一待标注图像的第一基础美学标签和第二待标注图像的第二基础美学标签。

3.根据权利要求1所述的方法，其特征在于，根据如下过程训练得到所述图像美学比较模型：

获取各类目的多组图像对，并通过人工方式对每一组图像对进行标注，以生成美学比较训练集；

对所述美学比较训练集进行训练，得到图像美学比较模型。

4.根据权利要求3所述的方法，其特征在于，对所述美学比较训练集进行训练，得到图像美学比较模型包括：

设置多种训练方法和多种模型复杂度；基于所述多种训练方法和多种模型复杂度，对所述美学比较训练集进行训练，得到多个图像美学比较模型。

5.根据权利要求4所述的方法，其特征在于，将所述第一待标注图像和第二待标注图像作为输入数据，输入预训练的图像美学比较模型，得到第一待标注图像的第一评价因子和第二待标注图像的第二评价因子包括：

将所述第一待标注图像和第二待标注图像作为输入数据，输入所述多个图像美学比较模型中的其中一个，得到第一待标注图像的第一评价因子和第二待标注图像的第二评价因子；

根据所述第一评价因子和所述第二评价因子，更新所述第一基础美学标签和所述第二基础美学标签包括：

根据所述第一评价因子和所述第二评价因子，更新所述第一基础美学标签和所述第二基础美学标签，将更新后的第一基础美学标签和第二基础美学标签分别作为新的第一基础美学标签和新的第二基础美学标签；

将所述第一待标注图像和第二待标注图像作为输入数据，输入其他图像美学比较模型，得到新的第一评价因子和新的第二评价因子；

基于所述新的第一评价因子和新的第二评价因子，更新所述新的第一基础美学标签和新的第二基础美学标签；

重复执行上述步骤，直至遍历完所述多个图像美学比较模型。

6.根据权利要求1-5任一项所述的方法，其特征在于，根据下式更新所述第一基础美学标签和所述第二基础美学标签：

其中，Ra表示第一基础美学标签，Rb表示第二基础美学标签，Sa表示第一评价因子，Sb表示第二评价因子，k和β分别为预设参数，Ra^*表示更新后的第一基础美学标签，Rb^*表示更新后的第二基础美学标签。

7.根据权利要求6所述的方法，其特征在于，将最终得到的基础美学标签作为待标注图像的最终美学标签包括：

将最终得到的基础美学标签进行归一化，将归一化后的基础美学标签作为待标注图像的最终美学标签。