[发明专利]基于离散小波分解的深度图像空洞填充方法及装置

权利要求书

1.一种基于离散小波分解的深度图像空洞填充方法，其特征在于，包括：

获取原始深度图像并进行离散小波变换，得到第一低频信息图像和第一高频信息图像；

对所述第一低频信息图像中的空洞点进行填充，得到空洞填充后的第一低频信息图像；

对所述原始深度图像进行二值化处理，得到二值化深度图像；

对所述二值化深度图像进行离散小波变换，得到第二低频信息图像和第二高频信息图像；

利用所述第二高频信息图像对所述第一高频信息图像进行对比修复，得到修复后的第一高频信息图像；

对所述空洞填充后的第一低频信息图像和所述修复后的第一高频信息图像进行离散小波反变换，得到空洞填充后的初始图像；

将所述空洞填充后的初始图像与所述原始深度图像进行融合处理，得到最终的深度图像；

其中，对所述第一低频信息图像中的空洞点进行填充，得到空洞填充后的第一低频信息图像包括：

根据所述第一低频信息图像中各像素点的深度值得到该图像中的空洞点和非空洞点；

计算当前空洞点与其邻域内各非空洞点的相似度权重；

根据所述相似度权重计算当前空洞点的深度估计值，并用该深度估计值填充当前空洞点；

遍历所述第一低频信息图像，以对所有空洞点进行填充，得到空洞填充后的第一低频信息图像。

2.根据权利要求1所述的基于离散小波分解的深度图像空洞填充方法，其特征在于，对所述原始深度图像进行二值化处理，得到二值化深度图像，包括：

获取所述原始深度图像中的空洞点和非空洞点；

按照空洞点为0、非空洞点为1的方式对所述原始深度图像进行二值化处理，得到二值化深度图像。

3.根据权利要求1所述的基于离散小波分解的深度图像空洞填充方法，其特征在于，利用所述第二高频信息图像对所述第一高频信息图像进行修复，得到修复后的第一高频信息图像，包括：

将所述第一高频信息图像与所述第二高频信息图像进行对比，当相同位置的像素值均不为零时，将所述第一高频信息图像中对应位置的像素值置零，以获得修复后的第一高频信息图像。

4.根据权利要求1所述的基于离散小波分解的深度图像空洞填充方法，其特征在于，将所述空洞填充后的初始图像与所述原始深度图像进行融合处理，包括：

遍历所述空洞填充后的初始图像，并对每个像素点重新进行深度填充，若当前点在所述原始深度图像中的对应点为非空洞点，则利用所述原始深度图像中该点的深度值替换当前点的深度值；否则，保留当前点的深度值。

5.一种基于离散小波分解的深度图像空洞填充装置，其特征在于，包括：

第一变换模块(1)，用于获取原始深度图像并进行离散小波变换，得到第一低频信息图像和第一高频信息图像；

低频填充模块(2)，用于对所述第一低频信息图像中的空洞点进行填充，得到空洞填充后的第一低频信息图像；

二值化模块(3)，用于对所述原始深度图像进行二值化处理，得到二值化深度图像；

第二变换模块(4)，用于对所述二值化深度图像进行离散小波变换，得到第二低频信息图像和第二高频信息图像；

高频修复模块(5)，用于利用所述第二高频信息图像对所述第一高频信息图像进行修复，得到修复后的第一高频信息图像；

反变换模块(6)，用于对所述空洞填充后的第一低频信息图像和所述修复后的第一高频信息图像进行离散小波反变换，得到空洞填充后的初始图像；

融合处理模块(7)，用于将所述空洞填充后的初始图像与所述原始深度图像进行融合处理，得到最终的深度图像；

其中，所述低频填充模块(2)具体用于：

根据所述第一低频信息图像中各像素点的深度值得到该图像中的空洞点和非空洞点；

计算当前空洞点与其邻域内各非空洞点的相似度权重；

根据所述相似度权重计算当前空洞点的深度估计值，并用该深度估计值填充当前空洞点；

遍历所述第一低频信息图像，以对所有空洞点进行填充，得到空洞填充后的第一低频信息图像。