一种交通执法影像中基于AI技术的车道线自动检测方法技术

本发明专利技术公开了一种交通执法影像中基于AI技术的车道线自动检测方法，属于图像识别领域。因此本发明专利技术方法提出一种只预测车道线两个端点的方法，并针对车道线的特殊性对端点特征点对组合分析，最终达到标注工作量小，要求低，预测速度快，预测准确的有益效果。本发明专利技术方法检测速度快、准确率高。该检测结果除应用在车辆违法变道检测(压实线检测)外，还可以应用在一些需要使用图像数据检测车道线位置的实际场景应用中如车道数判断等。

【技术实现步骤摘要】
一种交通执法影像中基于AI技术的车道线自动检测方法
本专利技术属于深度学习的目标检测领域，具体涉及到一种交通执法影像中基于AI技术的车道线自动检测方法。
技术介绍
车道线识别技术很早就有人开始研究，he/Rong等人使用Canny算子进行边缘检测，配合Hough变换方法进行车道线检测，时间复杂度较高，而且其车道线检测的适用范围小，主要是针对无人驾驶数据的车道线检测，而且其只关注左右两个车道。上述识别方法所针对的场景都是自动驾驶中的车道线检测。而近年来随着交通设施和违法抓拍系统的健全，在一定程度上减少了交通事故的发生。但是其违法数据都是由人工筛选，分类出违法、不违法数据。其数据量庞大，人工效率低，故而存在许多交通违法行为的二次检测方法。而在检测方法的前提中正确识别出车道线的所在位置至关重要。车道线识别的准确率直接影响违法判断的准确率。而原有的基于图像处理技术的车道线检测，在复杂的交通场景中受光照、天气、时间的影响，识别效果不佳。基于深度学习技术的车道线检测一般使用于自动驾驶，其只关注左右两个车道，而且数据标注工作量大，需耗费巨大的人力物力，因此也本文档来自技高网...

【技术保护点】
1.一种交通执法影像中基于AI技术的车道线自动检测方法，其特征在于，步骤如下：/nS1：获取交通执法设备所拍摄的含有路口处车道线的违法图像数据集，且数据集的每张图像中均预先对车道线的两个端点进行标注；/nS2：对违法图像数据集中的每张违法图像及其对应的标注数据，依次进行预处理和数据增强，得到相同大小的输入数据；/nS3：构建用于车道线检测的卷积神经网络模型，所述卷积神经网络模型以Hourglass网络作为基础网络结构，其输入为归一化处理后的违法图像，在模型卷积神经网络最后一层得到5个通道，每个通道各自输出一个数值矩阵，分别为车道线上端点热图、车道线下端点热图、车道线中心端点热图、车道线上端点偏...

【技术特征摘要】
1.一种交通执法影像中基于AI技术的车道线自动检测方法，其特征在于，步骤如下：
S1：获取交通执法设备所拍摄的含有路口处车道线的违法图像数据集，且数据集的每张图像中均预先对车道线的两个端点进行标注；
S2：对违法图像数据集中的每张违法图像及其对应的标注数据，依次进行预处理和数据增强，得到相同大小的输入数据；
S3：构建用于车道线检测的卷积神经网络模型，所述卷积神经网络模型以Hourglass网络作为基础网络结构，其输入为归一化处理后的违法图像，在模型卷积神经网络最后一层得到5个通道，每个通道各自输出一个数值矩阵，分别为车道线上端点热图、车道线下端点热图、车道线中心端点热图、车道线上端点偏移量和车道线下端点偏移量；所述卷积神经网络模型的总损失函数为5个通道的输出结果损失之和；
Hourglass网络的输出经过卷积和激活后，在卷积神经网络模型的最后一层得到5个通道，每个通道各自输出一个数值矩阵
S4：将S2的输入数据中的违法图像进行归一化处理后，转换为数值矩阵并输入所述卷积神经网络模型中，输出所述5个通道的数值矩阵；同时将S2的输入数据中的标注数据转换为所述5个通道输出结果的真值标签，一并输入所述卷积神经网络模型中；通过最小化总损失函数对卷积神经网络模型进行训练，直至模型收敛；
S5：将待检测的违法图像进行与S2中相同的预处理后，输入S4中训练完毕的卷积神经网络模型中，得到5个通道输出的数值矩阵，分别从车道线上端点热图、车道线下端点热图中提取符合车道线端点阈值范围的特征点，并将提取到的上端点特征点和下端点特征点两两组合得到若干特征点对，对所有特征点对进行异常值筛除，得到满足车道线分布规律的最终特征点对；以每一对最终特征点对作为车道线的两个端点，在图像中确定一条车道线。


2.如权利要求1所述的交通执法影像中基于AI技术的车道线自动检测方法，其特征在于，步骤S2中，所述预处理方法为：对每张违法图像进行灰度化处理，将其转换为灰度图像；然后将灰度图像进行二值化处理，使图像中的单个像素由0或255表示，形成黑白图像。


3.如权利要求1所述的交通执法影像中基于AI技术的车道线自动检测方法，其特征在于，步骤S2中，所述数据增强的方法为：将预处理后的违法图像缩放至512*512大小的图像中，并对其图像四周的空值区域进行填充；通过图像翻转、改变图像对比度或改变图像亮度进行调整，扩增新的图像样本。


4.如权利要求1所述的交通执法影像中基于AI技术的车道线自动检测方法，其特征在于，步骤S3中，输入所述卷积神经网络模型的违法图像，均需要预先除以255以进行归一化处理，保证数据值在0-1之间。


5.如权利要求1所述的交通执法影像中基于AI技术的车道线自动检测方法，其特征在于，步骤S3中，车道线上端点热图、车道线下端点热图和车道线中心端点热图的损失采用二分类交叉熵损失，车道线上端点偏移量和车道线下端点偏移量的损失采用...

【专利技术属性】
技术研发人员：李万清，刘俊，林永杰，袁友伟，
申请(专利权)人：杭州电子科技大学，
类型：发明
国别省市：浙江;33