分水嶺算法code

1、分水嶺算法code /Image Watershed Segmentation /This is the implementation of the algorithm based on immersion model. / = = / = Module: Watershed.cpp / = - = = / = Version 01 Date: 04/21/2003 / = - = = / = = / = Written by F / = e-mail: gong200 / = = / Permission to use, copy, or modify this

2、software and its documentation / for educational and research purposes only is hereby granted without / fee, provided that this copyright notice appear on all copies and / related documentation. For any other uses of this software, in original / or modified form, including but not limited to distrib

3、ution in whole / or in part, specific prior permission must be obtained from / the author(s). / / THE SOFTWARE IS PROVIDED "AS-IS" AND WITHOUT WARRANTY OF ANY KIND, / EXPRESS, IMPLIED OR OTHERWISE, INCLUDING WITHOUT LIMITATION, ANY / WARRANTY OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR

4、PURPOSE. / / IN NO EVENT SHALL RUTGERS UNIVERSITY BE LIABLE FOR ANY SPECIAL, / INCIDENTAL, INDIRECT OR CONSEQUENTIAL DAMAGES OF ANY KIND, OR ANY / DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, / WHETHER OR NOT ADVISED OF THE POSSIBILITY OF DAMAGE, AND ON ANY / THEORY OF LIABILITY,

5、ARISING OUT OF OR IN CONNECTION WITH THE USE / OR PERFORMANCE OF THIS SOFTWARE. / =#include <queue> #include <vector> #include <windows.h> /*= 函數(shù)名 ： Watershed 功能 ： 用標(biāo)記-分水嶺算法對(duì)輸入圖像進(jìn)行分割 算法實(shí)現(xiàn) ： 無(wú) 輸入?yún)?shù)說(shuō)明 ： OriginalImage -輸入圖像（灰度圖，0255） SeedImage -標(biāo)記圖像（二值圖，0-非 標(biāo) 記，1-標(biāo)記） LabelImage -輸出圖像（

6、1-第一個(gè)分割 區(qū) 域，2-第二個(gè)分割區(qū)域，.） row -圖像行數(shù) col -圖像列數(shù) 返回值說(shuō)明 ： 無(wú) =*/ void Watershed(const int *OriginalImage, char* SeedImage, int *LabelImage , int row, int col) using namespace std; /標(biāo)記區(qū)域標(biāo)識(shí)號(hào)，從1開(kāi)始 int Num=0; int i,j; /保存每個(gè)隊(duì)列種子個(gè)數(shù)的數(shù)組 vector<int*> SeedCounts; /臨時(shí)種子隊(duì)列 queue<POINT> que; /保存所有標(biāo)記區(qū)域種子隊(duì)列的

7、數(shù)組 vector<queue<POINT>* > qu; int* array; queue<POINT> *uu; POINT temp; for(i=0;i<row;i+) for(j=0;j<col;j+) LabelImageij=0; int m,n,k=0; int up,down,right,left,upleft,upright,downleft,downright; /預(yù)處理,提取區(qū)分每個(gè)標(biāo)記區(qū)域，并初始化每個(gè)標(biāo)記的種子隊(duì)列 /種子是指標(biāo)記區(qū)域邊緣的點(diǎn)，他們可以在水位上升時(shí)向外淹沒(méi)（或者說(shuō)生長(zhǎng)） for(i=0;i<ro

8、w;i+) for(j=0;j<col;j+) /如果找到一個(gè)標(biāo)記區(qū)域 if(SeedImageij=1) /區(qū)域的標(biāo)識(shí)號(hào)加一 Num+; /分配數(shù)組并初始化為零 array=new int256; ZeroMemory(array,256*sizeof(int); / SeedCounts.push_back(array); /分配本標(biāo)記的優(yōu)先隊(duì)列 uu=new queue<POINT>256; /加入到隊(duì)列數(shù)組中 qu.push_back(uu); /當(dāng)前點(diǎn)放入本標(biāo)記區(qū)域的臨時(shí)種子隊(duì)列中 temp.x=i; temp.y=j; que.push(temp); /當(dāng)前點(diǎn)標(biāo)記

9、為已處理 LabelImageij=Num; SeedImageij=127; /讓種子隊(duì)列中的種子進(jìn)行生長(zhǎng)直到所有的種子都生 長(zhǎng) 完畢while(!que.empty() up=down=right=left=0; upleft=upright=downleft=downright=0; /隊(duì)列中取出一個(gè)種子 temp=que.front(); m=temp.x; n=temp.y; que.pop(); if(m>0) /上方若為可生長(zhǎng)點(diǎn)則加為新種子 if(SeedImagem-1n=1) temp.x=m-1; temp.y=n; que.push(temp); /新種子點(diǎn)標(biāo)記為已

10、淹沒(méi) 區(qū) 域 LabelImagem-1n=Num ; SeedImagem-1n=127; else/否則上方為不可生長(zhǎng) up=1; if(m>0&&n>0) if(SeedImagem-1n-1=1)/左 上 方若為可生長(zhǎng)點(diǎn)則加為新種子 temp.x=m-1; temp.y=n-1; que.push(temp); /新種子點(diǎn)標(biāo)記為已淹沒(méi) 區(qū) 域 LabelImagem-1n-1=Num; SeedImagem-1n-1=12 7; else/否則左上方為不可生長(zhǎng) upleft=1; if(m<row-1) if(SeedImagem+1n=1)/下方 若

11、 為可生長(zhǎng)點(diǎn)則加為新種子 temp.x=m+1; temp.y=n; que.push(temp); /新種子點(diǎn)標(biāo)記為已淹沒(méi) 區(qū) 域 LabelImagem+1n=Num ; SeedImagem+1n=127; else/否則下方為不可生長(zhǎng) down=1; if(m<(row-1)&&n<(col-1) if(SeedImagem+1n+1=1)/下 方 若為可生長(zhǎng)點(diǎn)則加為新種子 temp.x=m+1; temp.y=n+1; que.push(temp); /新種子點(diǎn)標(biāo)記為已淹沒(méi) 區(qū) 域 LabelImagem+1n+1=Num; SeedImagem+1n+1

12、=12 7; else/否則下方為不可生長(zhǎng) downright=1; if(n<col-1) if(SeedImagemn+1=1)/右方 若 為可生長(zhǎng)點(diǎn)則加為新種子 temp.x=m; temp.y=n+1; que.push(temp); /新種子點(diǎn)標(biāo)記為已淹沒(méi) 區(qū) 域 LabelImagemn+1=Num ; SeedImagemn+1=127; else/否則右方為不可生長(zhǎng) right=1; if(m>0&&n<(col-1) if(SeedImagem-1n+1=1)/右 上 方若為可生長(zhǎng)點(diǎn)則加為新種子 temp.x=m-1; temp.y=n+1;

13、 que.push(temp); /新種子點(diǎn)標(biāo)記為已淹沒(méi) 區(qū) 域LabelImagem-1n+1=Num; SeedImagem-1n+1=12 7; else/否則右上方為不可生長(zhǎng) upright=1; if(n>0) if(SeedImagemn-1=1)/左方 若 為可生長(zhǎng)點(diǎn)則加為新種子 temp.x=m; temp.y=n-1; que.push(temp); /新種子點(diǎn)標(biāo)記為已淹沒(méi) 區(qū) 域 LabelImagemn-1=Num ; SeedImagemn-1=127; else/否則左方為不可生長(zhǎng) left=1; if(m<(row-1)&&n>0)

14、 if(SeedImagem+1n-1=1)/左 下 方若為可生長(zhǎng)點(diǎn)則加為新種子 temp.x=m+1; temp.y=n-1; que.push(temp); /新種子點(diǎn)標(biāo)記為已淹沒(méi) 區(qū) 域 LabelImagem+1n-1=Num; SeedImagem+1n-1=12 7; else/否則左方為不可生長(zhǎng) downleft=1; /上下左右只要有一點(diǎn)不可生長(zhǎng)，那么本點(diǎn) 為 初始種子隊(duì)列中的一個(gè) if(up|down|right|left| upleft|downleft|upright|downright) temp.x=m; temp.y=n; quNum-1OriginalImagem

15、n.push(temp); SeedCountsNum-1OriginalImagemn+; /while結(jié)束 bool actives;/在某一水位處，所有標(biāo)記的種子生長(zhǎng)完的標(biāo)志 int WaterLevel; /淹沒(méi)過(guò)程開(kāi)始，水位從零開(kāi)始上升 for(WaterLevel=0;WaterLevel<256;WaterLevel+) actives=true; while(actives) actives=false; /依次處理每個(gè)標(biāo)記區(qū)域 for(i=0;i<Num;i+) if(!quiWaterLevel.empty() actives=true; while(SeedC

16、ountsiWaterLevel>0) SeedCountsiWaterLevel-; temp=quiWaterLevel.front(); quiWaterLevel.pop(); m = temp.x; n = temp.y;/當(dāng)前種子的坐標(biāo) if(m>0) if(!LabelImagem-1n)/上方若 未處理 temp.x=m-1; temp.y=n; LabelImagem-1n=i+1;/上方點(diǎn)標(biāo)記為已淹沒(méi)區(qū)域 if(OriginalImagem-1n<=WaterLevel)/上方若為可生長(zhǎng)點(diǎn)則加入當(dāng)前隊(duì)列 quiWaterLevel.push(temp);

17、else/否則加入 OriginalImagem-1n級(jí)隊(duì)列 quiOriginalImagem-1n.push(temp); SeedCountsiOriginalImagem-1n+; if(m<row-1) if(!LabelImagem+1n)/下方若 未處理 temp.x=m+1; temp.y=n; LabelImagem+1n=i+1;/下方點(diǎn)標(biāo)記為已淹沒(méi)區(qū)域 if(OriginalImagem+1n<=WaterLevel)/下方若為可生長(zhǎng)點(diǎn)則加入當(dāng)前隊(duì)列 quiWaterLevel.push(temp); else/否則加入 OriginalImagem+1n級(jí)隊(duì)

18、列 quiOriginalImagem+1n.push(temp); SeedCountsiOriginalImagem+1n+; if(n<col-1) if(!LabelImagemn+1)/右邊若 未處理 temp.x=m; temp.y=n+1; LabelImagemn+1=i+1;/右邊點(diǎn)標(biāo)記為已淹沒(méi)區(qū)域 if(OriginalImagemn+1<=WaterLevel)/右邊若為可生長(zhǎng)點(diǎn)則加入當(dāng)前隊(duì)列 quiWaterLevel.push(temp); else/否則加入 OriginalImagemn+1級(jí)隊(duì)列 quiOriginalImagemn+1.push(temp); SeedCountsiOriginalImagemn+1+; if(n>0) if(!LabelImagemn-1)/左邊若未處理 temp.x=m; temp.y=n-1; LabelImagemn-1=i+1;/左邊點(diǎn)標(biāo)記為已淹沒(méi)區(qū)域 if(OriginalImagemn-1<=WaterLevel)/左邊若為可生長(zhǎng)點(diǎn)則加入當(dāng)前隊(duì)列 quiWaterLevel.push(temp); else/否則加入 OriginalImagemn-1級(jí)隊(duì)列 quiO