APP下载

基于SLIC和多尺度显著性的红枣图像分割算法

2016-05-03曹洪武王振磊姚娜姚江河

江苏农业科学 2016年3期
关键词:图像分割显著性

曹洪武+王振磊+姚娜+姚江河

摘要: 针对自然光条件下具有复杂背景的红枣图像,提出了1种新的图像分割方法。首先,该方法用简单线性迭代聚类(simple linear iterative clustering,SLIC)算法对图像进行超像素分割,并进行迭代合并处理,从而得到不同尺度的图层;其次,利用图像局部对比度估计和区域位置估计来获取相应的显著性图;最后,用K-means算法得到红枣图像聚类分割结果。结果表明,该方法能有效去除图像的复杂背景,消除红枣上光斑的影响,准确地将红枣图像从背景中分割出来。

关键词: 图像分割;简单线性迭代聚类;多尺度;红枣图像;显著性;K-means

中图分类号: TP391.4;S126 文献标志码: A 文章编号:1002-1302(2016)03-0455-03

新疆地区有着丰富的红枣资源,红枣的种植和加工已成为新疆特色支柱产业之一,目前,新疆红枣产量已居全国首位。因此,进行红枣采摘机器人研究,对提高采摘效率、节约劳动力成本有特别的现实意义,而其中红枣图像的分割是研发采摘机器人的关键步骤。近年来,已有较多国内外学者从不同角度开展了图像分割研究,典型的方法有直方图阈值法[1]、特征空间聚类法[2]、区域分级方法[3]、边缘检测方法[4]、显著性图像分割方法[5]等。虽然上述方法在各自领域均有较好的应用,但仍未有一种通用方法能适合于所有类型的图像,各种新思路和新算法还在不断地出现。本研究针对复杂环境背景中的红枣图像提出了1种基于简单线性迭代聚类[6](simple linear iterative clustering,SLIC)和多尺度显著性图的分割算法进行红枣图像分割的算法,先利用SLIC算法构建超像素,迭代后获得不同尺度图层,然后利用显著性检测方法获得不同尺度下的红枣显著度图像,最后利用K-means算法对显著性图像进行聚类处理,得到最终分割结果。

1 不同尺度的图层生成

通过SLIC算法可将图像分割为包含若干均质区域的超像素图像,将超像素区域按从小到大的多个尺度顺序进行合并处理可得到不同尺度的图层,可以避免因一次性选择过大的区域尺度而产生边界分割错误。通过从较小尺度开始处理超像素,再逐步增大尺度的方式,可以保留图像主体结构,舍弃部分非关键性细节,消除这些细节对关键目标的影响。主要步骤为:首先利用SLIC算法对图像进行超像素分割预处理,得到新图像I;其次,判定I中各区域尺度,迭代合并生成不同尺度图层,先得到图层L1,并在L1基础上,同理获得图层L2。

1.1 SLIC算法原理

SLIC是1种基于像素色彩和色彩空间来进行超像素分割的方法[6],其算法流程如下。

(1)初始化过程。在 CIELAB色彩空间(CIE 1976 L*,a*,b* Color Space)中,依据超像素区域数目控制因子k,将图像划分为k个正方形网格,每个网格边长S=N/K,并将每个网格中心点3×3领域内的最小梯度位置初始化为聚类中心,如第j个聚类中心为:Cj[lj,aj,bj,xj,yj]T (lj为像素点光亮度,aj、bj为像素点色度坐标,xj、yj为像素空间坐标);并初始化每个像素的分类标记l(i)=-1,像素点距离d(i)=∞。

(2)对每个聚类中心Cj进行迭代处理。计算Cj与每个像素的距离D,计算范围为Cj的2S×2S邻接区域,当D式中:dc为像素点i与聚类中心点j之间的色彩距离;ds为像素点i与聚类中线点j之间的空间距离;m用来确定分割结果与图像实际边界贴合度,m值越小,贴合程度越高。

1.2 区域的迭代合并处理

为减少区域的数量,必须对超像素分割所得图像的各个区域进行合并处理,主要步骤如下。

(1)定义超像素分割所得任意区域Ri的尺度。如果存在1个t×t大小的正方形区域Rt×t,在满足Rt×tRi的条件下,使得Rt×t面积最大的t值作为区域Rt×t的尺度值。

(2)为了选择与Ri区域相邻的区域,使用以下公式计算在CIELUV色彩空间[CIE 1976 (L*,u*,v*) color space]中区域Ri、Rj的色彩均值距离:

Dluv(Ri,Rj)=(lj-li)2+(uj-ui)2+(vj-vi)2。

(3)进行区域迭代合并处理。通过计算与Ri距离最小且阈值大于t的相邻区域,使用1个内核为t×t的方框滤波器对图像进行滤波处理,如果该区域能够完全包含该滤波核,则颜色不作改变;否则,用滤波核范围内该区域的色彩均值来更新其中心点对应位置的色彩值,通过迭代,可实现图像中各区域的合并。

2 显著性图像的生成

生成红枣图像与背景反差尽可能大的显著性图,有利于后续进行聚类处理,本研究利用上述生成的不同尺度图像来获取对应的多个尺度的显著性图。显著性图像的生成受到图像中所包含像素的数量、色彩、关键区域所处位置及其大小等因素影响,通过综合考虑区域局部对比度估计值和图像位置估计值来计算图像显著度[7]。

(1)区域局部对比度估计。图像区域局部对比度可以通过计算待测区域Ri与其周边n个区域的色差加权和Ci来估计,Ri与其周边区域对比度越大,视觉上越显著,公式如下:

式中:ci、cj分别是Ri、Rj的颜色;ρ(Rj)为Rj中像素数量,区域中拥有的像素数量越多,对局部对比度权值的贡献将越大;φ(i,j)=e-D(Ri,Rj)/σ2表示Ri、Rj间的空间距离影响权值,与Ri区域距离越近,则受影响越大,其中σ2为检测区域的范围,D(Ri,Rj)为Ri、Rj中心点的欧氏距离平方。

(2)区域位置估计。图像中心点在人类视觉中较容易引起关注,则距离图像中心越近的区域,视觉上越具有显著性。因此,使用Hi来计算这类区域,计算区域Ri中的任意像素坐标xi与图像中心的坐标xc间的距离影响权值,λ为权重调节参数,ρ(Ri)为Ri中的像素数量,计算公式如下:

3 基于K-means算法的显著性图像聚类处理

为了减少上述显著性图中的区域数量,实现红枣和背景的分割,本研究使用K-means算法对图像进行聚类处理。K-means算法是一种无监督自适应的基于样本间相似性度量的间接聚类方法,其算法主要步骤[8]如下。

(1)在n个像素的数据集上随机选定k个初始聚类中心,设置迭代次数为R次。

(2)计算每个像素到各聚类中心的欧氏距离D(xi,μj)。按最近邻原则将其划分到以μj为聚类中心的子类Cj中,得到k个子类:C1,C2,…,Ck。

(3)设di为第i个子类中所有像素到其聚类中心μi的欧氏距离和,则每个像素到其聚类中心的欧氏距离和为:

4 算法主要步骤

综上,本研究提出的算法主要步骤归纳如下。

(1)不同尺度图层的生成。利用SLIC算法对红枣图像进行超像素分割预处理,得到图像I,判定图像I中各区域所属尺度,迭代合并以生成不同尺度图层。初始化区域尺度阈值为t,逐一判断I中每个区域Ri的尺度,若小于t,则计算该区域与其相邻区域间的CIELUV色彩空间均值距离,找出其中距离最小且阈值大于t的区域,将其与Ri合并,并更新合并后区域的尺度和颜色。对图中满足条件的所有区域进行迭代合并处理,得到新图层L1,继续增大t值,在L1基础上重复上述操作得到图层L2。

(2)显著性图像的生成。通过综合考虑区域局部对比度估计值和图像位置估计值的方法计算图像显著度,分别得到图层L1、L2所对应的显著性图像S1、S2。

(3)利用K-means算法对各层显著性图层进行聚类处理。分别得到S1、S2的聚类分割图像,实现红枣图像与背景的分割。

5 结果与分析

采用3 000×2 250分辨率的红枣图像,CPU为Intel i5-3 470 3.2 GHz,内存4.0 GB,操作系统为Windows 7,用Matlab 2013b编程实现红枣图像的分割。

图1为原始图像,设置SLIC超像素分割数量k=3 500。为了突出细节,图2为SLIC分割的局部放大结果,截取了从左至右方向上第3个红枣放大图,可见图中各超像素区域内部色彩均匀,分割边缘与红枣图像实际边缘紧密贴合。

选取尺度t的值为55、70,对图2结果进行迭代合并处理,分别得到2个不同尺度图像L1、L2,如图3、图4所示。分别检测其显著性,得到对应的显著性图像S1、S2,如图5、图6所示。S2中红枣图像与S1相比,对比度得到进一步提高,红枣边缘与其周边区域有清晰边界,所包含的区域数量大幅减少,有利于后续进行区域聚类处理。

为了分割出红枣和背景,设置K-means算法聚类数量 k=2,迭代次数R=-1,即重复迭代直至收敛,分别得到图像S1、S2的聚类结果,如图7、图8所示。在图7中,红枣图像分割结果边缘部分仍有少量缺失,原因在于所缺失的区域在迭代过程中仍然没有被合并到红枣目标区域中;图8分割结果中,红枣目标已经被完整分割出来,并且边缘部分保持完整。为了验证分割效果,使用Photoshop 14.0软件对红枣图像进行手工分割,取得基准分割图像,如图9所示。对比图8中聚类结果和图9手工分割结果,可见本研究算法比手工分割结果实际外形总体略微偏小,除红枣果实被遮挡区域、相连区域、果蒂和部分边缘细节与手工分割结果相比有误差外,红枣图像整体分割效果良好。

6 结论

通过利用SLIC算法进行超像素分割,设定不同区域阈值来获得多个不同尺度的图层,能够确保分割边缘与实际边缘紧密贴合,避免分割错误,确保边缘完整性。另外,依据不同尺度的显著性图进行K-means聚类,能有效去除图中干扰因素,将红枣与背景有效分割,确保分割效果。试验结果表明,在选择合适的区域尺度及分层数量基础上,该方法能够很好地消除红枣上光斑影响,准确地将边缘光滑的红枣图像与背景分割开,取得较好的分割效果。在今后的工作中,如何进一步提高图像的分割精度和提高处理效率将是研究重点。

参考文献:

[1]Lan J H,Zeng Y L. Multi-threshold image segmentation using maximum fuzzy entropy based on a new 2D histogram[J]. Optik - International Journal for Light and Electron Optics,2013,124(18):3756-3760.

[2]Jin R,Kou C,Liu R,et al. A color image segmentation method based on improved k-means clustering algorithm[C].Chongqing,China:Proceedings of the International Conference on Information Engineering and Applications,2013:499-505.

[3]张长青. 葛文英与刘国英.一种基于区域分级合并的彩色图像分割方法[J]. 计算机工程与应用,2012,48(17):203-206.

[4]王江涛,练 煜,石红岩. 结合模糊C均值聚类和边缘检测算法的彩色图像分割[J]. 兰州文理学院学报:自然科学版,2015,29(2):61-65.

[5]张 辰,杨文柱,刘召海. 基于HSV综合显著性的彩色图像分割方法[J]. 计算机工程与设计,2013,34(11):3944-3947.

[6]Achanta R,Shaji A,Smith K,et al. SLIC superpixels compared to state-of-the-art superpixel methods[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence,2012,34(11):2274-2282.

[7]Yan Q,Xu L,Shi J,et al. Hierarchical saliency detection[C]. Portland,Oregon:2013 IEEE Conference on Computer Vision and Pattern Recognition,2013:1155-1162.

[8]霍迎秋,秦仁波,邢彩燕,等. 基于CUDA的并行K-means聚类图像分割算法优化[J]. 农业机械学报,2014,45(11):47-53,74.

猜你喜欢

图像分割显著性
本刊对论文中有关统计学表达的要求
基于显著性权重融合的图像拼接算法
基于视觉显著性的视频差错掩盖算法
一种基于显著性边缘的运动模糊图像复原方法
论商标固有显著性的认定
纹理粗糙度在红外图像显著性检测中的应用