[发明专利]一种基于测地距离的抠图处理方法、装置和图像处理设备

说明书

本申请提供了一种基于测地距离的抠图处理方法、装置和图像处理设备，该方法和装置应用于图像处理设备。本方案引入了通过反比例函数得到的置信度，该置信度与相应像素的两个测地距离的差值成负相关，因此，在相应像素的两个测地距离相近的情况时，通过该像素的置信度确定该像素的初始α值的可靠性并对其进行修正，从而使最终α值为概率图的概率α值和原图的初始α值的线性组合，使得最终α值更具有说服力，有效避免了像素被误判的发生，提高了抠图效果。

技术领域

本公开涉及图像处理技术，尤其涉及一种基于测地距离的抠图处理方法、装置和图像处理设备。

背景技术

在图像处理中，自然抠图是一项非常重要且具有挑战性的课题，其主要目的是将图像中的前景从背景中抠出，以便将前景与一个新的背景结合从而实现特定的效果。

现有技术是基于像素的测地距离进行α抠图，具体来说是求出每个像素与被选定的前景样本点和背景样本点之间的测地距离，进而根据这两个测地距离得到当前像素的α值，即当前像素的非透明度。最后根据每个像素的α值进行抠图处理，即将原图的内容中区分出前景和背景。

但是在前景与背景的衔接处会出现两个测地距离相近的情况，导致该处的像素有可能属于前景也有可能属于背景，在此情况下像素极有可能被误判，从而导致抠图效果较差。

发明内容

为克服相关技术中存在的问题，本公开提供一种基于测地距离的抠图处理方法、最终和图像处理设备。

根据本公开实施例的第一方面，提供一种基于测地距离的抠图处理方法，应用于图像处理设备，所述抠图处理方法包括步骤：

获取待处理的原图和所述原图的概率图；

计算所述原图中每个像素相对于前景样本点的前景测地距离，以及相对于背景样本点的背景测地距离；

根据每个像素的所述前景测地距离和所述背景测地距离计算相应像素的初始α值；

根据预设的反比例函数对所述前景测地距离和所述背景测地距离进行计算，得到每个像素的置信度；

利用所述置信度和所述概率图中每个像素对应的概率α值对所述初始α值进行修正计算，得到每个像素的最终α值；

根据每个像素的所述最终α值对所述原图进行抠图处理，得到所述原图的前景和背景。

可选的，所述概率图是利用预先训练的深度神经网络对所述原图进行处理得到的。

可选的，所述计算所述原图中每个像素相对于前景样本点的前景测地距离，以及相对于背景样本点的背景测地距离，包括：

利用迪杰斯特拉算法分别计算所述前景测地距离和所述背景测地距离。

可选的，所述根据每个像素的所述前景测地距离和所述背景测地距离计算相应像素的初始α值，包括：

根据如下公式计算每个像素的所述初始α值：

其中，α(x)为每个像素的初始α值，D_B为相应像素的背景测地距离，D_F为相应像素的前景测地距离。

可选的，所述反比例函数为：

其中，λ为置信度，e为欧拉数，D_F为背景测地距离，D_B为前景测地距离，β为预设常数。

可选的，所述最终α值的计算公式如下：

最终α值＝(1-λ)*初始α(x)+λ*概率α值。