Meanshift,聚類演算法

編著人發表於2017-07-31

Meanshift,聚類演算法

http://www.cnblogs.com/liqizhou/archive/2012/05/12/2497220.html

記得剛讀研究生的時候,學習的第一個演算法就是meanshift演算法,所以一直記憶猶新,今天和大家分享一下Meanshift演算法,如有錯誤,請線上交流。

Mean Shift演算法,一般是指一個迭代的步驟,即先算出當前點的偏移均值,移動該點到其偏移均值,然後以此為新的起始點,繼續移動,直到滿足一定的條件結束.

 1. Meanshift推導

給定d維空間Rd的n個樣本點 ,i=1,…,n,在空間中任選一點x,那麼Mean Shift向量的基本形式定義為:                             

 Sk是一個半徑為h的高維球區域,滿足以下關係的y點的集合,

k表示在這n個樣本點xi中,有k個點落入Sk區域中.

以上是官方的說法,即書上的定義,我的理解就是,在d維空間中,任選一個點,然後以這個點為圓心,h為半徑做一個高維球,因為有d維,d可能大於2,所以是高維球。落在這個球內的所有點和圓心都會產生一個向量,向量是以圓心為起點落在球內的點位終點。然後把這些向量都相加。相加的結果就是Meanshift向量。

如圖所以。其中黃色箭頭就是Mh(meanshift向量)。

再以meanshift向量的終點為圓心,再做一個高維的球。如下圖所以,重複以上步驟,就可得到一個meanshift向量。如此重複下去,meanshift演算法可以收斂到概率密度最大得地方。也就是最稠密的地方。

最終的結果如下:

Meanshift推導:

 把基本的meanshift向量加入核函式,核函式的性質在這篇部落格介紹:http://www.cnblogs.com/liqizhou/archive/2012/05/11/2495788.html

那麼,meanshift演算法變形為

                                                         (1)

解釋一下K()核函式,h為半徑,Ck,d/nhd  為單位密度,要使得上式f得到最大,最容易想到的就是對上式進行求導,的確meanshift就是對上式進行求導.

(2)             

令:

K(x)叫做g(x)的影子核,名字聽上去聽深奧的,也就是求導的負方向,那麼上式可以表示

對於上式,如果才用高斯核,那麼,第一項就等於fh,k

第二項就相當於一個meanshift向量的式子:

 那麼(2)就可以表示為

下圖分析的構成,如圖所以,可以很清晰的表達其構成。

要使得=0,當且僅當=0,可以得出新的圓心座標:

                          (3) 

 

上面介紹了meanshift的流程,但是比較散,下面具體給出它的演算法流程。

  1. 選擇空間中x為圓心,以h為半徑為半徑,做一個高維球,落在所有球內的所有點xi
  2. 計算,如果<ε(人工設定),推出程式。如果>ε, 則利用(3)計算x,返回1.

 

2.meanshift在影象上的聚類:

真正大牛的人就能創造演算法,例如像meanshift,em這個樣的演算法,這樣的創新才能推動整個學科的發展。還有的人就是把演算法運用的實際的運用中,推動整個工業進步,也就是技術的進步。下面介紹meashift演算法怎樣運用到影象上的聚類核跟蹤。

一般一個影象就是個矩陣,畫素點均勻的分佈在影象上,就沒有點的稠密性。所以怎樣來定義點的概率密度,這才是最關鍵的。

如果我們就算點x的概率密度,採用的方法如下:以x為圓心,以h為半徑。落在球內的點位xi   定義二個模式規則。

(1)x畫素點的顏色與xi畫素點顏色越相近,我們定義概率密度越高。

(2)離x的位置越近的畫素點xi,定義概率密度越高。

所以定義總的概率密度,是二個規則概率密度乘積的結果,可以(4)表示

(4)

其中:代表空間位置的資訊,離遠點越近,其值就越大,表示顏色資訊,顏色越相似,其值越大。如圖左上角圖片,按照(4)計算的概率密度如圖右上。利用meanshift對其聚類,可得到左下角的圖。

 

----------------------------------------------------------------------------------------------------------------------

meanshift演算法

http://blog.sina.com.cn/s/blog_6fd15d5f01016agj.html

Mean Shift演算法,一般是指一個迭代的步驟,即先算出當前點的偏移均值,移動該點到其偏移均值,然後以此為新的起始點,繼續移動,直到滿足一定的條件結束.

 1. Meanshift推導

給定d維空間Rd的n個樣本點 ,i=1,…,n,在空間中任選一點x,那麼Mean Shift向量的基本形式定義為:                             

 Sk是一個半徑為h的高維球區域,滿足以下關係的y點的集合,

k表示在這n個樣本點xi中,有k個點落入Sk區域中.

以上是官方的說法,即書上的定義,我的理解就是,在d維空間中,任選一個點,然後以這個點為圓心,h為半徑做一個高維球,因為有d維,d可能大於2,所以是高維球。落在這個球內的所有點和圓心都會產生一個向量,向量是以圓心為起點落在球內的點位終點。然後把這些向量都相加。相加的結果就是Meanshift向量。

如圖所以。其中黃色箭頭就是Mh(meanshift向量)。

再以meanshift向量的終點為圓心,再做一個高維的球。如下圖所以,重複以上步驟,就可得到一個meanshift向量。如此重複下去,meanshift演算法可以收斂到概率密度最大得地方。也就是最稠密的地方。

最終的結果如下:

Meanshift推導:

 把基本的meanshift向量加入核函式,核函式的性質在這篇部落格介紹:http://www.cnblogs.com/liqizhou/archive/2012/05/11/2495788.html

那麼,meanshift演算法變形為

                                                         (1)

解釋一下K()核函式,h為半徑,Ck,d/nh 為單位密度,要使得上式f得到最大,最容易想到的就是對上式進行求導,的確meanshift就是對上式進行求導.

(2)             

令:

K(x)叫做g(x)的影子核,名字聽上去聽深奧的,也就是求導的負方向,那麼上式可以表示

對於上式,如果才用高斯核,那麼,第一項就等於fh,k

第二項就相當於一個meanshift向量的式子:

 那麼(2)就可以表示為

下圖分析的構成,如圖所以,可以很清晰的表達其構成。

要使得=0,當且僅當=0,可以得出新的圓心座標:

                          (3) 

 

上面介紹了meanshift的流程,但是比較散,下面具體給出它的演算法流程。

  1. 選擇空間中x為圓心,以h為半徑為半徑,做一個高維球,落在所有球內的所有點xi
  2. 計算,如果<ε(人工設定),推出程式。如果>ε, 則利用(3)計算x,返回1.

 

2.meanshift在影象上的聚類:

真正大牛的人就能創造演算法,例如像meanshift,em這個樣的演算法,這樣的創新才能推動整個學科的發展。還有的人就是把演算法運用的實際的運用中,推動整個工業進步,也就是技術的進步。下面介紹meashift演算法怎樣運用到影象上的聚類核跟蹤。

一般一個影象就是個矩陣,畫素點均勻的分佈在影象上,就沒有點的稠密性。所以怎樣來定義點的概率密度,這才是最關鍵的。

如果我們就算點x的概率密度,採用的方法如下:以x為圓心,以h為半徑。落在球內的點位xi   定義二個模式規則。

(1)x畫素點的顏色與xi畫素點顏色越相近,我們定義概率密度越高。

(2)離x的位置越近的畫素點xi,定義概率密度越高。

所以定義總的概率密度,是二個規則概率密度乘積的結果,可以(4)表示

(4)

其中:代表空間位置的資訊,離遠點越近,其值就越大,表示顏色資訊,顏色越相似,其值越大。如圖左上角圖片,按照(4)計算的概率密度如圖右上。利用meanshift對其聚類,可得到左下角的圖。

 

http://www.cnblogs.com/liqizhou/archive/2012/05/12/2497220.html

Matlab中meanshift演算法

mean-shift 的特點是把支撐空間和特徵空間在資料密度的框架下綜合了起來。對影象來講,支撐空間就是畫素點的座標,特徵空間就是對應畫素點的灰度或者RGB三分量。將這兩個空間綜合後,一個資料點就是一個5維的向量:[x,y,r,g,b]。

這在觀念上看似簡單,實質是一個飛躍,它是mean-shift方法的基點。

mean-shift方法很寶貴的一個特點就是在這樣迭代計算的框架下,求得的mean-shift向量必收斂於資料密度的區域性最大點。可以細看[ComaniciuMeer2002]的文章。

寫了點程式,可以對影象做簡單的mean-shift filtering,供參考:

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
function [DRGB, DSD, MSSD] = MScut(sMode, RGB_raw, hs, hf, m );
% designed for segmenting a colour image using mean-shift [ComaniciuMeer 2002]
% image must be color
% procedure in mean-shift
% 1. combine support space and feature space to make a mean-shift space
%    based data description
% 2. for every mean-shift space data
% 3.   do mean-shift filtering
%      until convergence
% 4. end
% 5. find the converged mean-shift space data that you are interested in
%    and label it
% 6. repeat the above steps
%
% a     -- data in support space
% b     -- data in feature space
% x     -- data in mean-shift space
% f(.)  -- data density function
% k(.)  -- profile function (implicit)
% g(.)  -- profile function (explicit)
% m     -- mean shift vector
% hs    -- bandwidth in support space
% hf    -- bandwidth in feature space
% M     -- threshold to make a distinct cluster
%% enter $hs$, $hf$, $m$ if necessary
if ~exist('hs')
    hs = input('please enter spatial bandwidth (hs):n');
end
if ~exist('hf')
    hf = input('please enter feature bandwidth (hf):n');
end
if ~exist('m')
    m = input('please enter minimum cluster size (m):n');
end
switch upper(sMode)
    case 'RGB'
        RGB = double( RGB_raw );
    case 'gray'
        error('FCMcut must use colored image to do segmentation!')
end
sz = size(RGB);
mTCUT = Tcut( RGB(:,:,1) ); % trivial segmentation

%% project data into mean-shift space to make $MSSD$ (mean-shift space data)
mT = repmat([1:sz(1)]', 1, sz(2));
vX = mT(1:end)';             % row
mT = repmat([1:sz(2)], sz(1), 1);
vY = mT(1:end)';  % column
mT = RGB(:,:,1);
vR = mT(1:end)'; % red
mT = RGB(:,:,2);
vG = mT(1:end)'; % green
mT = RGB(:,:,3);
vB = mT(1:end)'; % blue
MSSD = [vX, vY, vR, vG, vB];
%% make $g$ - explicit profile function
disp('Using flat kernel: Epanechnikov kernel...')
g_s = ones(2*hs+1, 2); % 's' for support space
g_f = ones(2*hf+1, 3); % 'f' for feature space
%% main part $$
nIteration = 4;
nData   = length(MSSD); % total number of data
DSD     = MSSD*0; % 'DSD' for destination space data
for k = 1:nData
    %
    tMSSD = MSSD(k,:); % 't' for temp
    for l = 1:nIteration
        %
        mT = abs( MSSD - repmat(tMSSD, nData, 1));
        vT = logical( (mT(:,1)<=hs).*(mT(:,2)<=hs).*(mT(:,3)<=hf).*(mT(:,4)<=hf).*(mT(:,5)<=hf) );
        v  = MSSD(vT,:);
        % update $tMSSD$
        tMSSD = mean( v, 1 );
        if nIteration == l
            DSD(k,:) = tMSSD;
        end
    end
end
% show result
DRGB = RGB * 0;
DRGB(:,:,1) = reshape(DSD(:,3), sz(1), sz(2)); % red
DRGB(:,:,2) = reshape(DSD(:,4), sz(1), sz(2)); % red
DRGB(:,:,3) = reshape(DSD(:,5), sz(1), sz(2)); % red

figure, imshow(uint8(DRGB), [])

-------------------------------------------------------------------------------------------------------------------------

matlab練習程式(meanshift影象聚類)

http://www.cnblogs.com/tiandsp/archive/2012/11/20/2779601.html

 關於這個meanshift,一來可以用來作為目標跟蹤,二來可以用來進行影象聚類。我這裡只實現了影象聚類,當然,是按自己的理解編寫的程式。至於目標跟蹤將來一定也是要實現的,因為我最初看這個演算法的原因就是想用他來跟蹤目標的。

  meanshift的基本原理我就不介紹了,比起我的介紹,網上有不少牛人們比我解釋的好,最後我會列出我參考的文章。我這裡說一下我是怎麼理解meanshift影象聚類的。這裡的聚類也像過去的濾波一樣,需要一個模板矩陣,不過這個模板不是事先設定好的矩陣,而是在當前處理的畫素周圍提取一個r*r的矩陣,然後把這個矩陣化為一維向量,再對這個向量進行meanshift,最終迭代到的值再賦值給當前處理的畫素。所以可以這樣理解,把影象經過meanshift迭代到相同值的畫素聚為一類。

  我這裡使用的是灰度影象,至於彩色影象,我看到一篇部落格上把rgb域轉換到luv域上再去做處理,這個我就不太清楚了,不過我看他的程式碼其中有一部分很像均值濾波。雖然我沒有和他用一樣的方法,不過他的程式碼也可以參考一下。傳送門在此

  下面是程式碼(這都是我自己的理解,不能保證都正確,不過至少可以為你的編碼提供一些思路):

main.m

複製程式碼
clear all;
close all;
clc;

r=2;        %濾波半徑
img=imread('lena.jpg');
imshow(img);
img=double(img);
[m n]=size(img);

imgn=zeros(m+2*r+1,n+2*r+1);

imgn(r+1:m+r,r+1:n+r)=img;
imgn(1:r,r+1:n+r)=img(1:r,1:n); 
imgn(1:m+r,n+r+1:n+2*r+1)=imgn(1:m+r,n:n+r);
imgn(m+r+1:m+2*r+1,r+1:n+2*r+1)=imgn(m:m+r,r+1:n+2*r+1);
imgn(1:m+2*r+1,1:r)=imgn(1:m+2*r+1,r+1:2*r);
imshow(mat2gray(imgn))

for i=1+r:m+r
    for j=1+r:n+r
        ser=imgn(i-r:i+r,j-r:j+r);
        ser=reshape(ser,[1 (2*r+1)^2]);         %將二維模板變為一維
        imgn(i,j)=mean_shift(ser,2*r^2+2*r+1);   %取模板最中間的那個值作為迭代初值
    
    end    
end

figure;
imgn=imgn(r+1:m+r,r+1:n+r);
imshow(mat2gray(imgn));
複製程式碼

meanshift.m

複製程式碼
function   re= mean_shift( ser,p)
    [m n]=size(ser);
    tmp=double(ser);

    pre_w=tmp(p);
    point=p;
    while 1
        ser=tmp-pre_w;

        for i=1:m*n
            if i ~= point
                ser(i)=ser(i)/(i-point);            %i-point是距離,就是各種公式裡的h
            end
        end

        ser=ser.^2;
        K=(1/sqrt(2*pi))*exp(-0.5*ser);         %傳說中的核函式
        w=sum(tmp.*(K))/sum(K);

        if abs(w-pre_w)<0.01
            break;
        end
        pre_w=w; 
    end
 %   tmp1=abs(tmp-w);
 %   [i point]=min(tmp1);
    re=w;
 %   if max(tmp)-w<0.01
 %       point=0;
 %   end
 %   point=w;
end
複製程式碼

處理的效果:

原圖

半徑為2處理的效果

——————————下面是2013.5.30新增————————————

上一部分的meanshift影象聚類還需修改,下面實現最簡單的meanshift演算法,完全按照原理來。

最後的參考文獻都是很好的總結,不過這次我是參考的《影象處理、分析與機器視覺(第3版)》這本書。

下面是通常所見的迭代效果:

程式如下:

複製程式碼
clear all; close all; clc;

%測試資料
mu=[0 0];  %均值
S=[30 0;0 35];  %協方差
data=mvnrnd(mu,S,300);   %產生300個高斯分佈資料
plot(data(:,1),data(:,2),'o');

h=3;    %核的大小
x=[data(1,1) data(1,2)];    %以第一個資料為迭代初值
pre_x=[0 0];

hold on
while norm(pre_x-x)>0.01;
    
    pre_x=x;
    plot(x(1),x(2),'r+');
    u=0;        %分子累加項
    d=0;        %分母累加項
    for i=1:300
        %最關鍵的兩步,均值位移公式實現
        k=norm((x-data(i,:))/h).^2;        
        g=(1/sqrt(2*pi))*exp(-0.5*k);
        
        u=data(i,:)*g+u;
        d=g+d;
    end
    M=u/d;      %迭代後的座標位置
    x=M;
 
end
複製程式碼

 

參考:

1.http://en.wikipedia.org/wiki/Mean-shift wiki百科,介紹的簡介明瞭。

2.http://www.cnblogs.com/liqizhou/archive/2012/05/12/2497220.html 非常詳細的理解。

3.http://emuch.net/bbs/viewthread.php?tid=4626864 小木蟲上一個同學的理解。

4.http://en.wikipedia.org/wiki/Kernel_(statistics) 介紹核函式的。

5.http://wenku.baidu.com/view/11b6a7de6f1aff00bed51eac.html 提出meanshift演算法的論文,雖然我沒怎麼看,不過想對演算法徹底理解的還是看這篇好。

 

------------------------------------------------------------------------------------------------------------

Meanshift影象平滑之opencv實現

http://www.cnblogs.com/easymind223/archive/2012/07/03/2574887.html

一句話一幅圖理解meanshift演算法:

對於集合中的每一個元素,對它執行下面的操作:把該元素移動到它鄰域中所有元素的特徵值的均值的位置,不斷重複直到收斂。

準確的說,不是真正移動元素,而是把該元素與它的收斂位置的元素標記為同一類。對於影象來說,所有元素程矩陣排列,特徵值便是畫素的灰度值。

 

Meanshift的這種思想可以應用於目標跟蹤、影象平滑、邊緣檢測、聚類等,是一種適應性很好的演算法,缺點是速度非常慢。

本文以影象平滑為例對其說明

  從網上找程式碼不如自己動手寫。說明一下兩個引數的含義,hs和hr是核函式的視窗大小,hs是距離核函式,控制子視窗的大小,同時也影響計算速度。hr是顏色核函式,是顏色差值的閾值,maxiter是最大迭代次數。轉載請註明出處,謝謝。本文演算法只是用作實驗之用,沒有進行優化,計算時會有重複計算的地方,速度非常慢,且只支援3通道影象。

複製程式碼
 1 void MyTreasureBox::MeanShiftSmooth(const IplImage* src, IplImage* dst, int hs, int hr, int maxIter)
 2 {
 3     if(!src)return ;
 4 
 5     IplImage* srcLUV = cvCreateImage( cvGetSize( src ), src->depth, src->nChannels );
 6     IplImage* dstLUV = cvCreateImage( cvGetSize( src ), src->depth, src->nChannels );
 7 
 8     cvCvtColor( src, srcLUV, CV_RGB2Luv);
 9     cvCopy( srcLUV, dstLUV );
10 
11     int widthstep = srcLUV->widthStep;
12     int channel = srcLUV->nChannels;
13 
14     for( int y = 0; y<src->height; y++ )
15     {
16         for( int x = 0; x<src->width; x++ )
17         {
18             uchar L = (uchar)srcLUV->imageData[y *widthstep + x *channel];
19             uchar U = (uchar)srcLUV->imageData[y *widthstep + x *channel + 1];
20             uchar V = (uchar)srcLUV->imageData[y *widthstep + x *channel + 2];
21             int xx = x;
22             int yy = y;
23 
24             int nIter = 0;
25             int count, sumL, sumu, sumv, sumx, sumy;
26 
27             while(nIter < maxIter)
28             {
29                 count = 0;
30                 sumL = sumu = sumv = 0;
31                 sumx = sumy = 0;
32 
33                 for( int m = y - hs; m <= y + hs; m++ )
34                 {
35                     for( int n = x - hs; n <= x + hs; n++ )
36                     {
37                         if(m >= 0 && m < src->height && n >= 0 && n < src->width)
38                         {
39                             uchar l = (uchar)srcLUV->imageData[m *widthstep + n *channel];
40                             uchar u = (uchar)srcLUV->imageData[m *widthstep + n *channel + 1];
41                             uchar v = (uchar)srcLUV->imageData[m *widthstep + n *channel + 2];
42 
43                             double dist = sqrt( (double)((L - l)^2 + (U - u)^2 + (V - v)^2) );
44                             if( dist < hr )
45                             {
46                                 count++;
47                                 sumL += l;
48                                 sumu += u;
49                                 sumv += v;
50                                 sumx += n;
51                                 sumy += m;
52                             }
53                         }
54                     }                    
55                 }
56                 if(count == 0)break;
57                 L = sumL / count;
58                 U = sumu / count;
59                 V = sumv / count;
60                 xx = sumx / count;
61                 yy = sumy / count;
62 
63                 nIter++;
64             }
65             dstLUV->imageData[y *widthstep + x *channel] = L;
66             dstLUV->imageData[y *widthstep + x *channel + 1] = U;
67             dstLUV->imageData[y *widthstep + x *channel + 2] = V;
68         }
69     }
70 
71     cvCvtColor( dstLUV, dst, CV_Luv2RGB );
72     cvReleaseImage(&srcLUV);
73     cvReleaseImage(&dstLUV);
74 }
複製程式碼

hs和hr的控制可以參閱下圖

 

相關文章