OpenCV学习笔记-霍夫线变换2

原创

superdont 2022-08-15 12:20:55 博主文章分类：图像处理 ©著作权

文章标签 dst float 出版文档 #include 文章分类 后端开发

©著作权归作者所有：来自51CTO博客作者superdont的原创作品，请联系作者获取转载授权，否则将追究法律责任

采用参数形式为：

CV_HOUGH_PROBABILISTIC进行变化。

具体实现代码为：

#include "stdafx.h"
#include <cv.h>
#include <highgui.h>
#include <math.h>
int main(int argc, char** argv)
{
  IplImage* src;
  src=cvLoadImage("building.jpg",0);
  IplImage* dst = cvCreateImage( cvGetSize(src), 8, 1 );
  IplImage* color_dst = cvCreateImage( cvGetSize(src), 8, 3 );
  CvMemStorage* storage = cvCreateMemStorage(0);
  CvSeq* lines = 0;
  int i;
  cvCanny( src, dst, 50, 200, 3 );
  cvCvtColor( dst, color_dst, CV_GRAY2BGR );
  lines = cvHoughLines2( dst, storage, CV_HOUGH_STANDARD, 1, CV_PI/180, 150, 0, 0 );
  for( i = 0; i < lines->total; i++ )
  {
    float* line = (float*)cvGetSeqElem(lines,i);
    float rho = line[0];
    float theta = line[1];
    CvPoint pt1, pt2;
    double a = cos(theta), b = sin(theta);
    if( fabs(a) < 0.001 )
    {
      pt1.x = pt2.x = cvRound(rho);
      pt1.y = 0;
      pt2.y = color_dst->height;
    }
    else if( fabs(b) < 0.001 )
    {
      pt1.y = pt2.y = cvRound(rho);
      pt1.x = 0;
      pt2.x = color_dst->width;
    }
    else
    {
      pt1.x = 0;
      pt1.y = cvRound(rho/b);
      pt2.x = cvRound(rho/a);
      pt2.y = 0;
    }
    cvLine( color_dst, pt1, pt2, CV_RGB(255,0,0), 3, 8 );
  }

  cvNamedWindow( "Source", 1 );
  cvShowImage( "Source", src );
  cvNamedWindow( "Hough", 1 );
  cvShowImage( "Hough", color_dst );
  cvWaitKey(0);
}