[发明专利]生成深度图的方法和装置

摘要

本发明公开了一种生成深度图的方法和装置，属于计算机视觉技术领域。该方法包括：将输入图像对中的左右视图分别作为参考图像和目标图像做灰度变换；计算灰度变换后的参考图像的上下左右梯度矩阵；按视差计算代价函数矩阵并进行边界处理；计算原始信息存储矩阵且初始化上下左右信息存储矩阵；在GPU显存中利用CUDA置信传播算法进行迭代计算，得到迭代后的上下左右信息存储矩阵，根据迭代结果计算置信矩阵；根据置信矩阵计算深度图。该装置包括：灰度变换模块、梯度计算模块、代价函数矩阵计算模块、原始信息存储矩阵计算模块、迭代模块、置信矩阵计算模块和深度图计算模块。本发明提高了深度图的运算效率，实现了高质量且快速的深度图生成。

主权项

1、一种生成深度图的方法，其特征在于，所述方法包括：将输入图像对中的左视图作为参考图像，将所述输入图像对中的右视图作为目标图像，对所述参考图像和目标图像分别做灰度变换；计算灰度变换后的参考图像的上梯度矩阵、下梯度矩阵、左梯度矩阵和右梯度矩阵；根据灰度变换后的参考图像、灰度变换后的目标图像以及所述参考图像和目标图像之间的视差，计算代价函数矩阵并进行边界处理；根据所述代价函数矩阵计算得到原始信息存储矩阵，初始化上信息存储矩阵、下信息存储矩阵、左信息存储矩阵和右信息存储矩阵；将所述上梯度矩阵、下梯度矩阵、左梯度矩阵和右梯度矩阵，以及所述原始信息存储矩阵、上信息存储矩阵、下信息存储矩阵、左信息存储矩阵和右信息存储矩阵复制到图形处理器GPU的显存中，采用计算统一设备架构CUDA在所述GPU上采用置信传播算法进行迭代计算，分别得到迭代后的上信息存储矩阵、迭代后的下信息存储矩阵、迭代后的左信息存储矩阵和迭代后的右信息存储矩阵；根据所述迭代后的上信息存储矩阵、迭代后的下信息存储矩阵、迭代后的左信息存储矩阵和迭代后的右信息存储矩阵计算得到置信矩阵；根据所述置信矩阵计算得到深度图。