从零开始详解OpenCV车道线检测

前言

车道线检测是智能驾驶和智能交通系统中的重要组成部分,对于提高道路安全、交通效率和驾驶舒适性具有重要意义。在本篇文章中将介绍使用OpenCV进行车道线的检测

详解

导入包

python 复制代码
import cv2
import matplotlib.pyplot as plt
import numpy as np

读入图像并灰度化

python 复制代码
img = cv2.imread('road.png')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
plt.imshow(gray, cmap='gray')
plt.show()

二值化

python 复制代码
thresh, im = cv2.threshold(gray, 200, 255, cv2.THRESH_BINARY)
plt.imshow(im, cmap=plt.cm.gray)
plt.show()

边缘检测

python 复制代码
output = cv2.Canny(im, 180, 255)
plt.imshow(output, cmap='gray')
plt.show()

cv2.Canny()函数的语法和参数说明:

python 复制代码
语法:
	edges = cv2.Canny(image, threshold1, threshold2[, apertureSize[, L2gradient]])
参数:
	image:输入图像,通常是8位灰度图像。
	threshold1:第一个阈值。边缘梯度值高于这个阈值的像素被视为边缘。
	threshold2:第二个阈值。在这两个阈值之间的边缘会被认为是强边缘,而低于threshold1但高于threshold2的边缘被认为是弱边缘。只有与强边缘相连的弱边缘才会被认为是真正的边缘。
	apertureSize(可选):Sobel算子的孔径大小,用于计算梯度。默认是3。
	L2gradient(可选):一个布尔值,指定是否使用L2范数来计算梯度幅度。如果为True,则使用L2范数;否则,使用L1范数(这更快)。默认是False。
返回值:
	edges:输出图像,包含检测到的边缘。

掩码操作

从边缘检测的结果看,仍然存在很多干扰区域,我们可以通过掩码操作除去这一部分的干扰。

在获取掩码的步骤中,首先构建一个全为0的掩码,即黑色部分。然后使用cv2,fillPoly来绘制掩码中值为255的白色部分。

python 复制代码
polygons = np.array([[(0, output.shape[0]), (0, 400), (output.shape[1]//2, 100), (output.shape[1], 400), (output.shape[1], output.shape[0])]])
mask = np.zeros_like(output)
cv2.fillPoly(mask, polygons, 255)

cv2.fillPoly用于在图像上绘制并填充多边形

python 复制代码
语法:
	cv2.fillPoly(img, pts, color[, lineType[, shift[, offset]]])
参数:
	img:原图像。
	pts:多边形的顶点坐标列表。其中每个元素是一个二维坐标点(即顶点的坐标)。
	color:填充颜色。这是一个三元组,表示 BGR 颜色。
	lineType:线条类型。默认值为 cv2.LINE_8, 可选参数:
		cv2.LINE_8(8-连通)
		cv2.LINE_4(4-连通)
		cv2.LINE_AA(反锯齿线条)。
	shift:点的精度。这是一个整数值,表示每个坐标点 (x,y) 的偏移量。默认值为 0。
	offset:绘制多边形的偏移量。这是一个元组,表示在 x 和 y 方向上的偏移量。默认值为 (0,0)。

获取到的掩码如下图

随后可以通过按位与运算对图像进行掩码操作

python 复制代码
masked_image = cv2.bitwise_and(output, mask)

操作后的结果

我们将以上两个步骤写成一个函数

python 复制代码
def mask_of_image(image):
    polygons = np.array([[(0, image.shape[0]), (0, 400), (image.shape[1]//2, 100), (image.shape[1], 400), (image.shape[1], image.shape[0])]])
    mask = np.zeros_like(image)
    cv2.fillPoly(mask, polygons, 255)
    masked_image = cv2.bitwise_and(image, mask)
    return masked_image

绘制到原图像

在这一步可以先获取车道线的直线部分,再将获取到的直线线段绘制到原图像

python 复制代码
mask_img = mask_of_image(output)

img = cv2.imread('road.png')
lines = cv2.HoughLinesP(mask_img, 1, np.pi/180, 5)
for line in lines:
    x1, y1, x2, y2 = line[0]
    cv2.line(img, (x1, y1), (x2, y2), (0, 255, 0), 4)

plt.imshow(img)

cv2.HoughLinesP 用于在二值图像中执行概率霍夫线变换(Probabilistic Hough Transform)以检测直线,能够返回检测到的直线的起点和终点的坐标。

以下是 cv2.HoughLinesP 的函数原型和参数说明:

python 复制代码
语法
	cv2.HoughLinesP(image, rho, theta, threshold, lines=None, minLineLength=None, maxLineGap=None)
参数:
	image:是通过边缘检测(如 Canny 边缘检测)处理后的图像。
	rho:参数 ρ 的步长(分辨率)。ρ 是从原点到直线的垂直距离(以像素为单位)。
	theta:参数 θ 的步长(分辨率)。θ 是直线的角度(以弧度为单位)。
		通常,可以使用较小的角度步长来提高角度的精度。
	threshold:累加器阈值。只有那些累加器值大于阈值的直线才会被返回。
		这个值越高,检测到的直线就越强。
	lines:一个可选参数,用于存储检测到的直线的端点坐标。
		它是一个 NumPy 数组,每行包含四个元素 [x1, y1, x2, y2],
		分别表示直线的起点和终点的坐标。
	minLineLength:可选参数,表示检测到的直线的最小长度。
		小于此长度的直线将被忽略。默认值为 None。
	maxLineGap:可选参数,表示在检测到的直线上被认为是同一部分的两个线段之间的最大距离。
		如果两个线段之间的间隔小于此值,则它们将被视为同一条直线。默认值为 None。
相关推荐
凤枭香6 小时前
Python OpenCV 傅里叶变换
开发语言·图像处理·python·opencv
ctrey_7 小时前
2024-11-4 学习人工智能的Day21 openCV(3)
人工智能·opencv·学习
可均可可8 小时前
C++之OpenCV入门到提高004:Mat 对象的使用
c++·opencv·mat·imread·imwrite
蒙娜丽宁8 小时前
《Python OpenCV从菜鸟到高手》——零基础进阶,开启图像处理与计算机视觉的大门!
python·opencv·计算机视觉
plmm烟酒僧9 小时前
Windows下QT调用MinGW编译的OpenCV
开发语言·windows·qt·opencv
AI小杨9 小时前
【车道线检测】一、传统车道线检测:基于霍夫变换的车道线检测史诗级详细教程
人工智能·opencv·计算机视觉·霍夫变换·车道线检测
冷凝女子10 小时前
【QT】海康视频及openCv抓拍正脸接口
qt·opencv·音视频·海康
撞南墙者12 小时前
OpenCV自学系列(1)——简介和GUI特征操作
人工智能·opencv·计算机视觉
柳鲲鹏13 小时前
OpenCV视频防抖源码及编译脚本
人工智能·opencv·计算机视觉
jndingxin13 小时前
OpenCV视觉分析之目标跟踪(8)目标跟踪函数CamShift()使用
人工智能·opencv·目标跟踪