opencv学习日记——检测图像中的线段（效果很棒）

opencv学习日记——检测图像中的线段

1、图像输出结果

效果很好，我很满意

2、代码部分

#include<iostream>
#include<opencv2/opencv.hpp>

using namespace std;
using namespace cv;

int main()
{
    Mat img = imread("直线1.png", IMREAD_GRAYSCALE);
    if (img.empty())
    {
        cout << "输入图像为空";
        return -1;
    }

    Mat edge;
    //Canny(原图， 新图， 阈值， 阈值， 卷积核大小， 梯度)
    Canny(img, edge, 80, 180, 3, false);
    //threshold(原图，新图， 阈值， 超过阈值变成255， 类型（二值化）)
    threshold(edge, edge, 170, 255, THRESH_BINARY);

    vector<Vec4i> lines1, lines2;
    //HoughLinesp（原图， 向量， r的单位长度， theta单位长度， 阈值（大于该阈值认定为直线）， 直线最小长度阈值， 直线相邻两点最大距离）
    HoughLinesP(edge, lines1, 1, CV_PI / 180, 150, 30, 10);
    HoughLinesP(edge, lines2, 1, CV_PI / 180, 150, 30, 30);
    
    Mat img1;
    img.copyTo(img1);
    //连接计算出来的直线
    for (size_t i = 0; i < lines1.size(); i++)
    {
        line(img, Point(lines1[i][0], lines1[i][1]), Point(lines1[i][2], lines1[i][3]), Scalar(255), 3 );
    }
    for (size_t i = 0; i < lines2.size(); i++)
    {
        line(img1, Point(lines2[i][0], lines2[i][1]), Point(lines2[i][2], lines2[i][3]), Scalar(255), 3 );
    }

    imshow("img", img);
    imshow("img1", img1);

    waitKey(0);
    return 0;
}