[发明专利]年龄估计模型的训练方法、年龄估计方法以及对应的装置

摘要

本申请实施例提供了一种年龄估计模型的训练方法、年龄估计方法以及对应的装置，该训练方法包括：获取训练样本，各样本为标注有对应的年龄标签的人脸图像；确定各样本的年龄标签所对应的年龄标签分布；基于样本以及年龄标签分布，对年龄估计模型进行训练，直至年龄估计模型的总损失函数收敛；其中，总损失函数包括分布损失函数和回归损失函数，分布损失函数表征了样本的年龄标签分布和预测年龄分布的差异，回归损失函数表征了样本的年龄标签和预测年龄值的差异。基于本申请实施例所提供的方案，能够有效提供年龄预测的准确性，更好的满足了实际应用需求。

主权项

1.一种年龄估计模型的训练方法，其特征在于，所述年龄估计模型包括依次级联的特征提取模块、概率分布预测模块和年龄值预测模块，其中，所述特征提取模块，用于提取输入至模型的人脸图像的图像特征，所述概率分布预测模块，用于根据所述图像特征输出所述人脸图像中的人脸年龄对应于各个预划分的年龄端点值的预测年龄分布；所述年龄值预测模块，用于根据所述预测年龄分布输出所述人脸年龄的预测年龄值；所述训练方法包括：获取训练样本，各所述样本为标注有对应的年龄标签的人脸图像；确定各所述样本的年龄标签所对应的年龄标签分布；基于所述样本以及所述年龄标签分布，对所述年龄估计模型进行训练，直至所述年龄估计模型的总损失函数收敛；其中，所述总损失函数包括分布损失函数和回归损失函数，所述分布损失函数表征了所述样本的年龄标签分布和预测年龄分布的差异，所述回归损失函数表征了所述样本的年龄标签和预测年龄值的差异。