霍夫变换

#include<iostream>
#include<opencv2\opencv.hpp>
#include<cmath>
const double PI=3.1415926;
int main()
{
cv::Mat srcImage = cv::imread("g:\\road.jpg",1);
cv::Mat grayImage,contours;
cv::cvtColor(srcImage,grayImage,CV_BGR2GRAY);
cv::Canny(grayImage,contours,125,300);  
cv::vector<cv::Vec2f> lines;
cv::HoughLines(contours,lines,1,PI/180,160);
cv::vector<cv::Vec2f>::const_iterator it = lines.begin();
while(it!=lines.end()) 草稿箱
{
float rho =(*it)[0];
float therta = (*it)[1];
if(therta<PI/4||therta>PI*3/4)
{
cv::Point pt1(rho/cos(therta),0);//线与第一行的交点
cv::Point pt2((rho-srcImage.rows*sin(therta))/cos(therta),srcImage.rows-1);
cv::line(srcImage,pt1,pt2,cv::Scalar(255),1);
}
else
{
cv::Point pt1(0,rho/sin(therta));//与第一列的交点
cv::Point pt2(srcImage.cols-1,(rho-srcImage.cols*cos(therta))/sin(therta));
cv::line(srcImage,pt1,pt2,cv::Scalar(255),1);
}
it++;
}
cv::imshow("srcHoughlines",srcImage);
cv::waitKey();
return 0;


}