opencv 金字塔图像分割

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

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

Segment函数：

 

 

[cpp]  view plain  copy

 

 

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.   
17.     parameters initialization//  
18.     int mode = CV_RETR_LIST;  
19.     int level = 2, block_size = 1000;  
20.     CvMemStorage* storage = cvCreateMemStorage(block_size);  
21.     CvSeq* components;  
22.     double threshold1 = 100,threshold2 = 50;  
23.   
24.     IplImage *src = cvCreateImage(cvGetSize(I),IPL_DEPTH_8U,1);  
25.     cvCvtColor(I,src,CV_RGB2GRAY);  
26.     IplImage* dst = cvCreateImage(cvGetSize(I),IPL_DEPTH_8U,1);  
27.   
28.     src->width = dst->width = (I->width & -(1 << level));  
29.     src->height = dst->height = (I->height & -(1 << level));  
30.   
31.     //Segment/  
32.     cvPyrSegmentation(src,dst,storage,&components,level,threshold1,threshold2);  
33.     int ncomp = components->total;  
34.     cout<<"segemented into "<<ncomp<<" components."<<endl;  
35.   
36.     //Get Components/  
37.     map<int, int>mapping;//map color value to component id (classify)  
38.     for (int i = 0; i<ncomp; i++)//foreach connection component  
39.     {  
40.         CvConnectedComp* cc = (CvConnectedComp*) cvGetSeqElem(components,i);  
41.         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));  
42.         mapping.insert(pair<int,int>(cc->value.val[0],i));  
43.     }  
44.     cvShowImage("segementation",dst);  
45.     waitKey();  
46.     cvReleaseMemStorage(&storage);  
47. }  

 

彩色图分割版本：

 

[cpp]  view plain  copy

 

 

1. 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.   
   10.     IplImage *src = cvCreateImage(cvGetSize(I),IPL_DEPTH_8U,3);  
   11.     src = cvCloneImage(I);  
   12.     IplImage* dst = cvCreateImage(cvGetSize(I),IPL_DEPTH_8U,3);  
   13.   
   14.     src->width = dst->width = (I->width & -(1 << level));  
   15.     src->height = dst->height = (I->height & -(1 << level));  
   16.   
   17.     //Segment/  
   18.     cvPyrSegmentation(src,dst,storage,&components,level,threshold1,threshold2);  
   19.     int ncomp = components->total;  
   20.     cout<<"segemented into "<<ncomp<<" components."<<endl;  
   21.   
   22.     //Get Components/  
   23.     map<int, int>mapping;//map color value to component id (classify)  
   24.     for (int i = 0; i<ncomp; i++)//foreach connection component  
   25.     {  
   26.         CvConnectedComp* cc = (CvConnectedComp*) cvGetSeqElem(components,i);  
   27.         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));  
   28.         mapping.insert(pair<int,int>(cc->value.val[0],i));  
   29.     }  
   30.     cvShowImage("segementation",dst);  
   31.     waitKey();  
   32.     cvReleaseMemStorage(&storage);  
   33. }  
   34. // from: http://blog.csdn.net/abcjennifer/article/details/18215953

转载于:https://www.cnblogs.com/GarfieldEr007/p/5369631.html