关于轮廓的各种使用

#include "cv.h"
#include "highgui.h"
#include "cvaux.h"
#include "cxcore.h"
#include <stdio.h>
#include <vector>
using namespace std;
using namespace cv;

#define PI 3.14159f
void DrawBox(CvBox2D box,IplImage* img)
{
 CvPoint2D32f points[4];
 cvBoxPoints(box,points);

 CvPoint pt[4];
 for (int i=0; i<4; i++)
 {
  pt[i].x = (int)points[i].x;
  pt[i].y = (int)points[i].y;
 }
 cvLine( img, pt[0], pt[1],CV_RGB(255,0,0), 2, 8, 0 );
 cvLine( img, pt[1], pt[2],CV_RGB(255,0,0), 2, 8, 0 );
 cvLine( img, pt[2], pt[3],CV_RGB(255,0,0), 2, 8, 0 );
 cvLine( img, pt[3], pt[0],CV_RGB(255,0,0), 2, 8, 0 );
}

void main()
{
 IplImage* img1 = cvLoadImage("5.jpg",1);
 IplImage* img1_gray =cvCreateImage(cvGetSize(img1),8,1);
 cvCvtColor(img1,img1_gray,CV_BGR2GRAY);
 IplImage* img_edge1 = cvCreateImage(cvGetSize(img1),8,1);
 cvThreshold(img1_gray,img_edge1,240,255,CV_THRESH_BINARY);

 printf("图1：\n\n");
 CvMemStorage* storage1 = cvCreateMemStorage();
 CvSeq* contour1 = NULL;
 int Nc1= cvFindContours(img_edge1,storage1,&contour1,sizeof(CvContour),CV_RETR_LIST);//最后一个参数的可变性
 printf("轮廓数：%d\n",Nc1);
 
 //轮廓的矩
 CvMoments *moments1=new CvMoments();
 cvMoments(contour1,moments1,0);
 //Hu矩
 CvHuMoments *huMonents1=new CvHuMoments();
 cvGetHuMoments(moments1,huMonents1);
 printf("遍历三阶矩：\n");
 for (int xOrder = 0; xOrder <= 3; xOrder++)
  for (int yOrder = 0; yOrder <= 3; yOrder++)
  {
   if (xOrder + yOrder <= 3)
   {
    double spatialMoment =cvGetSpatialMoment(moments1,xOrder, yOrder);
    double centralMoment =cvGetCentralMoment(moments1,xOrder, yOrder);
    double normalizedCentralMoment =cvGetNormalizedCentralMoment(moments1,xOrder, yOrder);
    printf("  %d,%d ： 空间距-%.2f, 中心距-%.2f, 归一化中心距-%.2f\n",xOrder,yOrder,spatialMoment,centralMoment,normalizedCentralMoment);
   }
  }
 printf("Hu矩：%f,%f,%f,%f,%f,%f,%f\n",huMonents1->hu1,huMonents1->hu2,huMonents1->hu3,huMonents1->hu3,huMonents1->hu5,huMonents1->hu6,huMonents1->hu7);

 printf("\n图2：\n\n");
 IplImage* img2 = cvLoadImage("6.jpg",1);
 IplImage* img2_gray =cvCreateImage(cvGetSize(img2),8,1);
 cvCvtColor(img2,img2_gray,CV_BGR2GRAY);
 IplImage* img_edge2 = cvCreateImage(cvGetSize(img2),8,1);
 cvThreshold(img2_gray,img_edge2,128,255,CV_THRESH_BINARY);

 CvMemStorage* storage2 = cvCreateMemStorage();
 CvSeq* contour2 = NULL;
 int Nc2 = cvFindContours(img_edge2,storage2,&contour2,sizeof(CvContour),CV_RETR_LIST);
 printf("轮廓数：%d\n\n",Nc2);

 printf("遍历三阶矩：\n");
 CvMoments*moments2=new CvMoments(); 
 cvMoments(contour2,moments2,0); 
 CvHuMoments *huMonents2=new CvHuMoments();
 cvGetHuMoments(moments2,huMonents2);
 for (int xOrder = 0; xOrder <= 3; xOrder++)
  for (int yOrder = 0; yOrder <= 3; yOrder++)
  {
   if (xOrder + yOrder <= 3)
   {
    double spatialMoment =cvGetSpatialMoment(moments2,xOrder, yOrder);
    double centralMoment =cvGetCentralMoment(moments2,xOrder, yOrder);
    double normalizedCentralMoment =cvGetNormalizedCentralMoment(moments2,xOrder, yOrder);
    printf("  %d,%d ： 空间距-%.2f, 中心距-%.2f, 归一化中心距-%.2f\n",xOrder,yOrder,spatialMoment,centralMoment,normalizedCentralMoment);
   }
  }
 printf("Hu矩：%f,%f,%f,%f,%f,%f,%f\n",huMonents2->hu1,huMonents2->hu2,huMonents2->hu3,huMonents2->hu3,huMonents2->hu5,huMonents2->hu6,huMonents2->hu7);

 //Hu矩匹配
 double hu = cvMatchShapes(contour1,contour2,CV_CONTOURS_MATCH_I1,0);
 printf("\n\nHu矩匹配：%f\n",hu);

 //轮廓树匹配
 CvMemStorage* storage3 = cvCreateMemStorage();
 CvMemStorage* storage4= cvCreateMemStorage();
 CvContourTree* tree1 = cvCreateContourTree(contour1,storage3,200);
 CvContourTree* tree2 = cvCreateContourTree(contour2,storage4,200);
 double tree = cvMatchContourTrees(tree1,tree2,CV_CONTOURS_MATCH_I1,200);
 printf("\n轮廓树匹配：%.4f\n\n",tree);

 //将轮廓线画出
 double s1,len1,h1,d1;
 CvRect rect1;
 CvBox2D box1,ellipse1;
 CvPoint2D32f center1;
 float radius1;
 IplImage* mask1=cvCreateImage(cvGetSize(img1),8,1);
 IplImage* dst1=cvCreateImage(cvGetSize(img1),8,3);

 //轮廓的成对几何直方图匹配  
 int sizes[2] = {60, 200};
 float ranges[2][2] = {{0,PI}, {0,200}};
 float** rangesPtr = new float* [2];
 rangesPtr[0] = ranges[0];
 rangesPtr[1] = ranges[1];

 CvHistogram* hist1,*hist2;
 hist1 = cvCreateHist(2, sizes, CV_HIST_ARRAY, rangesPtr, 1);
 hist2 = cvCreateHist(2, sizes, CV_HIST_ARRAY, rangesPtr, 1);
 cvCalcPGH(contour1, hist1); 
 cvCalcPGH(contour2, hist2);
 cvNormalizeHist(hist1, 1);
 cvNormalizeHist(hist2, 1);

 double hist= cvCompareHist(hist1,hist2, CV_COMP_INTERSECT);
 printf("\n成对几何直方图匹配:%.2f\n\n\n", hist);

 printf("图1：\n\n");
 for( int i=1; contour1!= 0; contour1 = contour1->h_next,i++)  
 {  
  IplImage* img1_copy =cvCloneImage(img1);
  printf("\n第%d个轮廓：\n",i);
  CvScalar color = CV_RGB( 0, 0,255);  
  cvZero( mask1);
  cvZero( dst1 ); 
  cvDrawContours(mask1, contour1, color, color, -1, 1);//边界线的厚度，默认为 1，得到的是轮廓外形这里设置为 -1即CV_FILLED，
  //是指对轮廓内的区域进行填充,通过cvCopy可将原图轮廓中的内容保留下来
  cvCopy(img1_copy, dst1, mask1); //可将原图轮廓中的内容保留下来
  cvSaveImage("111.bmp",dst1);//debug查看效果
  s1 = fabs(cvContourArea(contour1));//轮廓面积
  printf("  轮廓面积：%f\n",s1);
  len1=cvArcLength(contour1);
  printf("  轮廓周长：%f\n",len1);
  rect1=cvBoundingRect(contour1);//轮廓的边界框
  cvRectangle(img1_copy,cvPoint(rect1.x,rect1.y),cvPoint(rect1.x+rect1.width,rect1.y+rect1.height),CV_RGB(255,255,0));
  box1=cvMinAreaRect2(contour1);//最小矩形
  DrawBox(box1,img1_copy);
  cvMinEnclosingCircle(contour1,¢er1,&radius1);//外围圆
  cvCircle(img1_copy,cvPoint((int)center1.x,(int)center1.y),radius1,CV_RGB(0,255,0));
  ellipse1=cvFitEllipse2(contour1);//外围椭圆
  cvEllipseBox(img1_copy,ellipse1,CV_RGB(0,0,255));
  cvSaveImage("222.bmp",img1_copy);//debug查看效果
  int flag=cvCheckContourConvexity(contour1);//凹凸性，检测不准确???
  if (flag)
   printf("  为凸包\n");
  else
   printf("  为凹包\n");
  CvSeq* convexHull1=cvConvexHull2(contour1,NULL,1,1);//凸外多边形
  //当return_points=0时，用cvConvexHull2函数得到的凸外形，包含的是轮廓的定点的指针或下标，
  //而当return_points为非0时，得到的是外形点的本身。大家知道计算整个轮廓或部分轮廓的面积的函数定义为：
  cvZero( mask1);
  cvZero( dst1 ); 
  cvDrawContours(mask1,convexHull1, color, color, -1, 2);
  cvCopy(img1_copy, dst1, mask1); //可将原图轮廓中的内容保留下来
  cvSaveImage("333.bmp",dst1);//debug查看效果
  h1 = fabs(cvContourArea(convexHull1));//轮廓面积
  printf("  凸外多边形面积为：%f\n",h1);
  //CvSeq *defects1=cvConvexityDefects(contour1,convexHull1);//缺陷
  //Seq<CvConvexityDefect> defects2=cvConvexityDefects(contour1,convexHull1);//为什么都不可以
  //d1 = fabs(cvContourArea(defects1));//轮廓面积
  //printf("  缺陷面积为：%f\n",d1);
 }
 cvReleaseMemStorage(&storage1);
 cvReleaseMemStorage(&storage2);
 cvReleaseMemStorage(&storage3);
 cvReleaseMemStorage(&storage4);

}