[发明专利]一种建立训练集的方法、装置、电子设备以及介质

权利要求书

1.一种建立训练集的方法，其特征在于，所述方法包括：

获取预设训练集，所述预设训练集包括：预先收集的多个具有特定人脸属性的样本图像；

将所述预设训练集中的样本图像作为标注，对初始图像生成模型进行训练，得到人脸图像生成模型；

基于所述人脸图像生成模型，生成多个具有所述特定人脸属性的候选图像；

基于所述人脸图像生成模型生成的多个具有所述特定人脸属性的候选图像，生成用于训练人脸识别模型的目标训练集。

2.根据权利要求1所述的方法，其特征在于，所述基于所述人脸图像生成模型生成的多个具有所述特定人脸属性的候选图像，生成用于训练人脸识别模型的目标训练集，包括：

确定每个候选图像的噪声特征值和/或分辨率；

将噪声特征值小于预设噪声特征值阈值，和/或分辨率大于预设分辨率阈值的候选图像加入所述目标训练集。

3.根据权利要求1或2所述的方法，其特征在于，在所述生成用于训练人脸识别模型的目标训练集之后，所述方法还包括：

基于预设的图像变换工具，分别对所述目标训练集中的每个图像进行图像变换，分别得到每个图像对应的多个变换图像；

将所述目标训练集中的每个图像对应的多个变换图像加入所述目标训练集。

4.根据权利要求3所述的方法，其特征在于，所述基于预设的图像变换工具，分别对所述目标训练集中的每个图像进行图像变换，分别得到每个图像对应的多个变换图像，包括：

针对所述目标训练集中的每个图像，基于预设的图像编辑算法，对所述目标训练集中的每个图像进行镜像、旋转、变形操作中的至少一种操作；和/或，

基于预设的图像增强网络模型，增强或者抑制所述目标训练集中的每个图像中的像素点单元。

5.一种建立训练集的装置，其特征在于，所述装置包括：

获取模块，用于获取预设训练集，所述预设训练集包括：预先收集的多个具有特定人脸属性的样本图像；

训练模块，用于将所述预设训练集中的样本图像作为标注，对初始图像生成模型进行训练，得到人脸图像生成模型；

第一生成模块，用于基于所述人脸图像生成模型，生成多个具有所述特定人脸属性的候选图像；

第二生成模块，用于基于所述人脸图像生成模型生成的多个具有所述特定人脸属性的候选图像，生成用于训练人脸识别模型的目标训练集。

6.根据权利要求5所述的装置，其特征在于，所述第二生成模块，具体用于：

确定每个候选图像的噪声特征值和/或分辨率；

将噪声特征值小于预设噪声特征值阈值，和/或分辨率大于预设分辨率阈值的候选图像加入所述目标训练集。

7.根据权利要求5或6所述的装置，其特征在于，所述装置还包括：图像变换模块；

所述图像变换模块，用于基于预设的图像变换工具，分别对所述目标训练集中的每个图像进行图像变换，分别得到每个图像对应的多个变换图像；

所述第二生成模块，还用于将所述目标训练集中的每个图像对应的多个变换图像加入所述目标训练集。

8.根据权利要求7所述的装置，其特征在于，所述图像变换模块，具体用于：

针对所述目标训练集中的每个图像，基于预设的图像编辑算法，对所述目标训练集中的每个图像进行镜像、旋转、变形操作中的至少一种操作；和/或，

基于预设的图像增强网络模型，增强或者抑制所述目标训练集中的每个图像中的像素点单元。

9.一种电子设备，其特征在于，包括处理器、通信接口、存储器和通信总线，其中，处理器，通信接口，存储器通过通信总线完成相互间的通信；

存储器，用于存放计算机程序；

处理器，用于执行存储器上所存放的程序时，实现权利要求1-4任一所述的方法步骤。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质内存储有计算机程序，所述计算机程序被处理器执行时实现权利要求1-4任一所述的方法步骤。