[发明专利]数据增广方法、装置、电子设备及存储介质

权利要求书

1.一种数据增广方法，其特征在于，所述方法包括：

获取口罩模板数据和待增广人脸数据集，其中，所述口罩模板数据包括彩色口罩模板数据和深度口罩模板数据；

利用预设的人脸重建网络对所述彩色口罩模板数据、所述深度口罩模板数据和所述待增广人脸数据集进行处理，生成增广人脸数据集；

其中，所述获取口罩模板数据包括：

获取样本数据集，其中，所述样本数据集包括多视角人脸彩色数据和多视角人脸深度数据；对所述多视角人脸彩色数据进行分割处理获取多视角口罩数据；

利用所述人脸重建网络对所述多视角人脸彩色数据、所述多视角人脸深度数据和所述多视角口罩数据进行重建，获取多视角彩色数据、多视角点云数据和多视角口罩掩膜数据；

对所述多视角彩色数据和所述多视角口罩掩膜数据进行融合处理获取所述彩色口罩模板数据；对所述多视角点云数据和所述多视角口罩掩膜数据进行融合处理获取所述深度口罩模板数据。

2.根据权利要求1所述的数据增广方法，其特征在于，所述利用所述人脸重建网络对所述多视角人脸彩色数据、所述多视角人脸深度数据和所述多视角口罩数据进行重建之前还包括：对所述多视角人脸深度数据进行滤波处理和补洞处理。

3.根据权利要求1所述的数据增广方法，其特征在于，所述待增广人脸数据集包括待增广人脸彩色数据和待增广人脸深度数据；

所述利用所述人脸重建网络对所述彩色口罩模板数据、所述深度口罩模板数据和所述待增广人脸数据集进行处理，生成增广人脸数据集包括：

利用所述人脸重建网络对所述待增广人脸彩色数据和所述待增广人脸深度数据进行重建处理，获取待增广彩色数据和待增广深度数据；

对所述待增广彩色数据和所述彩色口罩模板数据进行融合处理获取彩色口罩数据；

对所述待增广深度数据和所述深度口罩模板数据进行融合处理获取深度口罩数据；

分别将所述彩色口罩数据和所述深度口罩数据渲染至所述待增广人脸彩色数据和所述待增广人脸深度数据上，生成所述增广人脸数据集，其中，所述增广人脸数据集包括人脸增广彩色数据和人脸增广深度数据。

4.根据权利要求3所述的数据增广方法，其特征在于，所述对所述待增广彩色数据和所述彩色口罩模板数据进行融合处理获取彩色口罩数据之前还包括：

对所述待增广彩色数据和所述彩色口罩模板数据进行颜色空间转换处理，获取彩色转换数据和彩色口罩转换数据；

根据所述彩色转换数据的亮度对所述彩色口罩转换数据的亮度进行线性处理；

根据所述彩色转换数据的饱和度对所述彩色口罩转换数据的饱和度进行线性处理；

对所述彩色转换数据和线性处理后的所述彩色口罩转换数据进行加色模式转换处理，获取所述待增广彩色数据和线性彩色口罩模板数据。

5.根据权利要求3所述的数据增广方法，其特征在于，所述对所述待增广深度数据和所述深度口罩模板数据进行融合处理获取深度口罩数据包括：

对所述深度口罩模板数据进行网络化处理，获取所述深度口罩模板数据的边界点数据；

利用预设的刚性损失函数、预设的非刚性损失函数对所述深度口罩模板数据的边界点数据进行处理；

对处理后的所述深度口罩模板数据的边界点数据和所述待增广深度数据进行融合处理，获取所述深度口罩数据。

6.根据权利要求1所述的数据增广方法，其特征在于，所述利用预设的人脸重建网络对所述彩色口罩模板数据、所述深度口罩模板数据和所述待增广人脸数据集进行处理之前还包括：获取彩色人脸训练数据，对所述彩色人脸训练数据进行口罩增广处理获取口罩人脸训练数据；利用所述彩色人脸训练数据和所述口罩人脸训练数据训练所述预设的人脸重建网络。