yolov5+车道线检测

news/2025/2/7 10:20:03/

        目标检测与车道线检测在自动驾驶以及车辆定位中起着重要的辅助作用,是环境感知中不可缺少的一个部分。基于深度学习的车道线检测方法近年来也在不断的提升,比如论文:Ultra Fast Deep Lane Detection with HybridAnchor Driven Ordinal Classification.该论文是提出了一种基于anchor的车道线检测方法,这种方法借鉴了目标检测的思想,预设定了anchor,比基于图像分割的车道线检测方法在速度上有很大的提升。到考虑到如果算法均采用深度学习方法,在最终算法落地的时候对于模型优化是比较高的,而且有些模型在参数和计算量上都比较大,因此本文采用深度学习的yolov5做为目标检测结合传统的图像处理方法做车道线检测

yolov5这部分不在详细展开,直接下载官方代码即可。重点是介绍车道线检测方法思路供大家学习。


主要步骤分四步:

1.颜色筛选

2.边缘检测

3.mask检测区域筛选

4.霍夫变换


yolov5+车道线检测

1.颜色筛选

        比如下面视频中,我们想先通过颜色筛选先对图像进行过滤,方法也很简单,就是通过设置阈值对RGB三通道先进行筛选。这个阈值也不是随便设置,教给大家一个方法。采用微信截图,然后把鼠标放到车道上面,记下RGB三通道的值即可【可以上下浮动调整,直到筛选出你较为满意的效果,注意opencv里的通道顺序是BGR】。过滤以后就如下所示。

2.边缘检测

        这里的边缘检测主要是检测车道线的边缘,采用的是Canny算子进行检测。在边缘检测之前需要先把图像转为灰度图,设置核的大小,一般设置为5,用高斯进行去噪处理后在采用Canny进行边缘检测。

def canny_edg_(img):gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) # 转为灰度图像kernel_size = 5blur_gray = cv2.GaussianBlur(gray, (kernel_size, kernel_size), 0) # 高斯滤波low_thres = 160high_thres = 240edg_img = cv2.Canny(blur_gray, low_thres, high_thres)return edg_img

边缘检测的效果图如下所示 ,可以看到这里已经将车道线检测出来了【当然这里还有其他边缘】。

3.mask检测区域筛选

        这一步是很重要的,我们可以分析一下步骤二中的图,在道路环境中不仅仅有车道线,还有山体的轮廓等等,那有什么办法可以进一步筛选吗?答案是有的,可以看到车道线基本是位于画面的三分之二左右的位置上,上面的部分是不需要检测,所以如果我们可以创建一个mask把需要检测的区域保留下来即可。mask图像如下所示,可以看到已经把车道区域筛选出来了【这个区域可以做更细微的调整,比如设置一个比例值,我这里只是做个演示】

4.霍夫变换        

        接下来是通过霍夫变换获取车道线的坐标进行绘制,可以直接用opencv中现有的函数即可,即cv2.HoughLinesP ,同时还可以计算出左右车道线坐标【通过霍夫空间映射到笛卡尔空间,这里的m表示斜率,c表示截距,那么我们可以斜率判断车道线是左还是右】,最后通过cv2.line进行绘制即可。

                # 求直线方程斜率判断左右车道m = (y2 - y1) / (x2 - x1)c = y1 - m * x1if m < 0:  # 左车道left_line.append((m, c))elif m >= 0:  # 右车道right_line.append((m, c))

效果图就是下面这样的。如果还想检测的区域更长一些,可以调整mask区域即可【但也会引入其他的噪声】 。


然后可以将以上车道线检测应用于目标检测中, 我这里是放到了yolov5中,大家也可以尝试放在其他的目标检测算法后,也可以放在图像分割中都是可以的。

【yolov5+车道线检测】 


 本文仅供研究学习使用,暂时不放源码的原因是由于以前碰到过一些不讲信誉的公司让我非常气愤【比如格灵深瞳】。大家有兴趣学习交流的可以私信或者留言都可以


http://www.ppmy.cn/news/11894.html

相关文章

[ 环境搭建篇 ] 安装 java 环境并配置环境变量(附 JDK1.8 安装包)

&#x1f36c; 博主介绍 &#x1f468;‍&#x1f393; 博主介绍&#xff1a;大家好&#xff0c;我是 _PowerShell &#xff0c;很高兴认识大家~ ✨主攻领域&#xff1a;【渗透领域】【数据通信】 【通讯安全】 【web安全】【面试分析】 &#x1f389;点赞➕评论➕收藏 养成习…

包体积优化 · 实战论 · 怎么做包体优化? 做好能晋升吗? 能涨多少钱?

“ 【小木箱成长营】包体积优化系列文章&#xff1a; 包体积优化 方法论 揭开包体积优化神秘面纱 包体积优化 工具论 初识包体积优化 BaguTree 包体积优化录播视频课 ”一、引言 Hello&#xff0c;我是小木箱&#xff0c;欢迎来到小木箱成长营系列教程&#xff0c;今天将分…

锁策略和synchronized

1.常见的锁策略&#xff08;1&#xff09;乐观锁 和 悲观锁乐观锁&#xff1a;预测锁竞争的情况不激烈&#xff08;工作量较少&#xff09;悲观锁&#xff1a;预测锁竞争的情况很激烈&#xff08;工作量较多&#xff09;&#xff08;2&#xff09;轻量级锁 和 重量级锁轻量级锁…

【每日一题】【LeetCode】【第十二天】区域和检索 - 数组不可变

解决之路 题目描述 测试案例&#xff08;部分&#xff09; 第一次 emmm&#xff0c;说实话&#xff0c;一开始我还真没看懂题目是什么意思。。。。 自己按我自己理解的方式写了一下代码&#xff0c;用测试案例跑了下&#xff0c;成功了。 不过&#xff0c;放进去跑不通&…

【Linux】六、Linux 基础IO(一)|重谈文件|C语言文件操作|操作系统文件操作(系统文件I/O)|文件描述符

目录 一、重谈文件 二、C语言文件操作 2.1 重谈C语言文件操作 2.2 补充细节 三、操作系统文件操作&#xff08;系统文件I/O&#xff09; 3.1 文件相关系统调用&#xff1a;close 3.2 文件相关系统调用&#xff1a;open 3.2.1 open 的第二个参数 flags 3.2.2 open 的第…

自定义类型:结构体,枚举,联合(2)

TIPS 1. 类型的定义可以考虑放在头文件里头。 2. 一个汉字存储的时候占两个字节空间 3. 关于结构体变量初始化的一些细节 4. 关于结构体内存对齐的补充 1. 2. S1和S2类型的成员一模一样&#xff0c;但是S1和S2所占空间的大小有了一些区别。 3. 这两个结构体类型成员都…

C++ Socket 构造函数参数解析

int socket(int af, int type, int protocol); 1、) af 为地址族&#xff08;Address Family&#xff09;&#xff0c;也就是 IP 地址类型&#xff0c;常用的有 AF_INET 和 AF_INET6。AF 是“Address Family”的简写&#xff0c;INET是“Inetnet”的简写。AF_INET 表示 IPv4 地…

Android ANR bugreport log分析

最近工作中频繁遇到设备ANR问题&#xff0c;而且是概率性的那种&#xff0c;于是决定花点时间找找规律复现分析下 说道这里&#xff0c;抓日志是问题解决的最有效途径&#xff0c;这里不得不说一下 bugreport log,其实网上关于它的分析方法有很多&#xff0c;在此仅仅是为了记录…