[发明专利]一种前景图像的提取方法、设备及计算机可读存储介质

说明书

本发明公开了一种前景图像的提取方法、设备及计算机可读存储介质，其中，该方法包括：通过获取输入的红外图像和色彩图像，并对所述红外图像和所述色彩图像进行特征点匹配，得到所述色彩图像到所述红外图像的变换矩阵；获取所述色彩图像中的前景掩模图像，并根据所述变换矩阵对所述前景掩模图像进行投影变换；获取输入的深度图像，并将投影变换后的所述前景掩模图像与所述深度图像进行对齐，计算得到对齐后的梯度图像；根据所述梯度图像的深度突变对所述前景掩模图像进行修正。实现了一种高效的前景图像提取方案，极大地提高了前景图像的提取精度，增强了用户对于景深拍摄功能的使用体验。

技术领域

本发明涉及移动通信领域，尤其涉及一种前景图像的提取方法、设备及计算机可读存储介质。

背景技术

现有技术中，随着智能终端设备的拍摄技术不断发展，用户对于景深拍摄的功能需求也越来越高。基于现有的景深拍摄原理，在所推出的各类景深拍摄的衍生功能中，待解决的难点在于对前景图像的提取，而如何提高景深拍摄过程中，对于前景图像提取的精准度，成为目前亟待解决的技术问题。

发明内容

为了解决现有技术中的上述技术缺陷，本发明提出了一种前景图像的提取方法，该方法包括：

获取输入的红外图像和色彩图像，并对所述红外图像和所述色彩图像进行特征点匹配，得到所述色彩图像到所述红外图像的变换矩阵。

获取所述色彩图像中的前景掩模图像，并根据所述变换矩阵对所述前景掩模图像进行投影变换。

获取输入的深度图像，并将投影变换后的所述前景掩模图像与所述深度图像进行对齐，计算得到对齐后的梯度图像。

根据所述梯度图像的深度突变对所述前景掩模图像进行修正。

可选地，所述获取输入的红外图像和色彩图像，并对所述红外图像和所述色彩图像进行特征点匹配，得到所述色彩图像到所述红外图像的变换矩阵，包括：

通过预设的深度摄像头获取当前取景的红外图像，以及通过预设的彩色镜头获取所述当前取景的色彩图像。

通过快速特征点提取和描述的算法，对所述红外图像与所述彩色图像的特征点进行匹配，计算得到所述特征点的所述变换矩阵。

可选地，所述获取所述色彩图像中的前景掩模图像，并根据所述变换矩阵对所述前景掩模图像进行投影变换，包括：

通过深度学习获取所述色彩图像中的前景轮廓信息。

通过所述变换矩阵将所述前景轮廓信息中的平面目标点投影至所述红外图像，得到所述平面目标点在所述红外图像中的目标点坐标。

可选地，所述获取输入的深度图像，并将投影变换后的所述前景掩模图像与所述深度图像进行对齐，计算得到对齐后的梯度图像，包括：

将所述深度图像设为A，以及，将所述梯度图像设为D。

根据所述深度图像计算得到所述梯度图像：

其中，(i，j)是所述目标点坐标，梯度图像D(i，j)是所述深度图像的像素在(i，j)和(i-1，j-1)的差值。

可选地，所述根据所述梯度图像的深度突变对所述前景掩模图像进行修正，包括：

预设与所述深度突变对应的阈值T，其中，所述阈值T表示为D(k1，k2)。

在所述梯度图像D(i，j)所述阈值T时，确定所述深度图像A在(i，j)的位置存在所述深度突变。

可选地，所述根据所述梯度图像的深度突变对所述前景掩模图像进行修正，还包括：

根据所述深度突变确定当前前景的边缘信息，并将所述边缘信息保存至坐标列表。