OpenCV实现图像的直线检测

Z时代
2024-01-10
分类：综合

上一篇博文介绍了图像的Canny边缘检测，本文主要介绍图像的直线检测部分，主要使用概率霍夫变换来检测直线，调用的函数为HoughLinesP()，下面给出代码部分以及直线检测效果图：

1、代码部分：


// Detect_Lines.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include <cv.h>
#include "highgui.h"
using namespace std;
using namespace cv;
void drawDetectLines(Mat& image,const vector<Vec4i>& lines,Scalar & color)
{ 
 // 将检测到的直线在图上画出来 
 vector<Vec4i>::const_iterator it=lines.begin(); 
 while(it!=lines.end()) 
 { 
 Point pt1((*it)[0],(*it)[1]); 
 Point pt2((*it)[2],(*it)[3]); 
 line(image,pt1,pt2,color,2); //线条宽度设置为2 
 ++it; 
 }
} 
int _tmain(int argc, _TCHAR* argv[])
{
 Mat src_img=imread("..\\image_norm\\71253.jpg");
 imshow("src_img",src_img);
 Mat I; 
 cvtColor(src_img,I,CV_BGR2GRAY);    
 Mat contours; 
 Canny(I,contours,125,350); 
 threshold(contours,contours,128,255,THRESH_BINARY); 
 vector<Vec4i> lines; 
 HoughLinesP(contours,lines,1,CV_PI/180,80,50,10); 
 drawDetectLines(src_img,lines,Scalar(0,255,0)); 
 imshow("Detect_Lines",src_img); 
 cvWaitKey(0);
 return 0;
}