Hough变换——Opencv版本

[cpp] view plaincopy

1. void CCVMFCView::OnHoughLines()  
2. {  
3.     IplImage* src = 0,*color_dst;  
4.     CvMemStorage* storage = cvCreateMemStorage(0);  
5.     CvSeq* lines = 0;  
6.     int i;  
7.   
8.     if (workImg->nChannels==3) {  
9.         src = cvCreateImage(cvGetSize(workImg), IPL_DEPTH_8U, 1);  
10.         cvCvtColor(workImg, src, CV_BGR2GRAY);  
11.     }  
12.     else {  
13.         src = cvCloneImage( workImg );  
14.     }  
15.   
16.     dst = cvCreateImage( cvGetSize(src), 8 ,1 );  
17.     color_dst = cvCreateImage( cvGetSize(src), 8 ,3);  
18.     cvFlip(src);  
19.     cvCanny( src, dst, 50, 120, 3 );  
20.     cvCvtColor(dst,color_dst,CV_GRAY2BGR);  
21.   
22.     lines = cvHoughLines2( dst, storage, CV_HOUGH_PROBABILISTIC, 1, CV_PI/180, 20, 20,30 );  
23.     //lines = cvHoughLines2( dst, storage, CV_HOUGH_PROBABILISTIC, 1,CV_PI/180, 80, 60,30 );  
24.   
25.     for( i = 0; i < lines->total; i++ )  
26.     {  
27.         CvPoint* line = (CvPoint*)cvGetSeqElem(lines,i);  
28.         cvLine( color_dst, line[0], line[1], CV_RGB(255,0,0), 3, CV_AA, 0 );  
29.     }  
30.   
31.     cvNamedWindow( "Hough", 1 );  
32.     cvShowImage( "Hough", color_dst );  
33.   
34.     cvWaitKey(0);  
35.   
36.     cvReleaseMemStorage(&storage );  
37.     cvReleaseImage( &src );  
38.     cvReleaseImage( &dst );  
39.     cvDestroyWindow( "Hough" );  
40. }