[发明专利]一种基于Transformer的人脸美丽预测方法、装置及存储介质

权利要求书

1.一种基于Transformer的人脸美丽预测方法，其特征在于，所述方法包括：

采集人脸美丽数据库的美丽人脸图像，以及对所述美丽人脸图像进行预处理，得到图像块信息；

将所述美丽人脸图像输入到预设的知识蒸馏网络中，得到蒸馏标记信息；

将所述图像块信息、所述蒸馏标记信息和预设类别标记输入到预设的Transformer网络中，得到人脸美丽分类信息。

2.根据权利要求1所述的基于Transformer的人脸美丽预测方法，其特征在于，所述对所述美丽人脸图像进行预处理，得到图像块信息，包括：

对所述美丽人脸图像进行数据增强处理，得到人脸图像增强信息；

对所述人脸图像增强信息进行裁剪处理，得到所述图像块信息。

3.根据权利要求1所述的基于Transformer的人脸美丽预测方法，其特征在于，所述知识蒸馏网络包括教师网络，所述将所述美丽人脸图像输入到预设的知识蒸馏网络中，得到蒸馏标记信息，包括：

将所述美丽人脸图像输入到预设的所述知识蒸馏网络中，得到从所述教师网络中输出的所述蒸馏标记信息。

4.根据权利要求1所述的基于Transformer的人脸美丽预测方法，其特征在于，所述Transformer网络包括自注意力层和多层感知机，所述将所述图像块信息、所述蒸馏标记信息和预设类别标记输入到预设的Transformer网络中，得到人脸美丽分类信息，包括：

将所述图像块信息、所述蒸馏标记信息和所述预设类别标记分别与所述自注意力层进行交互处理，得到人脸分类信息；

将所述人脸分类信息输入到所述多层感知机，以及根据所述多层感知机的损失函数，得到所述人脸美丽分类信息。

5.根据权利要求2所述的基于Transformer的人脸美丽预测方法，其特征在于，所述数据增强处理包括自增强、随机增强和随机擦除。

6.一种基于Transformer的人脸美丽预测装置，其特征在于，包括：

第一单元，用于采集人脸美丽数据库的美丽人脸图像，以及对所述美丽人脸图像进行预处理，得到图像块信息；

第二单元，用于将所述美丽人脸图像输入到预设的知识蒸馏网络中，得到蒸馏标记信息；

第三单元，用于将所述图像块信息、所述蒸馏标记信息和预设类别标记输入到预设的Transformer网络中，得到人脸美丽分类信息。

7.根据权利要求6所述的基于Transformer的人脸美丽预测装置，其特征在于，所述第一单元包括：

第四单元，用于对所述美丽人脸图像进行数据增强处理，得到人脸图像增强信息；

第五单元，用于对所述人脸图像增强信息进行裁剪处理，得到所述图像块信息。

8.根据权利要求6所述的基于Transformer的人脸美丽预测装置，其特征在于，所述第三单元包括：

第六单元，用于将所述图像块信息、所述蒸馏标记信息和所述预设类别标记分别与所述自注意力层进行交互处理，得到人脸分类信息；

第七单元，用于将所述人脸分类信息输入到所述多层感知机，以及根据所述多层感知机的损失函数，得到所述人脸美丽分类信息。

9.根据权利要求7所述的基于Transformer的人脸美丽预测装置，其特征在于，所述数据增强处理包括自增强、随机增强和随机擦除。

10.一种计算机可读存储介质，所述计算机可读存储介质存储有计算机可执行指令，其特征在于，所述计算机可执行指令被控制处理器执行时实现如权利要求1至5任一所述的基于Transformer的人脸美丽预测方法。