1. 程式人生 > >從貝葉斯理論到影象馬爾科夫隨機場

從貝葉斯理論到影象馬爾科夫隨機場

轉載自:http://blog.csdn.net/on2way/article/details/47307927(影象處理、機器學習的一些演算法)

本節主要介紹馬爾科夫的隨機場模型以及其用於影象的分割演算法中。基於馬爾科夫的隨機場(MRF)的影象分割是一種基於統計的影象分割演算法,其模型引數少,空間約束性強,使用較為廣泛。

首先了解一下馬爾科夫模型,純粹的馬爾科夫模型就是指一件事物的當前狀態只與它之前的1個或者n個狀態有關,而與再之前的狀態沒有關係,比如今天天氣好壞只與昨天天氣有關,而與前天乃至大前天都沒有關係。符合這樣的一種特性的事物認為其具有馬爾科夫性。那麼引申到影象領域,就是認為影象中某一點的特徵(一般都是畫素點灰色、顏色值等)只與其附近的一小塊領域有關,而與其他的領域無關。想想也是這樣,大多數時候,影象中某一點畫素和附近的畫素是相關的,附近領域畫素是黑的,那它八成也是黑的,很好理解。

馬爾科夫隨機場在影象領域的一大用途就是影象的分割,本文也主要介紹該方法的影象分割。

關於影象分割問題,從聚類角度講,就是一個影象的聚類問題,把具有相同性質的畫素點設定為一類。在說詳細點就是一個標籤分類問題,比如把一副影象分割成4類,那麼每一個畫素點必定屬於這四類中的某一類,假設四類為1,2,3,4類,那麼分割就是給每個畫素點找一個標籤類。好了,假設我們的待分割影象是S,大小m*n,那麼把每個畫素點放到集合S中,影象就是:S=S1,S2,...,Smn,把待分割的影象稱為觀測到的影象。現在假設分割好了,每個畫素點都分了類,那麼把最終的分割結果稱為W,那麼顯然w大小與S一樣大,W=w1,w

2,...,wmn,其中所有的w取值都在1-L之間,L是最大的分類數,比如上面L=4,那麼每個w都是在1-4之間取值的。

這是我們假設知道了最終分割結果W,但是W在開始是不知道的?現在的問題就是如何求W,我們所知道的不過是觀測到的影象S,也就是說在知道S情況下求W,轉化為概率就是求取P(W|S),問題就變成求取這個概率的最大值,也就是說根據S我們算出這個影象的最有可能的分割標籤是什麼。

基於此,問題才剛剛開始。我們由貝葉斯理論知道

P(W|S)=P(S|W)P(W)P(S)
觀察這個式子先不看能不能求出來,先給一些定義。因為W就是我們要求的分類標籤,我們稱P(W)為標記場w的先驗概率(要求它,事先又不知道,那就叫先驗概率吧,不知道是不是這樣來的)。P(S|W)是觀察值S的
條件概率分佈(也稱似然函式),看結構知道,這個是說在已知畫素點標記w的情況下,它是真實的畫素點s的概率,所以是一個似然函式,表示我的觀察畫素點和真實的分類情況w像不像的一種程度。那P(S)是什麼?S是觀察到的影象,在分割前它就已經定了,我就要分割這幅影象,那它還有概率嗎?沒有吧,所以P(S)認為是個定值。那麼現在要求P(W|S)的最大值,也就是要求P(S|W)P(W)的最大值。

先來說說P(W)這個標記場w的先驗概率。那麼我們落實到影象的每一個畫素點上,也就是說這個畫素點是標籤L的概率是多少,有人可能會說,分割之前影象的分類標籤都不知道,還怎麼算是某一類標籤L的概率,這個問題是有點較勁不好理解。我覺得,這就是一個雞蛋問題,是先有雞還是先有蛋的問題,我們要求這隻雞,但是又沒有蛋怎麼辦?一個簡單的辦法是我隨便弄一個蛋來,管他是雞蛋鴨蛋,有了蛋,雞(或者鴨)就可以有了,雖然這個雞不太像雞,比如說是隻野雞,那麼有了野雞,野雞再稍微進化一下,然後再下個蛋,蛋又長出野雞1號,野雞1號再稍微進化一下,然後再下個蛋,變成野雞2號,如此反覆反覆,知道野雞n號,然後驀然回首發現,野雞n號和我們所要的雞竟是那麼的相似,那麼把它就認為是我們要的雞了。又沒有糊塗?其實優化聚類裡面很多問題都是雞蛋問題,比如曾經介紹的FCM演算法,有個先給u還是先給c的雞蛋問題,u的計算需要c,c的計算有需要u,那怎麼辦,先假設一個吧。再比如EM演算法的迭代過程也可以看成雞蛋問題,有了E步算M步,在回來算E步,在M步。。。。最終得到最優解。

好,回到我們的問題,我們的問題要幹嘛?要求一個畫素點是標籤L的概率,開始標籤沒有,ok假設一個吧,一個不行,那麼在開始我們把整幅影象初始化一個分割標籤,儘管這個標籤不對(野雞),也可以假設一個稍微對一點的標籤(比如用其他分類演算法(kmeans)得到一個預分割標籤)(這個時候認為是野雞100號)(好處在於這個時候演算法不用迭代那麼多次就可以停止了)。好了有了初始的標籤,那麼再來求一個畫素點是標籤L的概率。從這個時候就需要馬爾科夫協助了。我們想,要求一個畫素點是標籤L的概率,此時這個畫素點附近的領域都也已經劃分標籤了,雖然這個畫素點也有一個標籤,但是這個標籤是上一代的標籤,那麼它到下一代是需要進化的,馬爾科夫性質告訴我們這個畫素點的分類情況只與附近一些領域分類情況有關,而與另一些領域沒有關係,也就是說我們可以根據這個畫素點的附近領域的分類情況決定這個畫素點在下一代是屬於哪一類的。

Ok既然我們認為每個畫素點的分類符合馬爾科夫隨機模型,而另一些學者證明了這種馬爾科夫隨機場可以與一個Gibbs隨機場等價(這就是Hammcrslcy-Clifford定理,人家證明的,那就這樣叫吧)。而Gibbs隨機場也有一個概率密度函式,這樣我們就用求影象的Gibbs隨機場的概率P代替了P(W)。那麼Gibbs隨機場的概率P怎麼表示呢?如下:

P(w)=z1e1TU2(w)
其中,z=weU2(w)T是歸一化常數,引數T可控制P(w)的形狀,T越大越平坦。而U2(w)=cCVc(wc),C為所有勢團集合,Vc(wc)為勢團勢能。對Vc(wc)的定義如下: 
Vc(wc)=Vs,r(ws,wr)={ββws=wtwswt β為耦合係數,通常0.5-1。 
糊不糊塗,不糊塗才怪,這麼多關係,Gibbs也不容易,還沒完,那麼什麼是勢團,說白了就是和這個畫素點構成一個小的檢測這個點連通效能的區域,一個圖如下: 
這裡寫圖片描述
總算完了,乍一看真是複雜,不知所云,其實細看還是可以理解的。說白了就是檢測一下這個點和周圍領域的相似程度。那些勢團就是要比較的物件。舉個例子,以一階勢團為例,假設某個點及其附近的8領域在某次迭代後的分類標籤是這樣的(假設分四類的情況): 
211134212
現在要計算中心的畫素點在下一次迭代中是屬於第幾類(這一代是第3類),ok採用一階勢能,這裡需要說明一點,這個畫素點無非是1-4之間的一類,那麼我們需要分別計算下一代它是第1,2,3,4類時的勢團。先假設這個點是第1類,先比較左右,發現都是1-1一樣,ok記一下β,在與上β,與下,不一樣,那麼記一下β,如果再來勢團的話,斜上方,1-2,不一樣β,斜下方,1-1一樣,β,一次類推,就可以將中心點如果是第1類的勢團計算出來。那麼在計算中心點如果是第2類,發現這個時候除了3個斜著的方向是一樣的外,其他的都不一樣。在計算假設是第三類,第四類。這樣有了每類下的勢團,就可以計算概率了,根據公式往回帶呀帶,最終得到這個點如果屬於第一類的概率,第二類的概率,第三類的概率,第四類的概率,這四個值在後面會一起用到來決定這個點到底屬於哪一類。你可能會說,知道了這四個概率,看哪個最大不就出來了嗎?注意這還只是第一項的先驗概率,後面還有個似然函式的概率沒有計算呢,後面你會看到這個似然函式的概率對於每個畫素點也有四個概率,在這四個概率都計算出來之後,那麼這個點屬於第一類的概率用著兩個都屬於第一類的概率相乘,第二類也是如此,最後再比較這四個概率的最大值作為要標記的標籤。

通觀Gibbs,其實就是一個求畫素點與周圍畫素點的不一致程度,或者說這個畫素點的周圍畫素點中,看看各個標籤出現的概率多大,就是這個點屬於對應標籤的概率。所以在實際程式設計上,也沒看到幾個人實實在在的用它的原版公式程式設計,反而簡化的計算,看看各個標籤出現的次數等等方式來計算的。

關於P(W)就說這麼多,下面說說P(S|W)。P(S|W),已知分類標籤那麼它的畫素值(灰度)是s的概率。現在就假設w=1,某個畫素點灰度為s,那麼表示的意思就是在第一類裡面畫素灰度為s的概率。因為分類標籤在前面說到,每次迭代的時候是有一個分類標籤的(儘管不是最後的標籤),那麼我們可以把屬於第一類的所有點都挑出來,考慮每個點都是獨立的,並且認為每一類裡面的所有點服從高斯分佈(正態分佈),那麼在每一類裡面我們是不是可以根據這一類裡面的這些點建立一個屬於這一類的高斯密度函數了,那麼再來一個畫素點值,把它帶到這類裡面密度函式中去就可以得到這個概率了吧。同理對於2,3,4類,每一類都可以建立一個高斯密度函式,這樣就有四個高斯密度函數了,那麼每一個點屬於這四類的概率就可以分別帶到這四個高斯密度函式中計算了。高斯密度函式一般形式為: 

P(x|wi)=12πσexp((xu)22σ2)
舉個例子假設現在我們求得的四類高斯函式,其影象如下所示: 
這裡寫圖片描述
某一點的灰度為s=110,那麼它對應的分別P(s|W1)、P(s|W2)、P(s|W3)、P(s|W3),就分別如上圖所示呢,很顯然的可以看到110在第三類下的概率最大,其他的幾個概率也可以出來的。

現在這一部分對於每個點也有了四個概率,結合上面每個點的四個概率,那麼對每個點,屬於每一類的概率對應相乘,得到最終每個點屬於四個類的概率。這個時候就可以挑其中最大的那麼就是該點所屬於的類了。 
這樣一次迭代,所有的點所屬於的類更新一遍,那這個新的類標籤作為下一次迭代的類標籤,反覆下去,程式結束的條件可以是設定迭代次數,也可以是觀察類中心不在變化為止。 
好了,上述的概率相乘取最大是原始一點的演算法。其實可以看到,這個計算包括兩個部分,一般的講法,認為這兩部分每一部分都組成一個能量,換個說法就是最優化能量函數了,可以表示為如下:

W=argmin(U1(w,S)+U2(w))
像上述的概率相乘在實際中取一下對數就變成相加了。更深層次的馬爾科夫隨機場可以去看看相關論文,雖然方法可能不太一樣,但是原理上是一樣的。

下面以條件迭代演算法(ICM)來例項闡述MRF在影象分割上的應用,程式設計平臺為matlab,相關程式碼如下:

<code class="hljs ruby has-numbering" style="display: block; padding: 0px; color: inherit; box-sizing: border-box; font-family: 'Source Code Pro', monospace;font-size:undefined; white-space: pre; border-radius: 0px; word-wrap: normal; background: transparent;">clc
clear
close all
img = double(imread(<span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">'lena.jpg'</span>));%more quickly
cluster_num = <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">4</span>;%設定分類數
maxiter = <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">60</span>;%最大迭代次數
<span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">%--</span>-----------隨機初始化標籤----------------
label = randi([<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>,cluster_num],size(img));
<span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">%--</span>---------kmeans最為初始化預分割----------
% label = kmeans(img(<span class="hljs-symbol" style="color: rgb(0, 102, 102); box-sizing: border-box;">:</span>),cluster_num);
% label = reshape(label,size(img));
iter = <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;
<span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">while</span> iter < maxiter
    <span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">%--</span>-----計算先驗概率---------------
    %這裡我採用的是畫素點和<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">3</span>*<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">3</span>領域的標籤相同
    %與否來作為計算概率
    <span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">%--</span>----收集上下左右斜等八個方向的標籤--------
    label_u = imfilter(label,[<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>],<span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">'replicate'</span>);
    label_d = imfilter(label,[<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>],<span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">'replicate'</span>);
    label_l = imfilter(label,[<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>],<span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">'replicate'</span>);
    label_r = imfilter(label,[<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>],<span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">'replicate'</span>);
    label_ul = imfilter(label,[<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>],<span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">'replicate'</span>);
    label_ur = imfilter(label,[<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>],<span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">'replicate'</span>);
    label_dl = imfilter(label,[<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>],<span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">'replicate'</span>);
    label_dr = imfilter(label,[<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>;<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>],<span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">'replicate'</span>);
    p_c = zeros(<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">4</span>,size(label,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>)*size(label,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">2</span>));
    %計算畫素點<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">8</span>領域標籤相對於每一類的相同個數
    <span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">for</span> i = <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span><span class="hljs-symbol" style="color: rgb(0, 102, 102); box-sizing: border-box;">:cluster_num</span>
        label_i = i * ones(size(label));
        temp = ~(label_i - label_u) + ~(label_i - label_d) + ...
            ~(label_i - label_l) + ~(label_i - label_r) + ...
            ~(label_i - label_ul) + ~(label_i - label_ur) + ...
            ~(label_i - label_dl) +~(label_i - label_dr);
        p_c(i,<span class="hljs-symbol" style="color: rgb(0, 102, 102); box-sizing: border-box;">:</span>) = temp(<span class="hljs-symbol" style="color: rgb(0, 102, 102); box-sizing: border-box;">:</span>)/<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">8</span>;%計算概率
    <span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">end</span>
    p_c(find(p_c == <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>)) = <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>.<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">001</span>;%防止出現<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>
    <span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">%--</span>-------------計算似然函式----------------
    mu = zeros(<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">4</span>);
    sigma = zeros(<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">4</span>);
    %求出每一類的的高斯引數--均值方差
    <span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">for</span> i = <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span><span class="hljs-symbol" style="color: rgb(0, 102, 102); box-sizing: border-box;">:cluster_num</span>
        index = find(label == i);%找到每一類的點
        data_c = img(index);
        mu(i) = mean(data_c);%均值
        sigma(i) = var(data_c);%方差
    <span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">end</span>
    p_sc = zeros(<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">4</span>,size(label,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>)*size(label,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">2</span>));
%     <span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">for</span> i = <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span><span class="hljs-symbol" style="color: rgb(0, 102, 102); box-sizing: border-box;">:size</span>(img,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>)*size(img,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">2</span>)
%         <span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">for</span> j = <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span><span class="hljs-symbol" style="color: rgb(0, 102, 102); box-sizing: border-box;">:cluster_num</span>
%             p_sc(j,i) = <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>/sqrt(<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">2</span>*pi*sigma(j))*...
%               exp(-(img(i)-mu(j))^<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">2</span>/<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">2</span>/sigma(j));
%         <span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">end</span>
%     <span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">end</span>
    <span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">%--</span>----計算每個畫素點屬於每一類的似然概率--------
    <span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">%--</span>----為了加速運算,將迴圈改為矩陣一起操作--------
    <span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">for</span> j = <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span><span class="hljs-symbol" style="color: rgb(0, 102, 102); box-sizing: border-box;">:cluster_num</span>
        <span class="hljs-constant" style="box-sizing: border-box;">MU</span> = repmat(mu(j),size(img,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>)*size(img,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">2</span>),<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>);
        p_sc(j,<span class="hljs-symbol" style="color: rgb(0, 102, 102); box-sizing: border-box;">:</span>) = <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>/sqrt(<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">2</span>*pi*sigma(j))*...
            exp(-(img(<span class="hljs-symbol" style="color: rgb(0, 102, 102); box-sizing: border-box;">:</span>)-<span class="hljs-constant" style="box-sizing: border-box;">MU</span>).^<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">2</span>/<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">2</span>/sigma(j));
    <span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">end</span> 
    %找到聯合一起的最大概率最為標籤,取對數防止值太小
    [~,label] = max(log(p_c) + log(p_sc));
    %改大小便於顯示
    label = reshape(label,size(img));
    <span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">%--</span>-------顯示----------------
    <span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">if</span> ~mod(iter,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">6</span>) 
        figure;
        n=<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>;
    <span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">end</span>
    subplot(<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">2</span>,<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">3</span>,n);
    imshow(label,[])
    title([<span class="hljs-string" style="color: rgb(0, 136, 0); box-sizing: border-box;">'iter = '</span>,num2str(iter)]);
    pause(<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">0</span>.<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>);
    n = n+<span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>;
    iter = iter + <span class="hljs-number" style="color: rgb(0, 102, 102); box-sizing: border-box;">1</span>;
<span class="hljs-keyword" style="color: rgb(0, 0, 136); box-sizing: border-box;">end</span>
</code><ul class="pre-numbering" style="box-sizing: border-box; position: absolute; width: 50px; top: 0px; left: 0px; margin: 0px; padding: 6px 0px 40px; border-right-width: 1px; border-right-style: solid; border-right-color: rgb(221, 221, 221); list-style: none; text-align: right; background-color: rgb(238, 238, 238);"><li style="box-sizing: border-box; padding: 0px 5px;">1</li><li style="box-sizing: border-box; padding: 0px 5px;">2</li><li style="box-sizing: border-box; padding: 0px 5px;">3</li><li style="box-sizing: border-box; padding: 0px 5px;">4</li><li style="box-sizing: border-box; padding: 0px 5px;">5</li><li style="box-sizing: border-box; padding: 0px 5px;">6</li><li style="box-sizing: border-box; padding: 0px 5px;">7</li><li style="box-sizing: border-box; padding: 0px 5px;">8</li><li style="box-sizing: border-box; padding: 0px 5px;">9</li><li style="box-sizing: border-box; padding: 0px 5px;">10</li><li style="box-sizing: border-box; padding: 0px 5px;">11</li><li style="box-sizing: border-box; padding: 0px 5px;">12</li><li style="box-sizing: border-box; padding: 0px 5px;">13</li><li style="box-sizing: border-box; padding: 0px 5px;">14</li><li style="box-sizing: border-box; padding: 0px 5px;">15</li><li style="box-sizing: border-box; padding: 0px 5px;">16</li><li style="box-sizing: border-box; padding: 0px 5px;">17</li><li style="box-sizing: border-box; padding: 0px 5px;">18</li><li style="box-sizing: border-box; padding: 0px 5px;">19</li><li style="box-sizing: border-box; padding: 0px 5px;">20</li><li style="box-sizing: border-box; padding: 0px 5px;">21</li><li style="box-sizing: border-box; padding: 0px 5px;">22</li><li style="box-sizing: border-box; padding: 0px 5px;">23</li><li style="box-sizing: border-box; padding: 0px 5px;">24</li><li style="box-sizing: border-box; padding: 0px 5px;">25</li><li style="box-sizing: border-box; padding: 0px 5px;">26</li><li style="box-sizing: border-box; padding: 0px 5px;">27</li><li style="box-sizing: border-box; padding: 0px 5px;">28</li><li style="box-sizing: border-box; padding: 0px 5px;">29</li><li style="box-sizing: border-box; padding: 0px 5px;">30</li><li style="box-sizing: border-box; padding: 0px 5px;">31</li><li style="box-sizing: border-box; padding: 0px 5px;">32</li><li style="box-sizing: border-box; padding: 0px 5px;">33</li><li style="box-sizing: border-box; padding: 0px 5px;">34</li><li style="box-sizing: border-box; padding: 0px 5px;">35</li><li style="box-sizing: border-box; padding: 0px 5px;">36</li><li style="box-sizing: border-box; padding: 0px 5px;">37</li><li style="box-sizing: border-box; padding: 0px 5px;">38</li><li style="box-sizing: border-box; padding: 0px 5px;">39</li><li style="box-sizing: border-box; padding: 0px 5px;">40</li><li style="box-sizing: border-box; padding: 0px 5px;">41</li><li style="box-sizing: border-box; padding: 0px 5px;">42</li><li style="box-sizing: border-box; padding: 0px 5px;">43</li><li style="box-sizing: border-box; padding: 0px 5px;">44</li><li style="box-sizing: border-box; padding: 0px 5px;">45&l