我所知的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<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<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);}
欢迎加入我爱机器学习QQ14群:336582044
微信扫一扫,关注我爱机器学习公众号