OpenCV-提取直线

#include <cv.h>   
#include <highgui.h>   
#include <math.h>  
 
  void cvThin( IplImage* src, IplImage* dst, int iterations=1)//将IPL_DEPTH_8U型二值图像进行细化
{
 CvSize size = cvGetSize(src);

 cvCopy(src, dst);//拷贝一个数组给另一个数组
    int n = 0,i = 0,j = 0;
 for(n=0; n<iterations; n++)
 {
IplImage* t_image = cvCloneImage(dst);

  for(i=0; i<size.height;  i++)
  {
   for(int j=0; j<size.width; j++)
   {
    if(CV_IMAGE_ELEM(t_image,byte,i,j)==1)
    {
     int ap=0;
     int p2 = (i==0)?0:CV_IMAGE_ELEM(t_image,byte, i-1, j);
     int p3 = (i==0 || j==size.width-1)?0:CV_IMAGE_ELEM(t_image,byte, i-1, j+1);
     if (p2==0 && p3==1)
     {
      ap++;
     }
     int p4 = (j==size.width-1)?0:CV_IMAGE_ELEM(t_image,byte,i,j+1);
     if(p3==0 && p4==1)
     {
      ap++;
     }
     int p5 = (i==size.height-1 || j==size.width-1)?0:CV_IMAGE_ELEM(t_image,byte,i+1,j+1);
     if(p4==0 && p5==1)
     {
      ap++;
     }
     int p6 = (i==size.height-1)?0:CV_IMAGE_ELEM(t_image,byte,i+1,j);
     if(p5==0 && p6==1)
     {
      ap++;
     }
     int p7 = (i==size.height-1 || j==0)?0:CV_IMAGE_ELEM(t_image,byte,i+1,j-1);
     if(p6==0 && p7==1)
     {
      ap++;
     }
     int p8 = (j==0)?0:CV_IMAGE_ELEM(t_image,byte,i,j-1);
     if(p7==0 && p8==1)
     {
      ap++;
     }
     int p9 = (i==0 || j==0)?0:CV_IMAGE_ELEM(t_image,byte,i-1,j-1);
     if(p8==0 && p9==1)
     {
      ap++;
     }
     if(p9==0 && p2==1)
     {
      ap++;
     }
     if((p2+p3+p4+p5+p6+p7+p8+p9)>1 && (p2+p3+p4+p5+p6+p7+p8+p9)<7)
     {
      if(ap==1)
      {
       if(p2*p4*p8==0)
       {
        if(p2*p6*p8==0)
        {
         CV_IMAGE_ELEM(dst, byte,i,j)=0;
        }
       }
      }
     }                    
    }

   }

  }            
  cvReleaseImage(&t_image);

}}

int main(int argc, char** argv)  
{  
    IplImage* src;  

    if(src=cvLoadImage("1.jpg",0))  
    {  
		//第一步 图像细化
		IplImage* srcThinBinary,*srcThin;
		srcThinBinary = cvCreateImage(cvGetSize(src),8,1);
		srcThin = cvCreateImage(cvGetSize(src),8,1);
		cvZero(srcThinBinary);
		cvZero(srcThin);
		cvThreshold(src,srcThinBinary,100,1,CV_THRESH_BINARY_INV);
		cvThin(srcThinBinary,srcThin,80);
		cvThreshold(srcThin,srcThin,0.5,255,CV_THRESH_BINARY);
		cvCopy(srcThin,src);

		//第二步 直线提取
        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, 150, 3 );  
        cvCvtColor( dst, color_dst, CV_GRAY2BGR );  

        lines = cvHoughLines2( dst, storage, CV_HOUGH_PROBABILISTIC, 1, CV_PI/180, 50, 30, 10 );  
        for( i = 0; i < lines->total; i++ )  
        {  
            CvPoint* line = (CvPoint*)cvGetSeqElem(lines,i);  
            cvLine( color_dst, line[0], line[1], CV_RGB(255,0,0), 3, 8 );  
        }  
  
		//显示结果
        cvNamedWindow( "Source", 1 );  
        cvShowImage( "Source", src );  
  
        cvNamedWindow( "Hough", 1 );  
        cvShowImage( "Hough", color_dst );  

		//释放资源
		cvReleaseImage(&src);
		cvReleaseImage(&srcThinBinary);
		cvReleaseImage(&srcThin);
		cvReleaseImage(&dst);
		cvReleaseImage(&color_dst);
		cvReleaseMemStorage(&storage);

  
        cvWaitKey(0);  
    }  
}