[发明专利]痘痘类型识别模型训练方法及相关装置

说明书

本申请公开了一种痘痘类型识别模型训练方法及相关装置。其中方法包括：获取样本数据集，所述样本数据集包括多个属于不同痘痘类型的样本图像；获取每个痘痘类型所对应的一个标准图像，所述一个标准图像用于代表一个痘痘类型的痘痘样式；计算各个所述样本图像的类别概率标签，所述类别概率标签包括所述样本图像与每个所述标准图像之间的相似度信息；基于所述各个样本图像和所述各个样本图像对应的类别概率标签训练分类网络模型，获得痘痘类型识别模型。

技术领域

本发明涉及计算机视觉技术领域，尤其是涉及一种痘痘类型识别模型训练方法及相关装置。

背景技术

随着移动通信技术的快速发展以及人民生活水平的提升，各种智能终端已广泛应用于人民的日常工作和生活，使得人们越来越习惯于使用终端应用程序，使得美颜自拍、拍照测肤此类功能的APP需求也变得越来越多，对脸部痘痘情况的分析也有很大的需求，可以根据痘痘类型，针对性地提出或查询皮肤改善方案。

然而痘痘面积小，不同类别之间可区分的特征不明显，导致目前对痘痘类型识别的准确度较低。

发明内容

本申请提供了一种痘痘类型识别模型训练方法及相关装置。

第一方面，提供了一种痘痘类型识别模型训练方法，包括：

获取样本数据集，所述样本数据集包括多个属于不同痘痘类型的样本图像；

获取每个痘痘类型所对应的一个标准图像，所述一个标准图像用于代表一个痘痘类型的痘痘样式；

计算各个所述样本图像的类别概率标签，所述类别概率标签包括所述样本图像与每个所述标准图像之间的相似度概率；

基于所述各个样本图像和所述各个样本图像对应的类别概率标签训练分类网络模型，获得痘痘类型识别模型。

第二方面，提供了一种痘痘类型识别模型训练装置，包括：

获取模块，用于获取样本数据集，所述样本数据集包括多个属于不同痘痘类型的样本图像；

所述获取模块还用于，获取每个痘痘类型所对应的一个标准图像，所述一个标准图像用于代表一个痘痘类型的痘痘样式；

计算模块，用于计算各个所述样本图像的类别概率标签，所述类别概率标签包括所述样本图像与每个所述标准图像之间的相似度信息；

训练模块，用于基于所述各个样本图像和所述各个样本图像对应的类别概率标签训练分类网络模型，获得痘痘类型识别模型。

第三方面，提供了一种计算机存储介质，所述计算机存储介质存储有一条或多条指令，所述一条或多条指令适于由处理器加载并执行如上述第一方面及其任一种可能的实现方式的步骤。

本申请实施例通过获取样本数据集，上述样本数据集包括多个属于不同痘痘类型的样本图像，获取每个痘痘类型所对应的一个标准图像，上述一个标准图像用于代表一个痘痘类型的痘痘样式，计算各个上述样本图像的类别概率标签，上述类别概率标签包括上述样本图像与每个上述标准图像之间的相似度信息，再基于上述各个样本图像和上述各个样本图像对应的类别概率标签训练分类网络模型，获得痘痘类型识别模型，可以对痘痘类型进行较准确识别。传统的分类标签方式，分类的准确度不高，而本申请中在模型训练时采用样本图像与每个所述标准图像之间的相似度信息进行标注，可以从标签上体现出类别之间的相似程度，将痘痘类型标签概率化，使得模型能够学习痘痘类型之间的相似特征，从而提升痘痘类型识别的准确度。

附图说明

为了更清楚地说明本申请实施例或背景技术中的技术方案，下面将对本申请实施例或背景技术中所需要使用的附图进行说明。

图1为本申请实施例提供的一种痘痘类型识别模型训练方法的流程示意图；