[发明专利]影像分割方法、存储介质及电子装置

说明书

一种影像分割方法，所述方法包括：获取影像和所述影像的深度图像；获取所述深度图像的直方图，所述直方图包括横坐标及纵坐标；采用聚类算法对直方图数据进行聚类，得到两个类别及两个类别的中心；确定直方图中两个类别的中心之间的对应纵坐标值最小的横坐标值为分割阈值；确定所述分割阈值是否满足预设条件；当所述分割阈值满足预设条件时，根据所述分割阈值将所述影像分割成前景和背景两个区域。本发明还提供一种计算机程序、存储介质及电子装置，可避免在影像深度分布连续的位置进行前背景分割。

技术领域

本发明涉及数字图像处理技术领域，特别涉及一种影像分割方法、计算机程序、存储介质及电子装置。

背景技术

目前，影像的后期处理，例如背景虚拟处理，需要事先将影像进行前背景分割处理。然而，现有的影像进行前背景分割处理往往存在在影像深度分布连续的位置进行分割导致影像的前景和背景分割不恰当，进而影响影像的后期处理。

发明内容

鉴于以上内容，有必要提供一种影像分割方法、计算机程序、存储介质及电子装置，可避免在影像深度分布连续的位置进行前背景分割。

本发明的第一方面提供一种影像分割方法，所述方法包括：

获取影像和所述影像的深度图像；

获取所述深度图像的直方图，所述直方图包括横坐标及纵坐标；

采用聚类算法对直方图数据进行聚类，得到两个类别及两个类别的中心；

确定直方图中两个类别的中心之间的对应纵坐标值最小的横坐标值为分割阈值；

确定所述分割阈值是否满足预设条件；

当所述分割阈值满足预设条件时，根据所述分割阈值将所述影像分割成前景和背景两个区域。

较佳的，所述获取所述影像的深度图像包括：

通过双目匹配算法或者深度图像采集装置获取所述影像的深度图像。

较佳的，所述预设条件为所述直方图中以所述分割阈值为中心的预设范围内的纵坐标值的平均值小于预设值。或者，所述预设条件为所述直方图中所述分割阈值对应的纵坐标值小于预设值。

较佳的，所述方法还包括：

当所述分割阈值不满足预设条件时，确定所述影像不适合分割成所述前景和所述背景两个区域。

较佳的，所述直方图中的所述横坐标代表所述深度图像的像素的深度值；

所述直方图中的所述纵坐标代表对应每一所述深度值的所述像素的个数。

较佳的，所述聚类算法为K-means算法。

本发明的第二方面提供一种计算机程序，所述计算机程序包括至少一个计算机程序指令，所述至少一个计算机程序指令用于被处理器执行以实现如上任意一项所述的影像分割方法。

本发明的第三方面提供一种存储介质，所述存储介质存储至少一个计算机程序指令，所述至少一个计算机程序指令用于被处理器执行以实现如上任意一项所述的影像分割方法。

本发明的第四方面提供一种电子装置，所述电子装置包括处理器及存储器，所述处理器用于执行所述存储器中存储的至少一个计算机程序指令以实现如上任意一项所述的影像分割方法。

本方案通过获取影像和所述影像的深度图像；获取所述深度图像的直方图，所述直方图包括横坐标及纵坐标；采用聚类算法对直方图数据进行聚类，得到两个类别及两个类别的中心；确定直方图中两个类别的中心之间的对应纵坐标值最小的横坐标值为分割阈值；确定所述分割阈值是否满足预设条件；当所述分割阈值满足预设条件时，根据所述分割阈值将所述影像分割成前景和背景两个区域，避免在影像深度分布连续的位置进行前背景分割。