C++ opencv-3.4.1 车道线的提取

贾志刚的github发现一个比较新奇的demo,使用opencv进行车道线的检测。
使用的方法都比较简单,但是效果看起来确实高端,可以和最新的无人驾驶相结合了。

  1. 读取视频
  2. 对视频的每一帧进行处理
  3. 首先截取本车道线的视角,保证只存在本车道线的区域
  4. 进行二值化处理
  5. 查找轮廓
  6. 通过角度过滤轮廓
  7. 确定直线方向

每个处理都很简洁,但是用在车道线检测上立马看起来不一样了。

#include <opencv2/opencv.hpp>
#include <iostream>
#define PI 3.1415926

using namespace cv;
using namespace std;

RNG rng(12345);
void find_Lanes(Mat &frame);
int main() {
    String win_title = "input frame";
    namedWindow(win_title, CV_WINDOW_AUTOSIZE);
    VideoCapture capture("images/lane.avi");
    if (!capture.isOpened()) {
        printf("could not load video file");
        return -1;
    }

    
    int i = 0;
    Mat frame;
    while (capture.read(frame)) {
        imshow(win_title, frame);
        find_Lanes(frame);
        char c = waitKey(10);
        
        if (c == 27) {
            break;
        }

    }

    waitKey(0);
    return 0;
}

void find_Lanes(Mat &frame) {

    // 裁剪出本车道需要的车道线
    int offx = frame.cols / 5;
    int offy = frame.rows / 3;
    Rect rect;
    rect.x = offx;
    rect.y = frame.rows - offy;
    rect.width = frame.cols - offx * 2;
    rect.height = offy - 50;
    
    // 得到本车道的车道线
    Mat copy = frame(rect).clone();
    imshow("copy", copy);

    // 对车道进行二值化
    GaussianBlur(copy, copy, Size(3, 3), 0);
    Mat gray, binary;
    cvtColor(copy, gray, COLOR_BGR2GRAY);
    threshold(gray, binary, 0, 255, THRESH_BINARY | THRESH_OTSU);
    Mat mask = Mat::zeros(frame.size(), CV_8UC1);
    binary.copyTo(mask(rect));  
    imshow("binary", binary);

    // 查找轮廓
    vector<vector<Point> > contours;
    vector<Vec4i> hierarchy;
    findContours(mask, contours, hierarchy, RETR_TREE, CHAIN_APPROX_SIMPLE, Point(0, 0));
    Mat drawing = Mat::zeros(mask.size(), CV_8UC3);
    for (size_t i = 0; i< contours.size(); i++)
    {
        // 使用角度过滤轮廓
        RotatedRect rrt = minAreaRect(contours[i]);
        int angle = abs(rrt.angle);
        if (angle < 20 || angle > 160 || angle == 90)
            continue;
        printf("rrt.angle: %.2f\n", rrt.angle);
        Scalar color = Scalar(rng.uniform(0, 255), rng.uniform(0, 255), rng.uniform(0, 255));
        //drawContours(frame, contours, (int)i, color, 2, 8, hierarchy, 0, Point());
        Point pt1(-1, -1);
        Point pt2(-1, -1);
        int miny = 100000;
        int maxy = 0;
        
        // 取得最大轮廓的两个端点
        for (int p = 0; p < contours[i].size(); p++) {
            Point onep = contours[i][p];
            if (miny > onep.y) {
                miny = onep.y;
                pt1.y = onep.y;
                pt1.x = onep.x;
            }
            if (maxy < onep.y) {
                maxy = onep.y;
                pt2.y = onep.y;
                pt2.x = onep.x;
            }
            
        }
        if (pt1.x < 0 || pt2.x< 0)
            continue;
        // printf("line Point1 (x = %d, y = %d) to Point2 (x=%d, y=%d)\n", pt1.x, pt1.y, pt2.x, pt2.y);
        line(frame, pt1, pt2, Scalar(255, 0, 0), 3, 8);
    }

    imshow("mask", mask);
    imshow("lane-lines", frame);
}

原始摄像头的视角

裁剪之后的视角

二值化

轮廓查找

参考:

贾志刚的git
视频链接:https://pan.baidu.com/s/1Ifkq9vZmIF9tEf8Johy_Og
提取码:li5y

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,670评论 5 460
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,928评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,926评论 0 320
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,238评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,112评论 4 356
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,138评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,545评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,232评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,496评论 1 290
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,596评论 2 310
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,369评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,226评论 3 313
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,600评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,906评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,185评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,516评论 2 341
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,721评论 2 335

推荐阅读更多精彩内容