[发明专利]一种图像抠图方法及装置

说明书

技术领域

本发明涉及计算机视觉技术领域，特别涉及一种图像抠图方法及装置。

背景技术

图像抠图技术是一种利用算法对图像中的前景目标进行提取的技术。抠图技术在影视制作，图像合成中具有非常广泛的应用。

该技术通常需要用户给予一定的交互，比如用一个矩形框粗略标记出物体所在位置；或者在前景，背景上勾勒出一定得形状。通过交互，就可以给出粗略的前景，背景信息，然后放入算法中，就可以得到相对较为精细的前景提取结果。目前，图像抠图技术在算法实时性，鲁棒性，正确性上存在许多可改进的空间。特别是在对颜色以及光照变化方面精度较低。因此如何提高抠图在颜色，光照变化方面的鲁棒性和精确性，是本领域技术人员需要解决的技术问题。

发明内容

本发明的目的是提供一种图像抠图方法、装置，对颜色，光照变化具有更好的鲁棒性，且精度较高，只需少量用户交互即可得到较好的抠图结果。

为解决上述技术问题，本发明提供一种图像抠图方法，包括：

获取输入的图像以及目标框，并构建目标框对应图像的深度信息；

利用高斯混合模型对RGBD建模形成的图割算法对所述目标框对应图像进行粗分割，获取目标区域；其中，D为所述深度信息；

对所述目标区域进行腐蚀膨胀操作，获得三元图；

确定所述三元图的透明度值，获得前景精确提取结果。

可选的，利用高斯混合模型对RGBD建模形成的图割算法对所述目标框对应图像进行粗分割，获取目标区域，包括：

构建所述目标框对应图像的S-T图；

利用高斯混合模型对所述S-T图中每一个像素点的RGBD进行建模得到对应的RGBD值，并根据所述S-T图中每一条边对应的两个像素点的RGBD值的差值赋予所述S-T图中每一条边的权重；

根据所述S-T图中每一条边的权重，利用最大流算法进行分割，并对分割后的图像进行轮廓检测，确定面积最大轮廓；

对所述面积最大轮廓进行填充获得目标区域。

可选的，对所述目标区域进行腐蚀膨胀操作，获得三元图，包括：

对所述目标区域，利用所述目标框的宽的5％以及高的5％的像素参数，进行腐蚀膨胀操作，获得三元图。

可选的，确定所述三元图的透明度值，包括：

利用共享样本点算法计算所述三元图的透明度值。

可选的，本方案还包括：

根据输入的修正框，迭代执行所述利用高斯混合模型对RGBD建模形成的图割算法对所述目标框对应图像进行粗分割，获取目标区域的步骤。

可选的，获取输入的图像，包括：

获取双摄像头图像采集装置采集的图像。

本发明还提供一种图像抠图装置，包括：

图像获取模块，用于获取输入的图像以及目标框；

深度信息构建模块，用于构建目标框对应图像的深度信息；

图像粗分割模块，用于利用高斯混合模型对RGBD建模形成的图割算法对所述目标框对应图像进行粗分割，获取目标区域；其中，D为所述深度信息；

三元图获取模块，用于对所述目标区域进行腐蚀膨胀操作，获得三元图；

图像精确提取模块，用于确定所述三元图的透明度值，获得前景精确提取结果。

可选的，所述深度信息构建模块，包括：

S-T图构件单元，用于构建所述目标框对应图像的S-T图；

权重赋值单元，用于利用高斯混合模型对所述S-T图中每一个像素点的RGBD进行建模得到对应的RGBD值，并根据所述S-T图中每一条边对应的两个像素点的RGBD值的差值赋予所述S-T图中每一条边的权重；

轮廓确定单元，用于根据所述S-T图中每一条边的权重，利用最大流算法进行分割，并对分割后的图像进行轮廓检测，确定面积最大轮廓；

目标区域确定单元，用于对所述面积最大轮廓进行填充获得目标区域。

可选的，所述图像精确提取模块，包括：

透明度值计算单元，用于利用共享样本点算法计算所述三元图的透明度值。

可选的，本方案还包括：

迭代模块，用于根据输入的修正框，迭代执行所述图像粗分割模块。

本发明所提供的一种图像抠图方法，包括：获取输入的图像以及目标框，并构建目标框对应图像的深度信息；利用高斯混合模型对RGBD建模形成的图割算法对目标框对应图像进行粗分割，获取目标区域；其中，D为深度信息；对目标区域进行腐蚀膨胀操作，获得三元图；确定三元图的透明度值，获得前景精确提取结果；