yolov5+车道线检测
创始人
2024-05-10 18:32:34
0

        目标检测与车道线检测在自动驾驶以及车辆定位中起着重要的辅助作用,是环境感知中不可缺少的一个部分。基于深度学习的车道线检测方法近年来也在不断的提升,比如论文: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+车道线检测】 


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

相关内容

热门资讯

喜欢穿一身黑的男生性格(喜欢穿... 今天百科达人给各位分享喜欢穿一身黑的男生性格的知识,其中也会对喜欢穿一身黑衣服的男人人好相处吗进行解...
发春是什么意思(思春和发春是什... 本篇文章极速百科给大家谈谈发春是什么意思,以及思春和发春是什么意思对应的知识点,希望对各位有所帮助,...
网络用语zl是什么意思(zl是... 今天给各位分享网络用语zl是什么意思的知识,其中也会对zl是啥意思是什么网络用语进行解释,如果能碰巧...
为什么酷狗音乐自己唱的歌不能下... 本篇文章极速百科小编给大家谈谈为什么酷狗音乐自己唱的歌不能下载到本地?,以及为什么酷狗下载的歌曲不是...
家里可以做假山养金鱼吗(假山能... 今天百科达人给各位分享家里可以做假山养金鱼吗的知识,其中也会对假山能放鱼缸里吗进行解释,如果能碰巧解...
华为下载未安装的文件去哪找(华... 今天百科达人给各位分享华为下载未安装的文件去哪找的知识,其中也会对华为下载未安装的文件去哪找到进行解...
四分五裂是什么生肖什么动物(四... 本篇文章极速百科小编给大家谈谈四分五裂是什么生肖什么动物,以及四分五裂打一生肖是什么对应的知识点,希...
怎么往应用助手里添加应用(应用... 今天百科达人给各位分享怎么往应用助手里添加应用的知识,其中也会对应用助手怎么添加微信进行解释,如果能...
客厅放八骏马摆件可以吗(家里摆... 今天给各位分享客厅放八骏马摆件可以吗的知识,其中也会对家里摆八骏马摆件好吗进行解释,如果能碰巧解决你...
苏州离哪个飞机场近(苏州离哪个... 本篇文章极速百科小编给大家谈谈苏州离哪个飞机场近,以及苏州离哪个飞机场近点对应的知识点,希望对各位有...