opencv 金字塔图像分割

我所知的opencv中分割函数：watershed（只是看看效果，不能返回每类pixel类属），cvsegmentImage，cvPyrSegmentation（返回pixel类属）

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

Segment函数：

1. #include<cv.h>
2. #include <cvaux.h>
3. #include <opencv\cxcore.hpp>
4. #include <opencv.hpp>
5. #include <nonfree.hpp>
6. #include <core/core.hpp>
7. #include <imgproc/imgproc.hpp>
8. #include <imgproc/imgproc_c.h>
9. #include <vector>
10. #include <map>
11. #include <highgui.h>
12. using namespace std;
13. using namespace cv;
14. void Segment(IplImage* I)
15. {
16. ////////////////////parameters initialization//////////////////////
17. int mode = CV_RETR_LIST;
18. int level = 2, block_size = 1000;
19. CvMemStorage* storage = cvCreateMemStorage(block_size);
20. CvSeq* components;
21. double threshold1 = 100,threshold2 = 50;
22. IplImage *src = cvCreateImage(cvGetSize(I),IPL_DEPTH_8U,1);
23. cvCvtColor(I,src,CV_RGB2GRAY);
24. IplImage* dst = cvCreateImage(cvGetSize(I),IPL_DEPTH_8U,1);
25. src->width = dst->width = (I->width & -(1 << level));
26. src->height = dst->height = (I->height & -(1 << level));
27. //////////////////////Segment/////////////////////////
28. cvPyrSegmentation(src,dst,storage,&components,level,threshold1,threshold2);
29. int ncomp = components->total;
30. cout<<"segemented into "<<ncomp<<" components."<<endl;
31. //////////////////////////Get Components/////////////////////////////
32. map<int, int>mapping;//map color value to component id (classify)
33. for (int i = 0; i<ncomp; i++)//foreach connection component
34. {
35. CvConnectedComp* cc = (CvConnectedComp*) cvGetSeqElem(components,i);
36. cvDrawRect(dst,cvPoint(cc->rect.x,cc->rect.y),cvPoint(cc->rect.x+cc->rect.width,cc->rect.y+cc->rect.height),cvScalar(255,255,0));
37. mapping.insert(pair<int,int>(cc->value.val[0],i));
38. }
39. cvShowImage("segementation",dst);
40. waitKey();
41. cvReleaseMemStorage(&storage);
42. }

彩色图分割版本：

1. void Segment(IplImage* I)
2. {
3. ////////////////////parameters initialization//////////////////////
4. int mode = CV_RETR_LIST;
5. int level = 2, block_size = 1000;
6. CvMemStorage* storage = cvCreateMemStorage(block_size);
7. CvSeq* components;
8. double threshold1 = 10,threshold2 = 50;
9. IplImage *src = cvCreateImage(cvGetSize(I),IPL_DEPTH_8U,3);
10. src = cvCloneImage(I);
11. IplImage* dst = cvCreateImage(cvGetSize(I),IPL_DEPTH_8U,3);
12. src->width = dst->width = (I->width & -(1 << level));
13. src->height = dst->height = (I->height & -(1 << level));
14. //////////////////////Segment/////////////////////////
15. cvPyrSegmentation(src,dst,storage,&components,level,threshold1,threshold2);
16. int ncomp = components->total;
17. cout<<"segemented into "<<ncomp<<" components."<<endl;
18. //////////////////////////Get Components/////////////////////////////
19. map<int, int>mapping;//map color value to component id (classify)
20. for (int i = 0; i<ncomp; i++)//foreach connection component
21. {
22. CvConnectedComp* cc = (CvConnectedComp*) cvGetSeqElem(components,i);
23. cvDrawRect(dst,cvPoint(cc->rect.x,cc->rect.y),cvPoint(cc->rect.x+cc->rect.width,cc->rect.y+cc->rect.height),cvScalar(255,255,0));
24. mapping.insert(pair<int,int>(cc->value.val[0],i));
25. }
26. cvShowImage("segementation",dst);
27. waitKey();
28. cvReleaseMemStorage(&storage);
29. }
30. // from: http://blog.csdn.net/abcjennifer/article/details/18215953