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

来源：互联网发布：js给数组添加对象编辑：程序博客网时间：2024/05/16 14:05

采用参数形式为：

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);}

结果为：

参考资料：

1.学习OpenCV，于仕祺，刘瑞祯，清华大学出版，pp.175-179

2.http://www.opencv.org.cn/index.php/Cv%E5%9B%BE%E5%83%8F%E5%A4%84%E7%90%86#HoughLines

3.OpenCV文档，路径："OpenCV2.2\doc\opencv.pdf"