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

权利要求书

1.一种基于测地距离的抠图处理方法，应用于图像处理设备，其特征在于，所述抠图处理方法包括步骤：

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

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

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

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

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

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

所述反比例函数为：

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

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

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

2.如权利要求1所述的抠图处理方法，其特征在于，所述概率图是利用预先训练的深度神经网络对所述原图进行处理得到的。

3.如权利要求1所述的抠图处理方法，其特征在于，所述计算所述原图中每个像素相对于前景样本点的前景测地距离，以及相对于背景样本点的背景测地距离，包括：

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

4.如权利要求1所述的抠图处理方法，其特征在于，所述根据每个像素的所述前景测地距离和所述背景测地距离计算相应像素的初始α值，包括：

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

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

5.一种基于测地距离的抠图处理装置，应用于图像处理设备，其特征在于，所述抠图处理装置包括：

图像获取模块，被配置为获取待处理的原图和所述原图的概率图；

第一计算模块，被配置为计算所述原图中每个像素相对于前景样本点的前景测地距离，以及相对于背景样本点的背景测地距离；

第二计算模块，被配置为根据每个像素的所述前景测地距离和所述背景测地距离计算相应像素的初始α值；

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

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

抠图执行模块，被配置为根据每个像素的所述最终α值对所述原图进行抠图处理，得到所述原图的前景和背景；

所述反比例函数为：

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

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

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

6.如权利要求5所述的抠图处理装置，其特征在于，所述概率图是利用预先训练的深度神经网络对所述原图进行处理得到的。

7.如权利要求5所述的抠图处理装置，其特征在于，所述第一计算模块被配置为利用迪杰斯特拉算法分别计算所述前景测地距离和所述背景测地距离。

8.如权利要求5所述的抠图处理装置，其特征在于，所述第二计算模块被配置为根据如下公式计算每个像素的所述初始α值：

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

9.一种图像处理设备，其特征在于，设置有如权利要求5～8任一项所述的抠图处理装置。

10.一种图像处理设备，其特征在于，包括：

处理器；

用于存储所述处理器可执行指令的存储器；

其中，所述处理器被配置为：

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

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

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

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

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

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

所述反比例函数为：

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

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

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