[发明专利]基于单目深度估计的深度图压缩方法、装置、设备及介质

权利要求书

1.一种基于单目深度估计的深度图压缩方法，其特征在于，所述方法包括下述步骤：

接收输入的彩色图像和与所述彩色图像对应的原始深度图；

根据接收到的所述彩色图像和所述原始深度图，通过预设的深度图压缩模型对所述原始深度图进行压缩，得到对应的压缩深度图，其中，所述深度图压缩模型为对称结构，且由单目深度估计模块、深度压缩模块以及掩膜生成与处理模块组成。

2.如权利要求1所述的方法，其特征在于，所述通过预设的深度图压缩模型对所述原始深度图进行压缩的步骤，包括：

通过所述单目深度估计模块对所述彩色图像进行处理，得到对应的预测深度图；

计算所述预测深度图与所述原始深度图之间的残差信息；

通过所述深度压缩模块对所述残差信息进行处理，得到对应的残差深度图；

通过所述掩膜生成与处理模块对所述原始深度图进行处理，得到对应的图像掩膜；

根据预设的重建公式，将所述预测深度图、所述残差深度图以及所述图像掩膜进行组合，得到所述压缩深度图。

3.如权利要求2所述的方法，其特征在于，所述通过所述单目深度估计模块对所述彩色图像进行处理的步骤，包括：

通过所述单目深度估计模块中的编码器提取所述彩色图像的深度特征；

根据提取到的所述深度特征，通过所述单目深度估计模块中的解码器生成所述预测深度图。

4.如权利要求2所述的方法，其特征在于，所述通过所述深度压缩模块对所述残差信息进行处理的步骤，包括：

通过所述深度压缩模块中的编码器对所述残差信息进行变换，得到对应的潜在表示；

通过所述深度压缩模块中的解码器对所述潜在表示进行重建，得到所述残差深度图。

5.如权利要求4所述的方法，其特征在于，在所述通过所述深度压缩模块中的编码器对所述残差信息进行变换的步骤之后，在所述通过所述深度压缩模块中的解码器对所述潜在表示进行重建的步骤之前，所述方法还包括：

通过所述深度压缩模块中的超先验模块对所述潜在表示中的相关性进行估计并消除。

6.一种基于单目深度估计的深度图压缩装置，其特征在于，所述装置包括：

图像输入单元，用于接收输入的彩色图像和与所述彩色图像对应的原始深度图；以及

图像压缩单元，用于根据接收到的所述彩色图像和所述原始深度图，通过预设的深度图压缩模型对所述原始深度图进行压缩，得到对应的压缩深度图，其中，所述深度图压缩模型为对称结构，由单目深度估计模块、深度压缩模块以及掩膜生成与处理模块组成。

7.如权利要求6所述的装置，其特征在于，所述图像压缩单元包括：

第一处理单元，用于通过所述单目深度估计模块对所述彩色图像进行处理，得到对应的预测深度图；

信息计算单元，用于计算所述预测深度图与所述原始深度图之间的残差信息；

第二处理单元，用于通过所述深度压缩模块对所述残差信息进行处理，得到对应的残差深度图；

第三处理单元，用于通过所述掩膜生成与处理模块对所述原始深度图进行处理，得到对应的图像掩膜；以及

图像组合单元，用于根据预设的重建公式，将所述预测深度图、所述残差深度图以及所述图像掩膜进行组合，得到所述压缩深度图。

8.如权利要求7所述的装置，其特征在于，所述第一处理单元包括：

深度特征提取单元，用于通过所述单目深度估计模块中的编码器提取所述彩色图像的深度特征；以及

深度图生成单元，用于根据提取到的所述深度特征，通过所述单目深度估计模块中的解码器生成所述预测深度图。

9.一种图像处理设备，包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现如权利要求1至5任一项所述方法的步骤。

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