图像中区域生长算法的详解和实现

#include "stdafx.h"
#include <opencv2/opencv.hpp>
#include <vector>
#include <stack>

using namespace std;
using namespace cv;

void regiongrowth(IplImage* &inputimg,IplImage* &outputimg,CvPoint pt,int th)
{

CvPoint ptGrowing;//待生长点的位置
int nGrowLabel=0;//是否被标记
int nScrValue=0;//生长起始点的灰度值
int nCurValue=0;//当前生长点的灰度值

IplImage* Dstimg=cvCreateImage(cvGetSize(inputimg),8,1);
if (Dstimg==NULL)
{
printf("image read error too");
return;
}
cvZero(Dstimg);

int DIR[8][2]={{-1,-1}, {0,-1}, {1,-1}, {1,0}, {1,1}, {0,1}, {-1,1}, {-1,0}};

vector<CvPoint>vcGrowpt;//生长点的堆栈
vcGrowpt.push_back(pt);//将初始生长点压入堆栈

//标记初始生长点
Dstimg->imageData[pt.x+pt.y*Dstimg->widthStep]=255;
nScrValue=inputimg->imageData[pt.x+pt.y*inputimg->widthStep];

while(!vcGrowpt.empty())
{
CvPoint curpt=vcGrowpt.back();//在堆栈中取出一个生长点
vcGrowpt.pop_back();

for(int i=0;i<8;i++)
{
ptGrowing.x=curpt.x+DIR[i][0];
ptGrowing.y=curpt.y+DIR[i][1];
//检查边缘点
if(ptGrowing.x<0||ptGrowing.y<0||(ptGrowing.x>inputimg->width-1)||(ptGrowing.y>inputimg->height-1))
continue;
nGrowLabel=Dstimg->imageData[ptGrowing.x+ptGrowing.y*Dstimg->widthStep];
if (nGrowLabel==0)//表示还未标记过
{
nCurValue=inputimg->imageData[ptGrowing.x+ptGrowing.y*inputimg->widthStep];
if(abs(nCurValue-nScrValue)<=th)
{
Dstimg->imageData[ptGrowing.x+ptGrowing.y*Dstimg->widthStep]=255;
vcGrowpt.push_back(ptGrowing);

}
}

}

}
cvCopy(Dstimg,outputimg);
cvReleaseImage(&Dstimg);

}

int _tmain(int argc, _TCHAR* argv[])
{

IplImage* sourceimage=cvLoadImage("H:\\programm\\test.jpg",0);

if (sourceimage==NULL)
{
printf("image read error");
}

IplImage* outputimg=cvCreateImage(cvGetSize(sourceimage),8,1);
IplImage* resultimg=cvCreateImage(cvGetSize(sourceimage),8,1);
cvZero(resultimg);
cvCopy(sourceimage,outputimg);
regiongrowth(sourceimage,outputimg,cvPoint(392,282),12);
cvAdd(outputimg,resultimg,resultimg);

regiongrowth(sourceimage,outputimg,cvPoint(587,195),8);
cvAdd(outputimg,resultimg,resultimg);

regiongrowth(sourceimage,outputimg,cvPoint(707,356),8);
cvAdd(outputimg,resultimg,resultimg);

regiongrowth(sourceimage,outputimg,cvPoint(546,549),10);
cvAdd(outputimg,resultimg,resultimg);

regiongrowth(sourceimage,outputimg,cvPoint(310,435),10);
cvAdd(outputimg,resultimg,resultimg);

cvNamedWindow("source",0);
cvShowImage("source",sourceimage);

cvNamedWindow("result",0);
cvShowImage("result",resultimg);
cvWaitKey(0);

cvReleaseImage(&sourceimage);
cvReleaseImage(&outputimg);
cvReleaseImage(&resultimg);
system("pause");
return 0;
}