houghlines.cpp 2.34 KB
Newer Older
wester committed
1 2 3
#include <cmath>
#include <iostream>

wester committed
4 5 6 7
#include "opencv2/core/core.hpp"
#include "opencv2/highgui/highgui.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/gpu/gpu.hpp"
wester committed
8 9 10

using namespace std;
using namespace cv;
wester committed
11
using namespace cv::gpu;
wester committed
12 13 14 15 16

static void help()
{
    cout << "This program demonstrates line finding with the Hough transform." << endl;
    cout << "Usage:" << endl;
wester committed
17
    cout << "./gpu-example-houghlines <image_name>, Default is pic1.png\n" << endl;
wester committed
18 19 20 21
}

int main(int argc, const char* argv[])
{
wester committed
22
    const string filename = argc >= 2 ? argv[1] : "pic1.png";
wester committed
23 24 25 26 27 28 29 30 31 32

    Mat src = imread(filename, IMREAD_GRAYSCALE);
    if (src.empty())
    {
        help();
        cout << "can not open " << filename << endl;
        return -1;
    }

    Mat mask;
wester committed
33
    Canny(src, mask, 100, 200, 3);
wester committed
34 35

    Mat dst_cpu;
wester committed
36
    cvtColor(mask, dst_cpu, CV_GRAY2BGR);
wester committed
37 38 39 40 41 42
    Mat dst_gpu = dst_cpu.clone();

    vector<Vec4i> lines_cpu;
    {
        const int64 start = getTickCount();

wester committed
43
        HoughLinesP(mask, lines_cpu, 1, CV_PI / 180, 50, 60, 5);
wester committed
44 45 46 47 48 49 50 51 52

        const double timeSec = (getTickCount() - start) / getTickFrequency();
        cout << "CPU Time : " << timeSec * 1000 << " ms" << endl;
        cout << "CPU Found : " << lines_cpu.size() << endl;
    }

    for (size_t i = 0; i < lines_cpu.size(); ++i)
    {
        Vec4i l = lines_cpu[i];
wester committed
53
        line(dst_cpu, Point(l[0], l[1]), Point(l[2], l[3]), Scalar(0, 0, 255), 3, CV_AA);
wester committed
54 55 56 57
    }

    GpuMat d_src(mask);
    GpuMat d_lines;
wester committed
58
    HoughLinesBuf d_buf;
wester committed
59 60 61
    {
        const int64 start = getTickCount();

wester committed
62
        gpu::HoughLinesP(d_src, d_lines, d_buf, 1.0f, (float) (CV_PI / 180.0f), 50, 5);
wester committed
63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78

        const double timeSec = (getTickCount() - start) / getTickFrequency();
        cout << "GPU Time : " << timeSec * 1000 << " ms" << endl;
        cout << "GPU Found : " << d_lines.cols << endl;
    }
    vector<Vec4i> lines_gpu;
    if (!d_lines.empty())
    {
        lines_gpu.resize(d_lines.cols);
        Mat h_lines(1, d_lines.cols, CV_32SC4, &lines_gpu[0]);
        d_lines.download(h_lines);
    }

    for (size_t i = 0; i < lines_gpu.size(); ++i)
    {
        Vec4i l = lines_gpu[i];
wester committed
79
        line(dst_gpu, Point(l[0], l[1]), Point(l[2], l[3]), Scalar(0, 0, 255), 3, CV_AA);
wester committed
80 81 82 83 84 85 86 87 88
    }

    imshow("source", src);
    imshow("detected lines [CPU]", dst_cpu);
    imshow("detected lines [GPU]", dst_gpu);
    waitKey();

    return 0;
}