王雷 黄晨雪
摘要:
针对传统的分层马尔可夫随机场(MRF)算法难以描述彩色图像像素值分布等问题,提出一种基于RGB色彩统计分布的分层MRF分割算法。在分层MRF模型的基础上,设定了相关参数并对分割过程进行了公式推导;结合RGB色彩统计分布模型,重写了分层MRF能量函数,利用kmeans算法作为预分割算法,实现了算法的无监督分割。相比传统的分层MRF分割模型,该算法充分利用了彩色图像的像素值的信息,可有效地减少颜色分布参数和计算成本,能更准确地描述各分割对象的颜色分布;且该算法不受目标和背景颜色区间分布、目标空间分布的限制,能够很好地描述不同目标和背景。通过大量实验验证了算法的有效性,其在运算速度、分割精度等方面均优于传统MRF算法和模糊C均值(FCM)算法。
关键词:
彩色图像;分层MRF; RGB色彩空间;图像分割;能量函数
中图分类号:
TN919.8
文献标志码:A
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 kmeans 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 CMeans (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×N, f为图像中s=(m,n)处的一个像素点,其像素值为(x,y,z),将其分割为目标区域S1与背景区域S2。分割结果用M×N的标记图像G表示,g为标记图像中对应(m,n)处的标记结果。Fl为l层的观测图像,Gl为l层的标记图像,由于图像分为3层,l=(1,2,3)。其中F1为最顶层图像,F3=F,G3为最终分割结果。二阶MRF模型参数如图1所示。
2RGB颜色分布
RGB颜色空间是计算机领域中应用最广泛的颜色系统之一。传统的MRF分割算法利用数学分布模型拟合图像中的像素值分布,在RGB色彩空间中一个像素点的像素值用三维坐标(x,y,z)表示,但用数学模型模拟像素值分布针对不同环境、分割不同对象时,很难表达,而且解算过程會带来比较高的计算成本。本文提出基于统计的RGB分布描述模型,根据图像中像素点在R、G、B通道分布密度和范围设定统计范围,能够描述任意颜色区间的颜色分布。假设R通道划分U个色彩段,G通道划分V个色彩段,B通道划分W个色彩段, f处的像素值(x,y,z)中,x∈u,y∈v,z∈w,其中u,v,w为R、G、B通道中的一个色彩段,则的隶属于哪个一个区域的概率公式如下:
基于RGB颜色分布的分层MRF彩色图像分割算法的具体步骤为:
1)对最顶层图像F1利用kmeans算法初始分割,将图像分割为两类。采用最小二乘法,根据式(1)、(2)、(3)利用预分割结果估算模型参{β11, β12,…, β18},对每一个节点,其颜色分布概率为所有子节点概率均值Ps(f)=116∑Pchild,利用式(4)、(5)得到颜色分布。利用式(6)、(7)通过后验能量函数最小重新得到最顶层的分割结果,采用条件迭代模式(Iterated Conditional Mode, ICM)进行迭代,直到得到最顶层最终结果。
2)利用最顶层分割结果和中间层与最顶层的映射关系,获得中间层初始分割结果,利用最小二乘法得到初始模型参数{β21,β22,…,β28,β29},子节点概率均值Ps(f)=14∑Pchild,通过ICM过程,得到中间层分割结果。
3)利用中间层分割结果和中间层与最底层的映射关系,获得最底层初始分割结果,利用最小二乘法得到初始模型参数{β31,β32,…,β38,β39},通过ICM过程,得到最终结果。
本文所有实验图片均来自Berkeley图库,综合考虑运算速度和算法的准确度,图2为采用本文算法对12003号图像进行分割的实验结果,其中RGB对三个颜色通道均采间隔16均匀划分。综合考虑运算速度和算法的准确度,对RGB三个颜色通道均采用间隔16的均匀划分,图2为采用本算法对图库中12003号图像进行分割的实验结果。
为了说明分割算法的有效性,本文利用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为最底层目标区颜色统计分布。
3实验结果和对比
将本文算法的分割结果与模糊C均值(Fuzzy CMeans, FCM)算法和传统的MRF算法分割结果进行比较。实验环境为Windows 7,i5四核CPU主频3.3GHz计算机,采用C#和Emgu CV图像处理库进行编程。为了说明算法的有效性,分别对不同类型、不同对象的彩色图像进行分割对比。三种算法与本文算法均采用相同的
kmeans预分割结果进行初始化。由于传统MRF算法无法直接对彩色图像进行分割,所以将其转化为灰度图像进行处理。采用比例分布模型描述目标区,高斯分布描述背景区。
图4、图6、图7中图像分辨率为481×321,图5的分辨率为321×481。其中图4采用1107号标准分割进行对比,图5~7采用1103标准分割进行对比。从表2中采用的Dice系数描述分割精度可以看出,不同图像的两种分割算法的定量对比能看出,本文算法比传统的聚类算法精度要高,在所有应用环境下均远远高于FCM的分割效果。
由于传统的MRF算法采用单一模型描述目标区或背景区,在面对不同对象和背景图像分割时,分割效果较差,且耗时较长。从表2中采用kappa系数衡量分割结果与实际分割类型相一致的概率比较中能看出,本文算法均获得了几乎完全一致的值,远远优于比较算法;尤其在目标区域包含颜色反差较大、纹理复杂的情况中,往往能获得较好的分割效果。从表2中对分割耗时进行的定量对比可以看出,本文的分割算法在运算速度上也具备一定优势。
4结语
在彩色图像分割中,针对目标包含颜色反差较大、纹理复杂等难以分割的情况,传统的分割算法难以获得较好的分割结果。基于统计的RGB色彩空间模型不受目标和背景颜色区间分布的限制,能够很好地描述这种目标。改进算法重写了分层MRF算法的能量函数,使得该算法在分割准确性、适应性及快速性上均有提高。通过该算法和其他算法进行比较,实验结果表明,针对不同特征的彩色图像,本文算法在较小的时间代价下获得了理想的分割结果。在后续研究中,可以对其他色彩空间、初始化算法及其他方面进行改进,以进一步提高分割效果。
参考文献:
[1]
JABBARI A H, ORIOLO G, BOLANDI H. An adaptive scheme for imagebased visual servoing of an underactuated UAV [J]. International Journal of Robotics and Automation, 2014, 29(1): 92-104.
[2]
DU B, ZHANG L. Randomselectionbased anomaly detector for hyperspectral imagery [J]. IEEE Transactions on Geoscience and Remote Sensing, 2011, 49(5): 1578-1589.
[3]
HSIAO J P, YEH S S, HSU P L, Target position estimation using multivision system implemented on distributed mobile robots [J]. International Journal of Robotics and Automation, 2013, 28(2): 154-169.
[4]
SHI Q, DU B, ZHANG L, Spatial coherence based batchmode active learning for remote sensing images classification [J]. IEEE Transactions on Image Processing, 2015, 24(7):2037-2050.
[5]
GAETANO R, POGGI G, SCARPA G. Hierarchical MRFbased segmentation of remotesensing images [C]// ICIP 2006: Proceedings of the 2006 International Conference on Image Processing. Piscataway, NJ: IEEE, 2006: 1121-1124.
[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.
[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.
[8]
褚一平,葉修梓,张引,等.基于分层MRF模型的抗抖动视频分割算法[J].浙江大学学报(工学版),2007,41(11):1793-1796.(CHU Y P, YE X Z, ZHANG Y, et al. Vibrationresisted video segmentation algorithm based on hierarchical Markov random field model[J]. Journal of Zhejiang University (Engineering Science), 2007, 41(11): 1793-1796.)
[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.)
[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.)
[11]
杨华勇,余正红,郑晨.结合多空间特征的多尺度马尔可夫随机场彩色图像分割[J].计算机应用,2011,31(12):3378-3381.(YANG H Y, YU Z H, ZHENG C. Color image segmentation of multiresolutin Markov random field in combination with multispace characteristics[J]. Journal of Computer Applications, 2011, 31(12):3378-3381.)
[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.)
[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.)
[14]
VANTARAM S R, SABER E. Survey of contemporary trends in color image segmentation [J]. Journal of Electronic Imaging, 2012, 21(4):177-187.
[15]
魏小莉,沈未名.一种基于马尔可夫随机场模型的彩色纹理图像分割[J].武汉大学学报(信息科学版),2014,38(8):955-958. (WEI X L, SHEN W M. GaborMRF model based on color texture image segmentation [J]. Geomatics and Information Science of Wuhan University, 2014, 38(8): 955-958.)