[发明专利]图像预处理方法、装置、设备以及存储介质

权利要求书

1.一种图像预处理方法，包括：

获取人脸的RGBD图像，其中，所述RGBD图像包括RGB图像和深度图像；

从所述RGB图像中提取RGB脸部和头发掩膜；

从所述深度图像中提取深度脸部和头发掩膜；

基于所述RGB脸部和头发掩膜对所述RGB图像进行过滤，得到RGB脸部区域和RGB头发区域，以及基于所述深度脸部和头发掩膜对所述深度图像进行过滤，得到深度脸部区域和深度头发区域。

2.根据权利要求1所述的方法，其中，所述从所述RGB图像中提取RGB脸部和头发掩膜，包括：

将所述RGB图像输入至预先训练的深度学习分割网络，得到所述RGB脸部和头发掩膜。

3.根据权利要求2所述的方法，其中，所述深度学习分割网络通过如下步骤训练得到：

获取样本RGB图像，其中，所述样本RGB图像存在人脸；

对所述样本RGB图像中的脸部和头发进行像素级标注，得到样本标注RGB图像；

基于所述样本标注RGB图像，生成样本RGB脸部和头发掩膜；

将所述样本RGB图像作为输入，将所述样本RGB脸部和/头发掩膜作为输出，训练得到所述深度学习分割网络。

4.根据权利要求2所述的方法，其中，所述从所述深度图像中提取深度脸部和头发掩膜，包括：

计算所述深度图像的深度直方图；

查找所述深度直方图的高峰；

在所述深度直方图中，从所述高峰处向前后两个方向前进，直至深度值小于预设阈值时停止，以及在停止处进行截断；

基于所述深度直方图中截断得到的部分，生成所述深度脸部和头发掩膜。

5.根据权利要求4所述的方法，其中，在所述查找所述深度直方图的高峰之前，还包括：

对所述深度直方图进行直方图平滑。

6.根据权利要求4所述的方法，其中，在所述基于所述RGB脸部和头发掩膜对所述RGB图像进行过滤，得到RGB脸部区域和RGB头发区域，以及基于所述深度脸部和头发掩膜对所述深度图像进行过滤，得到深度脸部区域和深度头发区域之前，还包括：

对所述RGB脸部和头发掩膜、所述深度脸部和头发掩膜中的至少一项做形态学处理，其中，所述形态学处理包括以下至少一项：形态学腐蚀、形态学膨胀、形态学开与形态学闭处理。

7.根据权利要求1所述的方法，其中，所述方法还包括：

将所述深度头发区域通过非零均值滤波器，对所述非零均值滤波器的滤波核内的像素统计非零均值，以及利用所述非零均值替代所述滤波核内的像素。

8.根据权利要求7所述的方法，其中，所述方法还包括：

对所述深度脸部区域做形态学处理。

9.根据权利要求1-8之一所述的方法，其中，所述方法还包括：

基于所述RGB脸部区域、所述RGB头发区域、所述深度脸部区域和所述深度头发区域进行人脸三维重建，得到人脸的三维模型。

10.一种图像预处理装置，包括：

获取模块，被配置成获取人脸的RGBD图像，其中，所述RGBD图像包括RGB图像和深度图像；

第一提取模块，被配置成从所述RGB图像中提取RGB脸部和头发掩膜；

第二提取模块，被配置成从所述深度图像中提取深度脸部和头发掩膜；

第一过滤模块，被配置成基于所述RGB脸部和头发掩膜对所述RGB图像进行过滤，得到RGB脸部区域和RGB头发区域，以及基于所述深度脸部和头发掩膜对所述深度图像进行过滤，得到深度脸部区域和深度头发区域。

11.根据权利要求10所述的装置，其中，所述第一提取模块进一步被配置成：

将所述RGB图像输入至预先训练的深度学习分割网络，得到所述RGB脸部和头发掩膜。