我所知的opencv中分割函数:watershed(只是看看效果,不能返回每类pixel类属),cvsegmentImage,cvPyrSegmentation(返回pixel类属)
金字塔分割原理篇在这里,本文只提供代码。
Segment函数:
- #include<cv.h>
- #include <cvaux.h>
- #include <opencvcxcore.hpp>
- #include <opencv.hpp>
- #include <nonfree.hpp>
- #include <core/core.hpp>
- #include <imgproc/imgproc.hpp>
- #include <imgproc/imgproc_c.h>
- #include <vector>
- #include <map>
- #include <highgui.h>
- 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 "<<ncomp<<" components."<<endl;
- //////////////////////////Get Components/////////////////////////////
- map<int, int>mapping;//map color value to component id (classify)
- for (int i = 0; i<ncomp; i++)//foreach connection component
- {
- CvConnectedComp* cc = (CvConnectedComp*) cvGetSeqElem(components,i);
- 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));
- mapping.insert(pair<int,int>(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 "<<ncomp<<" components."<<endl;
- //////////////////////////Get Components/////////////////////////////
- map<int, int>mapping;//map color value to component id (classify)
- for (int i = 0; i<ncomp; i++)//foreach connection component
- {
- CvConnectedComp* cc = (CvConnectedComp*) cvGetSeqElem(components,i);
- 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));
- mapping.insert(pair<int,int>(cc->value.val[0],i));
- }
- cvShowImage("segementation",dst);
- waitKey();
- cvReleaseMemStorage(&storage);
- }
- // from: http://blog.csdn.net/abcjennifer/article/details/18215953