Hough变换

先上代码,c++

1.hough检测线

//LineFinder.h
#include"opencv2/imgproc/imgproc.hpp"
#include"opencv2/highgui/highgui.hpp"
#include "opencv2/core/core.hpp"
#include<iostream>
//#include <math.h>
//#include <cmath>
using namespace std;
using namespace cv;
#define PI 3.141592
class LineFinder{
private:
    Mat img;
    vector<Vec4i>lines;
    double deltaRho;
    double deltaTheta;
    int minVote;
    double minLength;
    double maxGap;
public:
    LineFinder():deltaRho(1),deltaTheta(PI/180),minVote(10),minLength(0.),maxGap(0.){}
    void setAccResolution(double dRho,double dTheta)
    {
        deltaRho=dRho;
        deltaTheta=dTheta;
    }
    void setMinVote (int minv)
    {
        minVote=minv;
    }
    void setLineLengAndGap(double length,double gap)
    {
        minLength=length;
        maxGap=gap;
    }
    vector<Vec4i> findLines(Mat &binary)
    {
        lines.clear();
        HoughLinesP(binary,lines,deltaRho,deltaTheta,minVote,minLength,maxGap);
        return lines;
    }
 
    void drawDetectedLines(Mat &image,Scalar color=Scalar(255,255,255))
    {
        vector<Vec4i>::const_iterator it2=lines.begin();
        while (it2!=lines.end())
        {
            Point pt1((*it2)[0],(*it2)[1]);
            Point pt2((*it2)[2],(*it2)[3]);
            line(image,pt1,pt2,color);
            ++it2;
        }
 
    }
 
};

//main
#include "LineFinder.h"
void main()
{
    Mat img=imread("C:\\Users\\Administrator\\Desktop\\工作\\testp\\road.jpg",0);
   Mat out1;
    Canny(img,out1,125,350);
    Mat img1;
    img1=img.clone();
    vector<Vec2f> lines;
    HoughLines(out1,lines,1,PI/180,60);
    vector<Vec2f>::const_iterator it=lines.begin();
    while (it!=lines.end())
    {
        float rho=(*it)[0];
        float theta=(*it)[1];
        if (theta<PI/4.||theta>3.*PI/4.)
        {
            Point pt1(rho/cos(theta),0);
            Point pt2((rho-out1.rows*sin(theta))/cos(theta),out1.rows);
            line(img1,pt1,pt2,Scalar(255),1);
        }
        else
        {
            Point pt1(0,rho/sin(theta));
            Point pt2(out1.cols,(rho-out1.cols*cos(theta))/sin(theta));
            line(img1,pt1,pt2,Scalar(255),1);
        }
        ++it;
    }
 
    LineFinder finder;
    finder.setLineLengAndGap(100,20);
    finder.setMinVote(60);
    finder.findLines(out1);
    finder.drawDetectedLines(img);
 
 
 
 
    imshow("houghlinep",img);
    imshow("original",img1);
    imshow("out",out1);
    waitKey(0);
 
 
 
} 

2.Hough检测⚪

#include "opencv2/core/core.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"
using namespace cv;
void main()
{
    Mat img=imread("C:\\Users\\Administrator\\Desktop\\工作\\testp\\chariot.jpg",0);
    Mat img1;
    GaussianBlur(img,img1,Size(5,5),1.5);
vector<Vec3f>circles;
HoughCircles(img1,circles,CV_HOUGH_GRADIENT,2,50,200,100,25,100);
vector<Vec3f>::const_iterator itc=circles.begin();
while(itc!=circles.end())
{
    circle(img1,Point((*itc)[0],(*itc)[1]),(*itc)[2],Scalar(255),5);
    ++itc;
}
 
 
 
    imshow("original",img);
    imshow("img1",img1);
    waitKey(0);
 
 
 
 
}

3.点集的直线拟合

#include "LineFinder.h"
void main()
{
    Mat img=imread("C:\\Users\\Administrator\\Desktop\\工作\\testp\\road.jpg",0);
   Mat out1;
    Canny(img,out1,125,350);
    Mat img1;
    img1=img.clone();
    vector<Vec2f> lines;
    HoughLines(out1,lines,1,PI/180,60);
    vector<Vec2f>::const_iterator it=lines.begin();
    while (it!=lines.end())
    {
        float rho=(*it)[0];
        float theta=(*it)[1];
        if (theta<PI/4.||theta>3.*PI/4.)
        {
            Point pt1(rho/cos(theta),0);
            Point pt2((rho-out1.rows*sin(theta))/cos(theta),out1.rows);
            line(img1,pt1,pt2,Scalar(255),1);
        }
        else
        {
            Point pt1(0,rho/sin(theta));
            Point pt2(out1.cols,(rho-out1.cols*cos(theta))/sin(theta));
            line(img1,pt1,pt2,Scalar(255),1);
        }
        ++it;
    }
 
    LineFinder finder;
    finder.setLineLengAndGap(100,20);
    finder.setMinVote(60);
    finder.findLines(out1);
    finder.drawDetectedLines(img);
 
 
    vector<Vec4i>lines111=finder.findLines(out1);
    int n=0;
    Mat oneline(out1.size(),CV_8U,Scalar(0));
    line(oneline,Point(lines111[n][0],lines111[n][1]),Point(lines111[n][2],lines111[n][3]),Scalar(255),3);
    bitwise_and(out1,oneline,oneline);
    //threshold(oneline,oneline,100,255,THRESH_BINARY_INV);
    vector<Point>points;
    for (int y=0;y<oneline.rows;y++)
    {
        uchar *rowPtr=oneline.ptr<uchar>(y);
        for (int x=0;x<oneline.cols;x++)
        {
            if (rowPtr[x])
            {
                points.push_back(Point(x,y));
            }
        }
    }
    Vec4f line12;
    fitLine(points,line12,CV_DIST_L2,0,0.01,0.01);
    int x0=line12[2];
    int y0=line12[3];
    int x1=x0+100*line12[0];
    int y1=y0+100*line12[1];
    
    line(img,Point(x0,y0),Point(x1,y1),Scalar(0),3);
 
 
    imshow("src",img);
 
    imshow("oneline",oneline);
    waitKey(0);

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值