opencv 金字塔图像分割

我所知的opencv中分割函数:watershed(只是看看效果,不能返回每类pixel类属),cvsegmentImage,cvPyrSegmentation(返回pixel类属)

金字塔分割原理篇在这里,本文只提供代码。

Segment函数:


#include
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
using namespace cv;
void Segment(IplImage* I)
{

	parameters initialization//
	int mode = CV_RETR_LIST;
	int level = 2, block_size = 1000;
	CvMemStorage* storage = cvCreateMemStorage(block_size);
	CvSeq* components;
	double threshold1 = 100,threshold2 = 50;

	IplImage *src = cvCreateImage(cvGetSize(I),IPL_DEPTH_8U,1);
	cvCvtColor(I,src,CV_RGB2GRAY);
	IplImage* dst = cvCreateImage(cvGetSize(I),IPL_DEPTH_8U,1);

	src->width = dst->width = (I->width & -(1 << level));
	src->height = dst->height = (I->height & -(1 << level));

	//Segment/
	cvPyrSegmentation(src,dst,storage,&components,level,threshold1,threshold2);
	int ncomp = components->total;
	cout<<"segemented into "<mapping;//map color value to component id (classify)
	for (int i = 0; irect.x,cc->rect.y),cvPoint(cc->rect.x+cc->rect.width,cc->rect.y+cc->rect.height),cvScalar(255,255,0));
		mapping.insert(pair(cc->value.val[0],i));
	}
	cvShowImage("segementation",dst);
	waitKey();
	cvReleaseMemStorage(&storage);
}


彩色图分割版本:

void Segment(IplImage* I)
{
	parameters initialization//
	int mode = CV_RETR_LIST;
	int level = 2, block_size = 1000;
	CvMemStorage* storage = cvCreateMemStorage(block_size);
	CvSeq* components;
	double threshold1 = 10,threshold2 = 50;

	IplImage *src = cvCreateImage(cvGetSize(I),IPL_DEPTH_8U,3);
	src = cvCloneImage(I);
	IplImage* dst = cvCreateImage(cvGetSize(I),IPL_DEPTH_8U,3);

	src->width = dst->width = (I->width & -(1 << level));
	src->height = dst->height = (I->height & -(1 << level));

	//Segment/
	cvPyrSegmentation(src,dst,storage,&components,level,threshold1,threshold2);
	int ncomp = components->total;
	cout<<"segemented into "<mapping;//map color value to component id (classify)
	for (int i = 0; irect.x,cc->rect.y),cvPoint(cc->rect.x+cc->rect.width,cc->rect.y+cc->rect.height),cvScalar(255,255,0));
		mapping.insert(pair(cc->value.val[0],i));
	}
	cvShowImage("segementation",dst);
	waitKey();
	cvReleaseMemStorage(&storage);
}




你可能感兴趣的:(Computer,Vision,C/C++)