线和边缘检测器,opencv

时间:2014-12-02 19:52:09

标签: c++ opencv hough-transform

我正在尝试从迷宫中处理以下图像。我的问题是关于如何处理边缘。我正在使用OpenCV 2.4和c ++。 我想知道是否有办法区分地板和墙壁之间的边缘与地板上的线条?

地板是黑色的,墙壁是白色的,地板上画的线条也是白色的。

我想要做的是区分墙壁和地板上的标记。地板上的线条会给我一个距离参考,如果我可以转入迷宫。虽然墙壁只是告诉了迷宫大厅的极限。

在这里你可以找到我已经完成的过程图像。

http://oi61.tinypic.com/34hx7a0.jpg

http://oi60.tinypic.com/2zpqfq0.jpg

我正在使用Canny和HoughLinesP函数来检测和保存线条。但正如您在图像中看到的那样,程序不会将线条与边缘分开。

代码:

vector<Vec4i> get_lines(Mat dst, Mat cdst)
{
    vector<Vec4i> lines;
    HoughLinesP(dst, lines, 1, CV_PI/180, 100, 50, 10 );
    for( size_t i = 0; i < lines.size(); i++ )
    {
        Vec4i l = lines[i];
        double size = norm(Mat(Point(l[0], l[1])), Mat(Point(l[2], l[3])) );
        if(size > 100)
            line( cdst, Point(l[0], l[1]), Point(l[2], l[3]), Scalar(0,0,255), 3, CV_AA);
    }

    return lines;
}

主要功能是:

int main(int argc, char** argv)
{
    const char* filename = argc >= 2 ? argv[1] : "pic1.jpg";

    Mat src = imread(filename, 0);

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

    Mat dst, cdst;

    Canny(src, dst, 50, 200, 3);

    cvtColor(dst, cdst, CV_GRAY2BGR);
    vector<Vec4i> lines = get_lines(dst, cdst);

    imshow("source W&B", src);
    imshow("edges", dst);
    imshow("detected lines", cdst);

    imwrite("lines.jpg",cdst);
    imwrite("src.jpg",src);
    imwrite("canny.jpg",dst);

    waitKey();

    return 0;
}

1 个答案:

答案 0 :(得分:1)

要尝试的显而易见的事情是比较线路两侧的像素亮度。

制作三个区域:与线的一侧稍微距离的像素,与线的另一侧稍微距离的像素,以及靠近线的像素。计算任一区域的平均亮度。

墙壁为浅灰色,地板为黑色,线条为白色,所以

  • 如果一边明显比另一边明亮,那么它可能是边缘(你甚至可以分辨出哪一面是地板),
  • 如果两侧比中间明显更暗,则可能是地板上的标记。
  • (如果线是垂直的,则是墙壁边缘)