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. }

opencv 金字塔图像分割

彩色图分割版本:

    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
上一篇:ps-如何去水印


下一篇:C++程序在Windows平台上各种定位内存泄漏的方法,并对比了它们的优缺点