[发明专利]一种基于双重标签分布的轻量化人脸美学预测方法

说明书

本发明提供了一种基于双重标签分布的轻量化人脸美学预测方法，包括：获取人脸图片，对每张人脸图片进行预处理，将预处理后的人脸图片划分为训练集和测试集；对训练集中的每张人脸图片构建其对应的双重标签分布；所述双重标签分布包括吸引力分布和评分分布；将训练集中的人脸图片和其所对应的吸引力分布输入至轻量化卷积神经网络中，通过联合学习模块进行优化训练，得到优化后的人脸美学预测网络模型；将需要预测的人脸图片输入至优化后的人脸美学预测网络模型中，输出预测的吸引力分布，根据预测的吸引力分布得到预测的人脸图片的美丽分数。该方法提出了双重标签分布，同时进行轻量化设计，使预测性能优越、预测效率和速度都大幅提升。

技术领域

本发明涉及计算机视觉与图像处理技术领域，特别地，涉及一种基于双重标签分布的轻量化人脸美学预测方法。

背景技术

美丽一直为人们津津乐道的话题，然而如何评价一张人脸图像的美丽程度是一个抽象的问题。评价人脸的美丽程度很有可能受到主观因素的影响，不同的人有不同的评价标准，现有人研究通过计算机对人脸美丽进行较为客观的评价，人脸美丽评价可以广泛地应用在日常生活中，如在社交网络中让计算机为好友进行美丽评价并分享。

但是，现有的研究绝大部分采用了规模庞大的神经网络结构，例如：VGG、ResNet、ResNeXt系列等，忽略了轻量化设计。现有方法获得的预测模型虽然预测性能良好，但存在规模庞大、预测效率低下等诸多缺点，导致模型的适用范围十分有限，难以用于手机、嵌入式设备等资源有限的终端，在落地应用方面存在诸多约束与不便。同时，大部分研究将人脸美学预测视作简单分类或回归问题，没有充分利用数据集提供的各类信息。具体而言，在建模过程中，大部分方法只利用了数据集的分数标签作为监督信息，而忽略了分数的标准差和评分分布等信息，造成信息浪费，且预测不够准确，性能存在提升空间。

因此，业内急需用于人脸美学预测的方法的新型技术。

发明内容

针对上述现有技术中存在的技术问题，本发明提供了一种基于双重标签分布的轻量化人脸美学预测方法，该方法提出了双重标签分布，同时进行轻量化设计，使预测性能优越，在主流数据集上达到目前最好的预测性能，同时大幅提升预测效率和速度。

为实现上述目的，本发明提供了一种基于双重标签分布的轻量化人脸美学预测方法，包括以下步骤：

获取人脸图片，对每张人脸图片进行预处理，将预处理后的人脸图片划分为训练集和测试集；

对训练集中的每张人脸图片构建每张人脸图片对应的双重标签分布；所述双重标签分布包括吸引力分布和评分分布；

将训练集中的人脸图片和训练集中人脸图片所对应的吸引力分布输入至轻量化卷积神经网络中，通过联合学习模块进行优化训练，得到优化后的人脸美学预测网络模型；其中，所述联合学习模块包括吸引力分布学习模块、评分分布学习模块以及美丽分数回归学习模块；

将训练集中的人脸图片所对应的吸引力分布输入至所述吸引力分布学习模块，输出预估吸引力分布；通过训练集中人脸图片所对应的吸引力分布与输出的预估吸引力分布，得到吸引力分布损失函数；

将所述预估吸引力分布输入至所述评分分布学习模块，输出预估评分分布，通过训练集中人脸图片所对应的评分分布和预估评分分布，得到评分分布损失函数；

将训练集中的人脸图片所对应的预估吸引力分布输入至美丽分数回归学习模块，输出预测的美丽分数，通过训练集中人脸图片所对应的真实美丽分数与预测的美丽分数，得到美丽分数回归损失函数；

根据所述吸引力分布损失函数、评分分布损失函数和美丽分数回归损失函数构建联合损失函数；通过最小化联合损失函数，得到优化后的人脸美学预测网络模型；

将测试集中需要预测的人脸图片输入至优化后的人脸美学预测网络模型中，输出预测的吸引力分布，根据预测的吸引力分布得到预测的人脸图片的美丽分数。