计算机应用   2016, Vol. 36 Issue (9): 2576-2579  DOI: 10.11772/j.issn.1001-9081.2016.09.2576
0

引用本文 

王雷, 黄晨雪. 改进的分层马尔可夫随机场彩色图像分割算法[J]. 计算机应用, 2016, 36(9): 2576-2579.DOI: 10.11772/j.issn.1001-9081.2016.09.2576.
WANG Lei, HUANG Chenxue. Improved hierarchical Markov random field algorithm color image segmentation algorithm[J]. Journal of Computer Applications, 2016, 36(9): 2576-2579. DOI: 10.11772/j.issn.1001-9081.2016.09.2576.

基金项目

国家自然科学基金资助项目(51174084);湖北省自然科学基金资助项目(2014CFB583)

通信作者

王雷(1986-), 男, 湖北宜昌人, 讲师, 博士, 主要研究方向:图像分割、图像识别、模糊控制、机器学习、数据挖掘, releoy@foxmail.com

作者简介

黄晨雪(1993-), 女, 湖北荆州人, 硕士研究生, 主要研究方向:彩色图像分割、机器学习、数据挖掘

文章历史

收稿日期:2016-02-26
修回日期:2016-04-12
改进的分层马尔可夫随机场彩色图像分割算法
王雷, 黄晨雪    
湖北工业大学 湖北省农业机械工程研究设计院, 武汉 430068
摘要: 针对传统的分层马尔可夫随机场(MRF)算法难以描述彩色图像像素值分布等问题,提出一种基于RGB色彩统计分布的分层MRF分割算法。在分层MRF模型的基础上,设定了相关参数并对分割过程进行了公式推导;结合RGB色彩统计分布模型,重写了分层MRF能量函数,利用k-means算法作为预分割算法,实现了算法的无监督分割。相比传统的分层MRF分割模型,该算法充分利用了彩色图像的像素值的信息,可有效地减少颜色分布参数和计算成本,能更准确地描述各分割对象的颜色分布;且该算法不受目标和背景颜色区间分布、目标空间分布的限制,能够很好地描述不同目标和背景。通过大量实验验证了算法的有效性,其在运算速度、分割精度等方面均优于传统MRF算法和模糊C均值(FCM)算法。
关键词: 彩色图像    分层MRF    RGB色彩空间    图像分割    能量函数    
Improved hierarchical Markov random field algorithm color image segmentation algorithm
WANG Lei, HUANG Chenxue     
Hubei Agricultural Machinery Engineering Research and Design Institute, Hubei University of Technology, Wuhan Hubei 430068, China
Background: This work was partially supported by Natural Science Foundation of Hubei Province of China (Grant No. 2014CFB583). the National Natural Science Foundation of China (51174084)
WANG Lei, born in 1986, Ph. D., lecturer. His research interests include image segmentation, image recognition, fuzzy control, machine learning, data mining
HUANG Chenxue, born in 1993, M. S. candidate. Her research interests include color image segmentation, machine learning, data mining
Abstract: The distribution of color image pixel value is difficult to describe in hierarchical Markov Random Field (MRF) segmentation algorithm, therefore, a hierarchical MRF segmentation algorithm based on RGB color statistic distribution was proposed to solve this problem. The key parameters of the MRF model were set up, and the related formulas were deduced. With the RGB color statistic distribution model, the hierarchical MRF energy function was rewritten, and the k-means algorithm was used as presegmentation method to realize unsupervised segmentation. The proposed algorithm has fewer color distribution parameters and lower computational cost in comparison with traditional MRF segmentation model, which describes color distribution more accurately; and it can describe different targets and background very well without being restricted by target and background color distribution and target spatial distribution. Experimental results prove the effectiveness of the proposed algorithm, which is superior to the MRF algorithm and Fuzzy C-Means (FCM) algorithm in computing speed and segmentation accuracy.
Key words: color image    hierarchical Markov Random Field (MRF)    Red, Green and Blue color space    image segmentation    energy function    
0 引言

光学图像处理技术已经广泛应用于机器人、模式识别、机器视觉等领域中。Jabbari等[1]提出了一种基于图像的视觉伺服算法,用来控制无人飞行器的三维平动和旋翼偏航旋转; Hsiao等[2]对多视觉系统实现分布式移动机器人估计目标位置等方面进行了研究; 其他研究人员在对光学图像进行处理和识别算法研究中也取得了一些进展[3-4]

马尔可夫随机场(Markov Random Field, MRF)模型由于能够表达图像像素间的空间信息,并根据这些图像矩阵的像素值服从的概率分布来进行数学分割,因此在声呐图像等噪声干扰严重的图像分割中,最先获得广泛应用,而且在声呐图像分割中取得了较好的分割结果[5-6],但是只是在灰度的声呐图像上有较好的分割效果。Zhang等[7]提出一种自适应MRF模型参数的算法,对某一类灰度图像有较好的分割效果,但是对彩色图像进行分割时,需进行灰度化处理。褚一平等[8]对分层MRF模型在抗抖动视频分割上进行了研究,相对传统的灰度图像分割算法具有一定的分割速度和精度优势,但是难以满足图像精确分割的要求。而针对不同的应用环境对MRF模型进行改进的算法也层出不穷[9-10],只是针对彩色图像难以获得较高的分割精度。

彩色图像的分割任务是现阶段机器视觉中研究的重要课题与研究热点。杨华勇等[11]提出了一种结合多空间特征的多尺度马尔可夫随机场彩色图像分割算法;胡钦瑞等[12]提出了一种基于粗糙集和马尔可夫随机场的彩色图像分割方法,在基于粗糙集分割的基础上,采用颜色特征和纹理特征建立多特征的自适应可变权重马尔可夫随机场分割模型。上述算法均对MRF算法进行改进,对某些彩色图像分割获得了比较好的效果,但是不能实现对一个区域内颜色分布不均目标进行良好分割。其他研究人员在彩色图像分割算法研究中也取得了一些进展[13-15]。由于传统的MRF算法在对图像像素值分布模型进行描述时,往往采用单一的数学模型进行拟合,这种方法不但带来了大量的计算成本,而且难以用一个模型描述不同颜色分布、不同背景及不同的分割对象,适应性较差。本文中结合红绿蓝(Red Green and Blue, RGB)模型,采用一种基于统计的颜色分布描述方法来解决该问题。

1 分层MRF模型

分层MRF模型分为完全分层和不完全分层两种,它们之间的区别仅在于在顶层。不完全分层MRF模型的节点没有条件独立性,也就是说,每个节点不仅与父节点有关系,而且与相邻节点有关系[5];而完全分层模型每个节点仅与父节点有关系。为了获得良好的分割性能,考虑到求解计算困难度,我们选择了三次不完全分层MRF模型和二阶邻域。若有彩色图像F, 其图像大小为M×Nf为图像中s=(m, n)处的一个像素点,其像素值为(x, y, z), 将其分割为目标区域S1与背景区域S2。分割结果用M×N的标记图像G表示,g为标记图像中对应(m, n)处的标记结果。Fll层的观测图像,Gll层的标记图像,由于图像分为3层,l=(1, 2, 3)。其中F1为最顶层图像,F3=F,G3为最终分割结果。二阶MRF模型参数如图 1所示。

图 1 基团组合结构

图 1中{β1, β2, …, β8}为平面模型参数,利用β9来描述像素点和其父辈节点之间的关系;{α1, α2, …, α9}为后代2l×2l块中包含的基团数。利用最小二乘法估算模型参数[14]

$\begin{array}{l} \ln \frac{{P\left( {g_s^l,g_t^l\left| {g_s^l \in {S_1}} \right.} \right)}}{{P\left( {g_s^l,g_t^l\left| {g_s^l \in } \right.{S_2}} \right)}} = \sum\limits_{j = 1}^9 {\alpha _j^l\beta _j^l\left[{1 - g\delta \left( {g_s^l,g_t^l\left| {g_s^l \in {S_1}} \right.} \right)} \right]} - \\ \sum\limits_{j = 1}^9 {\alpha _j^l\beta _j^l\left[{1 - \delta \left( {g_s^l,g_t^l\left| {g_s^l \in {S_2}} \right.} \right)} \right]} \end{array}$ (1)

其中δ为Kronecker函数;βjl为第l层模型参数;αii为包含的后代节点的数,其服从以下关系:

$\begin{array}{l} \alpha _1^l = \alpha _2^l = \alpha _5^l = \alpha _6^l = \left[{{2^l} + 2\left( {{2^l} - 1} \right)} \right]\\ \alpha _3^l = \alpha _4^l = \alpha _7^l = \alpha _8^l = 1,\alpha _9^l = {4^l} \end{array}$

由于:

$\ln \frac{{P\left( {g_s^l,g_t^l\left| {g_s^l \in {S_1}} \right.} \right)}}{{\left( {g_s^l,g_t^l\left| {g_s^l \in {S_2}} \right.} \right)}} \approx \frac{{NUM\left\{ {s \in {S_1}:g = g_s^l,g_t^l} \right\}}}{{NUM\left\{ {s \in {S_2}:g = g_s^l,g_t^l} \right\}}}$ (2)

利用最小二乘法可以估算出模型参数,具体公式如下:

$\left\{ {{\beta _1},{\beta _2},\cdots ,{\beta _8}} \right\} = {\left( {{\boldsymbol{B}^{\rm{T}}} \cdot \boldsymbol{B}} \right)^{ - 1}} \cdot {\boldsymbol{B}^{\rm{T}}} \cdot \boldsymbol{A}$ (3)

其中:B为基团组合统计矩阵,A为对应包含的后代点统计向量[15]

2 RGB颜色分布

RGB颜色空间是计算机领域中应用最广泛的颜色系统之一。传统的MRF分割算法利用数学分布模型拟合图像中的像素值分布,在RGB色彩空间中一个像素点的像素值用三维坐标(x, y, z)表示,但用数学模型模拟像素值分布针对不同环境、分割不同对象时,很难表达,而且解算过程会带来比较高的计算成本。本文提出基于统计的RGB分布描述模型,根据图像中像素点在R、G、B通道分布密度和范围设定统计范围, 能够描述任意颜色区间的颜色分布。假设R通道划分U个色彩段,G通道划分V个色彩段,B通道划分W个色彩段,f处的像素值(x, y, z)中,xu, yv, zw,其中u, v, w为R、G、B通道中的一个色彩段,则的隶属于哪个一个区域的概率公式如下:

${P_{{S_1}}}\left( {f\left| g \right.} \right) = \frac{{Sum\left( {x \in u,y \in v,z \in w\left| {s \in {S_1}} \right.} \right)}}{{Sum\left( {s\left| {s \in {S_1}} \right.} \right)}}$ (4)
${P_{{S_2}}}\left( {f\left| g \right.} \right) = \frac{{Sum\left( {x \in u,y \in v,z \in w\left| {s \in {S_1}} \right.} \right)}}{{Sum\left( {s\left| {s \in {S_1}} \right.} \right)}}$ (5)

由此能量函数可以写成:

$\begin{array}{l} U_{{s_1}}^l\left( {g\left| f \right.} \right) = - \sum {\ln \frac{{Sum\left( {x \in u,y \in v,z \in w\left| {s \in {S_1}} \right.} \right)}}{{Sum\left( {s\left| {s \in {S_1}} \right.} \right)}}} \\ \sum {\alpha _{s,t}^l\beta _{s,t}^l\left[{1 - \delta \left( {g_s^l,g_t^l} \right)} \right] + \sum {\alpha _9^l{\beta _9}\left[{1 - \delta \left( {g_s^l,g_{parent\left( s \right)}^{l + 1}} \right)} \right]} } \end{array}$ (6)
$\begin{array}{l} U_{{s_2}}^l\left( {g\left| f \right.} \right) = - \sum {\ln \frac{{Sum\left( {x \in u,y \in v,z \in w\left| {s \in {S_2}} \right.} \right)}}{{Sum\left( {s\left| {s \in {S_2}} \right.} \right)}}} \\ \sum {\alpha _{s,t}^l\beta _{s,t}^l\left[{1 - \delta \left( {g_s^l,g_t^l} \right)} \right] + \sum {\alpha _9^l{\beta _9}\left[{1 - \delta \left( {g_s^l,g_{parent\left( s \right)}^{l + 1}} \right)} \right]} } \end{array}$ (7)

其中ts的邻域点。

基于RGB颜色分布的分层MRF彩色图像分割算法的具体步骤为:

1) 对最顶层图像F1利用k-means算法初始分割,将图像分割为两类。采用最小二乘法,根据式(1)、(2)、(3)利用预分割结果估算模型参{β11, β21, …, β81},对每一个节点,其颜色分布概率为所有子节点概率均值${P_s}\left( f \right) = \frac{1}{{16}}\sum {{P_{{\rm{child}}}}} $,利用式(4)、(5)得到颜色分布。利用式(6)、(7)通过后验能量函数最小重新得到最顶层的分割结果,采用条件迭代模式(Iterated Conditional Mode, ICM)进行迭代,直到得到最顶层最终结果。

2) 利用最顶层分割结果和中间层与最顶层的映射关系,获得中间层初始分割结果,利用最小二乘法得到初始模型参数{β12, β22, …, β82, β92},子节点概率均值${P_s}\left( f \right) = \frac{1}{4}\sum {{P_{{\rm{child}}}}} $,通过ICM过程,得到中间层分割结果。

3) 利用中间层分割结果和中间层与最底层的映射关系,获得最底层初始分割结果,利用最小二乘法得到初始模型参数{β13, β23, …, β83, β93},通过ICM过程,得到最终结果。

本文所有实验图片均来自Berkeley图库,综合考虑运算速度和算法的准确度,对RGB三个颜色通道均采用间隔16的均匀划分,图 2为采用本算法对图库中12003号图像进行分割的实验结果。

图 2 RGB分布模型分割实验

图 2(a)为原图像。图 2(b)为图像库中1105号人工分割标准图像。

图 2(c)为对最顶层图像利用k-means算法进行预分割的结果,分割出的区域为目标区,留白区域为背景区。图 2(d)为顶层图像分割结果,其分辨率为120×80,共迭代8次, 耗时0.15 s。图 2(e)为迭代6次后得到的中间层分割结果, 其分辨率为240×160,耗时0.33 s。(d)、(e)两幅图像均为分割过程图像,图 2(f)为迭代11次后得到的最终分割结果, 耗时1.52 s。

为了说明分割算法的有效性,本文利用Dice系数[14]进行评价,分割精度系数越大,说明分割精度越高。图 2的Dice系数为0.965,拥有比较好的分割结果精度。采用kappa系数衡量分割结果与实际分割类型相一致的概率,0.0~0.20极低的一致性、0.21~0.40表示一般的一致性、0.41~0.60表示中等的一致性、0.61~0.80表示高度的一致性和0.81~1表示几乎完全一致[11]表 1为各层最终迭代后得到的模型参数。图 3为最底层目标区颜色统计分布。

表 1 图 2分割过程中各层最终MRF模型参数
图 3 目标颜色分布
3 实验结果和对比

将本文算法的分割结果与模糊C均值(Fuzzy C-Means, FCM)算法和传统的MRF算法分割结果进行比较。实验环境为Windows 7,i5四核CPU主频3.3GHz计算机,采用C#和Emgu CV图像处理库进行编程。为了说明算法的有效性,分别对不同类型、不同对象的彩色图像进行分割对比。三种算法与本文算法均采用相同的k-means预分割结果进行初始化。由于传统MRF算法无法直接对彩色图像进行分割,所以将其转化为灰度图像进行处理。采用比例分布模型描述目标区,高斯分布描述背景区。

图 4图 6图 7中图像分辨率为481×321,图 5的分辨率为321×481。从表 2中采用的Dice系数描述分割精度可以看出,本文算法比传统的聚类算法精度要高,在所有应用环境下均远远高于FCM的分割效果。

图 4 动物图像分割实验
图 6 植物图像分割实验
图 7 纹理图像分割实验
图 5 人物图像分割实验
表 2 分割精度及一致性指标定量对比

由于传统的MRF算法采用单一模型描述目标区或背景区,在面对不同对象和背景图像分割时,分割效果较差,且耗时较长。从表 2中采用kappa系数衡量分割结果与实际分割类型相一致的概率比较中能看出,本文算法均获得了几乎完全一致的值,远远优于比较算法;尤其在目标区域包含颜色反差较大、纹理复杂的情况中,往往能获得较好的分割效果。从表 2中对分割耗时进行的定量对比可以看出,本文的分割算法在运算速度上也具备一定优势。

4 结语

在彩色图像分割中,针对目标包含颜色反差较大、纹理复杂等难以分割的情况,传统的分割算法难以获得较好的分割结果。基于统计的RGB色彩空间模型不受目标和背景颜色区间分布的限制,能够很好地描述这种目标。改进算法重写了分层MRF算法的能量函数,使得该算法在分割准确性、适应性及快速性上均有提高。实验结果表明,针对不同特征的彩色图像,本文算法在较小的时间代价下获得了理想的分割结果。在后续研究中,可以对其他色彩空间、初始化算法及其他方面进行改进,以进一步提高分割效果。

参考文献
[1] JABBARI A H, ORIOLO G, BOLANDI H. An adaptive scheme for image-based visual servoing of an underactuated UAV[J]. International Journal of Robotics and Automation, 2014, 29 (1) : 92-104. (0)
[2] DU B, ZHANG L. Random-selection-based anomaly detector for hyperspectral imagery[J]. IEEE Transactions on Geoscience and Remote Sensing, 2011, 49 (5) : 1578-1589. doi: 10.1109/TGRS.2010.2081677 (0)
[3] HSIAO J P, YEH S S, HSU P L. arget position estimation using multi-vision system implemented on distributed mobile robots[J]. International Journal of Robotics and Automation, 2013, 28 (2) : 154-169. (0)
[4] SHI Q, DU B, ZHANG L. Spatial coherence based batch-mode active learning for remote sensing images classification[J]. IEEE Transactions on Image Processing, 2015, 24 (7) : 2037-2050. doi: 10.1109/TIP.2015.2405335 (0)
[5] GAETANO R, POGGI G, SCARPA G. Hierarchical MRF-based segmentation of remote-sensing images [C]// ICIP 2006: Proceedings of the 2006 International Conference on Image Processing. Piscataway, NJ: IEEE, 2006: 1121-1124. (0)
[6] MIGNOTTE M, COLLET C, PEREZ P. Sonar image segmentation using an unsupervised hierarchical MRF model[J]. IEEE Transactions on Image Processing, 2000, 9 (7) : 1216-1231. doi: 10.1109/83.847834 (0)
[7] ZHANG X L, LAM K M, SHEN L. Image magnification based on adaptive MRF model parameter estimation [C]// ISPCS 2005: Proceedings of the 2005 International Symposium on Intelligent Signal Processing and Communication Systems. Piscataway, NJ: IEEE, 2005: 653-656. (0)
[8] 褚一平, 叶修梓, 张引, 等. 基于分层MRF模型的抗抖动视频分割算法[J]. 浙江大学学报(工学版), 2007, 41 (11) : 1793-1796. ( CHU Y P, YE X Z, ZHANG Y, et al. Vibration-resisted video segmentation algorithm based on hierarchical Markov random field model[J]. Journal of Zhejiang University Engineering Science), 2007, 41 (11) : 1793-1796. ) (0)
[9] 郑晨, 王雷光, 胡亦钧, 等. 利用小波域多尺度模糊MRF模型进行纹理分割[J]. 武汉大学学报(信息科学版), 2010, 35 (9) : 1074-1078. ( ZHENG C, WANG L G, HU Y J, et al. Texture segmentation based on multiscale fuzzy Markov random field model in wavelet domain[J]. Geomatics and Information Science of Wuhan University, 2010, 35 (9) : 1074-1078. ) (0)
[10] 王兴梅.水下声纳图像的MRF目标检测与水平集的轮廓提取方法研究[D].哈尔滨:哈尔滨工程大学, 2010:46-48. ( WANG X M. Research on underwater sonar image objects detection and contours extraction based respectively on MRF and level set [D]. Harbin: Harbin Engineering University, 2010: 46-48. ) (0)
[11] 杨华勇, 余正红, 郑晨. 结合多空间特征的多尺度马尔可夫随机场彩色图像分割[J]. 计算机应用, 2011, 31 (12) : 3378-3381. ( YANG H Y, YU Z H, ZHENG C. Color image segmentation of multi-resolutin Markov random field in combination with multi-space characteristics[J]. Journal of Computer Applications, 2011, 31 (12) : 3378-3381. ) (0)
[12] 胡钦瑞, 肖国强. 基于粗糙集和MRF的彩色图像分割方法[J]. 西南师范大学学报(自然科学版), 2014, 39 (4) : 113-118. ( HU Q R, XIAO G Q. On color image segmentation based on rough set and MRF[J]. Journal of Southwest China Normal University Natural Science Edition), 2014, 39 (4) : 113-118. ) (0)
[13] 赵瑶池, 胡祝华, 白勇, 等. 基于纹理差异度引导的DRLSE病虫害图像精准分割方法[J]. 农业机械学报, 2015, 46 (2) : 14-19. ( ZHAO Y C, HU Z H, BAI Y, et al. An accurate segmentation approach for disease and pest based on texture difference guided DRLSE[J]. Transaction of the Chinese Society for Agricultural Machinery, 2015, 46 (2) : 14-19. ) (0)
[14] VANTARAM S R, SABER E. Survey of contemporary trends in color image segmentation[J]. Journal of Electronic Imaging, 2012, 21 (4) : 177-187. (0)
[15] 魏小莉, 沈未名. 一种基于马尔可夫随机场模型的彩色纹理图像分割[J]. 武汉大学学报(信息科学版), 2014, 38 (8) : 955-958. ( WEI X L, SHEN W M. Gabor-MRF model based on color texture image segmentation[J]. Geomatics and Information Science of Wuhan University, 2014, 38 (8) : 955-958. ) (0)