CN106709921B - 一种基于空间Dirichlet混合模型的彩色图像分割方法 - Google Patents
一种基于空间Dirichlet混合模型的彩色图像分割方法 Download PDFInfo
- Publication number
- CN106709921B CN106709921B CN201611242922.9A CN201611242922A CN106709921B CN 106709921 B CN106709921 B CN 106709921B CN 201611242922 A CN201611242922 A CN 201611242922A CN 106709921 B CN106709921 B CN 106709921B
- Authority
- CN
- China
- Prior art keywords
- dirichlet
- model
- vector
- spatial
- color image
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Active
Links
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/10—Image acquisition modality
- G06T2207/10024—Color image
Landscapes
- Image Analysis (AREA)
Abstract
本发明涉及一种基于空间Dirichlet混合模型的彩色图像分割方法,所提出的统计模型基于Dirichlet混合模型,Dirichlet分布是一个多元泛化的Beta分布,对比其他分布(例如:高斯分布、)它能够包含对称与非对称模态,因此可以灵活处理各种图像以及体现出较好的分割准确率。本发明在Dirichlet混合模型中融入空间邻域特征,从而体现出较好的鲁棒性。在求解Dirichlet混合模型参数过程中,本发明采用基于变分贝叶斯推断的方法,使得模型求解更加准确高效。克服了现有技术的方法采用的最大似然估计(ML)和最大后验概率估计(MAP)存在的不足:这些方法通常容易在求解模型过程中陷入局部极小值,而且会造成迭代次数过多。
Description
技术领域
本发明涉及计算机图像分析与处理领域,更具体地说,涉及一种基于空间Dirichlet混合模型的彩色图像分割方法。
背景技术
近几年,随着信息时代特别是数字时代的来临,图像分割技术在医学、军事工程等领域中都有着广泛的应用。图像分割技术可以便于对获得的各种定量定性的图像数据进行分析,使这些数据更好的应用到工程实践中。例如在医学上,计算机断层成像(CT:ComputedTomography),核磁共振成像(MRI:Magnetic Resonance Imaging),超声(US:Ultrasonography)等新医学成像技术已经广泛应用在医疗诊断、术前计划、治疗、术后监测等各个环节中。
图像分割是对图像进行分析和理解的关键步骤,是数字图像处理中最重要和最基本的技术手段,是一种基本的计算机视觉技术,准确地对图像进行分割对工程实践及量化分析都有重要的价值。
图像分割方法主要包括以下四类。1)基于统计学方法:此方法利用图像的像素的统计特征和相关性准则进行分割,因此该方法需要建立合适的统计模型并进行模型的参数求解。此方法依据图像的灰度信息,利用特种提取和建立统计模型,方法灵活;2)基于边缘检测方法:此方法主要依赖于周围像素灰度有阶跃变化或“屋顶”变化的那些像素,然后利用边缘邻近一阶或二阶方向导数变化规律,用简单的边缘算子检测边缘,常见的检测算子有Robert算子、Sobel算子、Prewitt算子、Laplacian算子,该方法需要根据特定的图选定特定的算子;3)基于区域生长分割方法:此方法先对每个需要分割的区域找出一个种子象素作为生长的起点,然后将种子象素周围邻域中与种子有相同或相似性质的象素(根据事先确定的生长或相似准则来确定,如门限)合并到种子象素所在的区域中,而新的象素继续做种子向四周生长,需要选择合适的种子和门限,否则分割效果大打折扣;4)基于阈值分割方法:该方法通过设定不同的阈值,然后将像素跟阈值比较,看是否超过阈值,从而把图像像素点分成不同的类,该方法只在物体跟背景有较强对比的时候有很好的分割效果,而且阈值的合理与否关系到整个分割效果的好坏。
混合模型隶属统计模型中的一种,它可以看成是由2个或者2个以上的分布按照某种规则叠加,而每个分布对应各自不同的加权系数。最近几十年来,基于混合模型的图像分割的研究也取得了很多的成果,代表性的有:Y.Zhang等人(Y.Zhang,M.Brady,andS.Smith,“Segmentation of brain MR images through a hidden Markov random fieldmodel and the expectation-maximization algorithm”IEEE Trans.Med.Imag.,vol.20,no.1,pp.45–57,Jan.2001)提出的利用期望最大化(EM)算法求解隐马尔科夫随机场模型(HMRF-EM)进行图像分割。但是该方法采用EM算法估计模型参数,易陷入局部极小值。SKrinidis等人(SKrinidis,V Chatzis,“A robust fuzzy local information C-meansclustering algorithm”IEEE Trans.Imag.Proc.,vol.19,no.5,pp.1328-1337,May 2010)提出来一种新的模糊c均值聚类算法(FLICM)。该方法相比于HMRF-EM有较强的鲁棒性,因此分割效果进一步提高,但是该算法在最小化目标函数过程中并不是严格按照梯度下降,因此导致迭代次数过多,增加计算的复杂性。
在图像分割领域,需要考虑如何解决高噪声、计算开销大两大问题,因此提出一种鲁棒性和简单有效的算法是当今图像分割领域的重点研究方向之一。
发明内容
本发明的目的在于克服现有技术的不足,提供一种在计算开销小的情况下快速准确地得到彩色图像的分割结果的基于空间Dirichlet混合模型的彩色图像分割方法。
本发明的技术方案如下:
一种基于空间Dirichlet混合模型的彩色图像分割方法,步骤如下:
1)建立有限Dirichlet混合模型,对输入的彩色图像进行数据预处理,得到符合有限Dirichlet混合模型求解的图像数据;
2)利用有限Dirichlet混合模型对图像数据进行建模;
3)利用变分贝叶斯推断方法求解模型参数,并得到新的标签向量;
4)将标签向量作为输出向量,并转换成灰度矩阵进行图像分割。
作为优选,步骤1)中,对输入的原彩色图像进行预处理操作,得到N*M*3的数据矩阵,然后将相同像素点的3个维度数据转换成概率分布数据,得到3行N*M列的矩阵,作为有限Dirichlet混合模型的输入向量。
作为优选,对所述的输入向量中的三个数据通道(R、G、B)进行数据转换,转换公式如下:
作为优选,采用K-means方法对输入向量初始化,并设置聚类数K。
作为优选,步骤2)中,利用融合了空间邻域关系的Dirichlet混合模型对向量数据进行建模,并设置空间领域参数q、u、v、a、b;融合了空间邻域关系的Dirichlet混合模型如下:
ξi服从Dirichlet的先验分布,定义如下:
空间关系模型,定义如下:
<Zij>=rij,
最终得到期望值后验概率矩阵:
作为优选,循环迭代以下步骤,直到收敛或者达到预设值:
a)利用当前模型的预设值,计算如下公式:
<Zij>=rij,
作为优选,将得到标签向量转换成与原彩色图像对应的N*M的2D图像形式,从而得到最终的分割结果。
本发明的有益效果如下:
本发明所述的基于空间Dirichlet混合模型的彩色图像分割方法中,所提出的统计模型基于Dirichlet混合模型,Dirichlet分布是一个多元泛化的Beta分布,对比其他分布(例如:高斯分布、)它能够包含对称与非对称模态,因此可以灵活处理各种图像以及体现出较好的分割准确率。
本发明在Dirichlet混合模型中融入空间邻域特征,从而体现出较好的鲁棒性。
在求解Dirichlet混合模型参数过程中,本发明采用基于变分贝叶斯推断的方法,使得模型求解更加准确高效。克服了现有技术的方法采用的最大似然估计(ML)和最大后验概率估计(MAP)存在的不足:这些方法通常容易在求解模型过程中陷入局部极小值,而且会造成迭代次数过多。
附图说明
图1是本发明的方法流程图;
图2为原图2(a)分别在三种方法下的二类分割效果图,分别为图2(b)、图2(c)、图2(d);
图3为原图3(a)分别在三种方法下的四类分割效果图,分别为图3(b)、图3(c)、图3(d);
图4为原图4(a)分别在三种方法下的六类分割效果图,分别为图4(b)、图4(c)、图4(d);
具体实施方式
以下结合附图及实施例对本发明进行进一步的详细说明。
本发明所述的基于空间Dirichlet混合模型的彩色图像分割方法,如图1所示,步骤如下:
1)建立有限Dirichlet混合模型,对输入的彩色图像进行数据预处理,得到符合有限Dirichlet混合模型求解的图像数据;
2)利用有限Dirichlet混合模型对图像数据进行建模;
3)利用变分贝叶斯推断方法求解模型参数,并得到新的标签向量;主要通过两个子步骤进行:
3.1)利用贝叶斯变分推导估计参数模型;
3.2)利用贝叶斯最大后验概率准则得到输入向量数据对应的标签向量的后验概率矩阵;
4)将标签向量作为输出向量,并转换成灰度矩阵进行图像分割。
步骤1)中,对输入的原彩色图像进行预处理操作,得到N*M*3的数据矩阵,然后将相同像素点的3个维度数据转换成概率分布数据,得到3行N*M列的矩阵,作为有限Dirichlet混合模型的输入向量,以便进行标签向量生成操作。
对所述的输入向量中的三个数据通道(R、G、B)进行数据转换,RGB即为图像的三原色通道,转换公式如下:
采用K-means方法对输入向量初始化,并设置聚类数K。
步骤2)中,利用融合了空间邻域关系的Dirichlet混合模型对向量数据进行建模,并设置空间领域参数q、u、v、a、b;融合了空间邻域关系的Dirichlet混合模型如下:
ξi服从Dirichlet的先验分布,定义如下:
空间关系模型,定义如下:
步骤3)中,采用variational inference对Zij、进行估计,采用variationalinference,能极大的降低运算,而且不易陷入局部极小值,其中采用平均场方法进行参数估计的转换,如下所示:
<Zij>=rij,
最终得到期望值后验概率矩阵:
作为优选,循环迭代以下步骤,直到收敛或者达到预设值:
a)利用当前模型的预设值,计算如下公式:
<Zij>=rij,
步骤4)中,将后验概率矩阵转换成标签向量,然后将得到标签向量转换成与原彩色图像对应的N*M的2D图像形式,从而得到最终的分割结果。
本发明的效果可以通过以下仿真实验得到验证:
1.仿真条件:
在CPU为Intel(R)Core(TM)i5-4590处理器:主频3.30GHZ,内存8G,操作系统:WINDOWS 10,仿真平台:MATLAB 2014Ra。
2.评价指标:
概率边缘指数(PRI)是一种用于检测实际分割效果跟参考结果之间的属性共生的一致性的参量的评价指标,PRI计算公式如下:
上述公式中li、lj表示任意一对像元xi、xj在图像S中的像元标记,l′i、l′j表示在图像Stest中的像元标记,I表示判别函数。PRI的取值范围为[0,1],PRI越大表示分割结果越接近标准分割图,即实际分割结果跟参考结果之间的属性共生一致性越好。
3.仿真图像:
仿真图像采用自然图像图2(a)、图3(a)、图4(a)(BSDS500)原始测试图像,其中图2(b)-2(d)为原图2(a)分别在HMRF-EM、FLICM、本发明三种方法下的分割成二类结果,图3(b)-3(d)为原图3(a)分别在HMRF-EM、FLICM、本发明三种方法下的分割成四类结果,图4(b)-4(d)为原图4(a)分别在HMRF-EM、FLICM、本发明三种方法下的分割成六类结果。
4.仿真对比方法
对比方法1:Y.Zhang等人在文献“Segmentation of brain MR images through ahidden Markov random field model and the expectation-maximization algorithmIEEE Trans.Med.Imag.,vol.20,no.1,pp.45–57,Jan.2001”提出的利用期望最大化(EM)算法求解隐马尔科夫随机场模型的方法,简称HMRF-EM。
对比方法2:S Krinidis等人在文献“S Krinidis,V Chatzis,“A robust fuzzylocal information C-means clustering algorithm IEEE Trans.Imag.Proc.,vol.19,no.5,pp.1328-1337,may.2010”提出来一种新的模糊c均值聚类算法,简称FLICM。
5.仿真内容:
仿真1,利用现有的2种对比方法和本发明的方法对图2(a)进行图像分割,得到对应的分割效果图见图2(b)-2(d),另外仿真得到的数据结果见表1。
表1:图2(a)在3种方法下的分割结果。
结合图2以及表1,可以看出本发明在自然图像分割上较其他两种方法提高近5%~10%左右的效果,由此可以充分的说明本发明方法较其他三种方法对图像分割更具有效性。
仿真2,利用现有的2种对比方法和本发明的方法对图3(a)进行图像分割,得到对应的分割效果图见图3(b)-图3(d),另外仿真得到的数据结果见表2。
表2 图3(a)在二种方法下的分割结果。
结合图3跟表2,本发明的方法的PRI在90%以上,明显高于其他二种方法,而且从分割图效果来看,人物轮廓分割较为清楚。
仿真3,利用现有的2种对比方法和本发明的方法对图4(a)进行图像分割,得到对应的分割效果图见图4(b)-图4(d),另外仿真得到的数据结果见表3。
表3 图4(a)在二种方法下的分割结果。
结合图4跟表3,本发明的方法的PRI在80%左右,明显高于其他二种方法,而且从分割图效果来看,人物以及景物的区域轮廓分割得比较好,其他两种方法分割下的图像没有考虑到邻域一致性,所以分割效果差。
综上所述,本发明在图像分割方面,由于采用了空间邻域关系,以及采用Dirichlet混合模型,效果优于现有的二种分割方法。
上述实施例仅是用来说明本发明,而并非用作对本发明的限定。只要是依据本发明的技术实质,对上述实施例进行变化、变型等都将落在本发明的权利要求的范围内。
Claims (7)
1.一种基于空间Dirichlet混合模型的彩色图像分割方法,其特征在于,步骤如下:
1)建立有限Dirichlet混合模型,对输入的彩色图像进行数据预处理,得到符合有限Dirichlet混合模型求解的图像数据;
步骤1)中,对输入的原彩色图像进行预处理操作,得到N*M*3的数据矩阵,然后将相同像素点的3个维度数据转换成概率分布数据,得到3行N*M列的矩阵,作为有限Dirichlet混合模型的输入向量;
2)利用有限Dirichlet混合模型对图像数据进行建模;
步骤2)中,利用融合了空间邻域关系的Dirichlet混合模型对向量数据进行建模,并设置空间邻域参数q、u、v、a、b;融合了空间邻域关系的Dirichlet混合模型如下:
ξi服从Dirichlet的先验分布,定义如下:
3)利用变分贝叶斯推断方法求解模型参数,并得到新的标签向量;
4)将标签向量作为输出向量,并转换成灰度矩阵进行图像分割。
3.根据权利要求2所述的基于空间Dirichlet混合模型的彩色图像分割方法,其特征在于,采用K-means方法对输入向量初始化,并设置聚类数Km。
7.根据权利要求6所述的基于空间Dirichlet混合模型的彩色图像分割方法,其特征在于,将得到标签向量转换成与原彩色图像对应的N*M的2D图像形式,从而得到最终的分割结果。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201611242922.9A CN106709921B (zh) | 2016-12-29 | 2016-12-29 | 一种基于空间Dirichlet混合模型的彩色图像分割方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201611242922.9A CN106709921B (zh) | 2016-12-29 | 2016-12-29 | 一种基于空间Dirichlet混合模型的彩色图像分割方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN106709921A CN106709921A (zh) | 2017-05-24 |
CN106709921B true CN106709921B (zh) | 2020-06-09 |
Family
ID=58906004
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN201611242922.9A Active CN106709921B (zh) | 2016-12-29 | 2016-12-29 | 一种基于空间Dirichlet混合模型的彩色图像分割方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN106709921B (zh) |
Families Citing this family (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108805012B (zh) * | 2018-04-24 | 2019-10-25 | 西安理工大学 | 基于无穷Dirichlet过程混合模型的指节识别方法 |
CN109993756B (zh) * | 2019-04-09 | 2022-04-15 | 中康龙马(北京)医疗健康科技有限公司 | 一种基于图模型与连续逐步优化的通用医学图像分割方法 |
Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN102496146A (zh) * | 2011-11-28 | 2012-06-13 | 南京大学 | 一种基于视觉共生的图像分割方法 |
CN103870840A (zh) * | 2014-03-11 | 2014-06-18 | 西安电子科技大学 | 基于改进的潜在狄利克雷分配的自然图像分类方法 |
CN106023236A (zh) * | 2016-06-16 | 2016-10-12 | 华侨大学 | 基于截断Dirichlet过程无限Student’s t 混合模型的脑部核磁共振图像分割方法 |
Family Cites Families (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US7542604B2 (en) * | 2004-08-26 | 2009-06-02 | Siemens Medical Solutions Usa, Inc. | System and method for image segmentation by solving an inhomogenous dirichlet problem |
-
2016
- 2016-12-29 CN CN201611242922.9A patent/CN106709921B/zh active Active
Patent Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN102496146A (zh) * | 2011-11-28 | 2012-06-13 | 南京大学 | 一种基于视觉共生的图像分割方法 |
CN103870840A (zh) * | 2014-03-11 | 2014-06-18 | 西安电子科技大学 | 基于改进的潜在狄利克雷分配的自然图像分类方法 |
CN106023236A (zh) * | 2016-06-16 | 2016-10-12 | 华侨大学 | 基于截断Dirichlet过程无限Student’s t 混合模型的脑部核磁共振图像分割方法 |
Non-Patent Citations (1)
Title |
---|
《Dirichlet 过程混合模型SAR图像分割算法》;孙莉;《吉林大学学报(工学版)》;20090930;第1345-1346页 * |
Also Published As
Publication number | Publication date |
---|---|
CN106709921A (zh) | 2017-05-24 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN109409416B (zh) | 特征向量降维方法和医学图像识别方法、装置及存储介质 | |
Bi et al. | Automatic liver lesion detection using cascaded deep residual networks | |
CN110889852B (zh) | 基于残差-注意力深度神经网络的肝脏分割方法 | |
Pohl et al. | Using the logarithm of odds to define a vector space on probabilistic atlases | |
Wang et al. | Adaptive spatial information-theoretic clustering for image segmentation | |
Aranguren et al. | Improving the segmentation of magnetic resonance brain images using the LSHADE optimization algorithm | |
Banerjee et al. | Rough sets and stomped normal distribution for simultaneous segmentation and bias field correction in brain MR images | |
Wu et al. | Registration of longitudinal brain image sequences with implicit template and spatial–temporal heuristics | |
Joshi et al. | Non-parametric mixture model based evolution of level sets and application to medical images | |
Tang et al. | Reliability-driven, spatially-adaptive regularization for deformable registration | |
Galimzianova et al. | Stratified mixture modeling for segmentation of white-matter lesions in brain MR images | |
Cai et al. | Accurate weakly supervised deep lesion segmentation on CT scans: Self-paced 3D mask generation from RECIST | |
CN110097499B (zh) | 基于谱混合核高斯过程回归的单帧图像超分辨率重建方法 | |
CN109285176B (zh) | 一种基于正则化图割的大脑组织分割方法 | |
CN106709921B (zh) | 一种基于空间Dirichlet混合模型的彩色图像分割方法 | |
Pascal et al. | Strongly convex optimization for joint fractal feature estimation and texture segmentation | |
Ridhma et al. | Review of automated segmentation approaches for knee images | |
Ramana | Alzheimer disease detection and classification on magnetic resonance imaging (MRI) brain images using improved expectation maximization (IEM) and convolutional neural network (CNN) | |
Wen et al. | Brain tissue classification based on DTI using an improved fuzzy C-means algorithm with spatial constraints | |
Soliman et al. | Segmentationof pathological lungs from CT chest images | |
Song et al. | An extension Gaussian mixture model for brain MRI segmentation | |
CN112927235A (zh) | 基于多尺度超像素与核低秩表示的脑肿瘤图像分割方法 | |
Zanaty | An adaptive fuzzy C-means algorithm for improving MRI segmentation | |
He et al. | Multi-channel features based automated segmentation of diffusion tensor imaging using an improved FCM with spatial constraints | |
Bricq et al. | MS lesion segmentation based on hidden Markov chains |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |