opencv拟合直线并在屏幕上绘出

#include "stdafx.h"

#include <cv.h>

#include <highgui.h>

#include <stdio.h>

#include <math.h>

int main()

{

 IplImage *img=cvCreateImage(cvSize(500,500),8,3);//开辟500×500的8位3通道的图像内存空间。

 CvRNG rng=CvRNG(-1);//产生随机数产生器的状态。

 cvNamedWindow("fitline",0);

 for (;;)

 {

  char key;

  int i;

  int count =cvRandInt(&rng)%100+1;//产生1~100之间的随机整数

  int outliers=count/5;//选取1/5数量的点作为outliers。

  float a=cvRandReal(&rng)*200;//产生0~1随机浮点数

  float b=cvRandReal(&rng)*40;

  float angle=cvRandReal(&rng)*CV_PI;//随机产生弧度

  float cos_a=cos(angle);

  float sin_a=sin(angle);

  CvPoint pt1 , pt2;

  CvPoint * points=(CvPoint*)malloc(count *sizeof(points[0]));

  CvMat pointMat=cvMat(1,count,CV_32SC2,points);

  float line[4];

  float d,t;

  b=MIN(a*0.3,b);

  //产生噪声点

  for (i=0;i<count-outliers;i++)//被噪声污染的直线上的点

  {

   float x=(cvRandReal(&rng)*2-1)*a;

   float y=(cvRandReal(&rng)*2-1)*b;

   points[i].x=cvRound(x*cos_a-y*sin_a+img->width/2);

   points[i].y=cvRound(x*cos_a+y*sin_a+img->height/2);

  }

  for (;i<count;i++)//异常点

  {

   points[i].x=cvRandInt(&rng) % img->width;

   points[i].y=cvRandInt(&rng) % img->height;

  }

  //拟合直线

  cvFitLine(&pointMat,CV_DIST_L1,1,0.001,0.001,line);

 

  cvZero(img);

  //画图

  for (i=0;i<count;i++)

  {

   cvCircle(img,points[i],2,(i<count-outliers)?CV_RGB(255,0,0):CV_RGB(255,255,0),

    CV_FILLED,CV_AA,0);

  }

       

  t=(float)(img->width+img->height);

  pt1.x=cvRound(line[2]-line[0]*t);//将直线上的点（line[2],line[3])沿直线移动一段距离

  pt1.y=cvRound(line[3]-line[1]*t);//将直线上的点（line[2],line[3])沿直线移动一段距离

  pt2.x=cvRound(line[2]+line[0]*t);//将直线上的点（line[2],line[3])沿直线移动一段距离

  pt2.y=cvRound(line[3]+line[1]*t);//将直线上的点（line[2],line[3])沿直线移动一段距离

  cvLine(img,pt1,pt2,CV_RGB(0,255,0),3,CV_AA,0);

  cvShowImage("fitline",img);

  key=(char)cvWaitKey(0);

  if (key==27||key=='q'||key=='Q')

   break;

  free(points);

 }

 cvDestroyWindow("fitline");

 cvReleaseImage(&img);

 return 0;

}

注意：

 （1）：（line[0],line[1])不是直线上的点。它表示的是一个与拟合的直线平行的向量。且（line[0]*line[0]+line[1]*line[1]=1)，所以直线的倾斜角度=asin(line[1])=acos(line[0])=atg(line[1]/line[0])。

（2）：很明显，已知一个直线的方向（即向量）和直线上的一点（line[2],line[3])，当然能唯一确定一条直线。