车道线检测方法、装置、电子设备及存储介质转让专利

申请号 : CN201910536493.3

文献号 : CN110276293A

文献日 :

基本信息:

PDF:

法律信息:

相似专利:

发明人 : 潘杰

申请人 : 百度在线网络技术(北京)有限公司

摘要 :

本申请提出一种车道线检测方法、装置、电子设备及存储介质,属于计算机应用技术领域。其中,该方法包括:获取待检测的图像;将图像输入预设的目标检测模型,获取图像中各个网格的检测信息,检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值;对各个网格的检测信息进行非极大值抑制处理,获取图像中的各个车道中心点的位置以及对应的车道宽度;根据图像中的各个车道中心点的位置以及对应的车道宽度,确定图像中的车道线。由此,通过这种车道线检测方法,降低了噪声对车道线检测的干扰,提高了车道线检测的鲁棒性和准确度。

权利要求 :

1.一种车道线检测方法,其特征在于,包括:

获取待检测的图像;

将所述图像输入预设的目标检测模型,获取所述图像中各个网格的检测信息,所述检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值;

对所述各个网格的检测信息进行非极大值抑制处理,获取所述图像中的各个车道中心点的位置以及对应的车道宽度;

根据所述图像中的各个车道中心点的位置以及对应的车道宽度,确定所述图像中的车道线。

2.根据权利要求1所述的方法,其特征在于,所述对所述各个网格的检测信息进行非极大值抑制处理,获取所述图像中的各个车道中心点的位置以及对应的车道宽度,包括:针对所述图像中的每个网格,将所述网格中对应的车道中心点分数最大的预测框确定为所述网格对应的最优预测框;

针对每行网格,每隔预设步长选出分数大于阈值的最优预测框作为目标预测框;

针对每个目标预测框,根据所述目标预测框对应的车道中心点横向偏移以及预测框宽度调整值,确定所述图像中的一个车道中心点的位置以及对应的车道宽度。

3.根据权利要求2所述的方法,其特征在于,所述针对每个目标预测框,根据所述目标预测框对应的车道中心点横向偏移以及预测框宽度调整值,确定所述图像中的一个车道中心点的位置以及对应的车道宽度,包括:针对每个目标预测框,根据所述目标预测框对应的车道中心点横向偏移,以及所述目标预测框所属的网格的坐标,确定所述图像中的一个车道中心点的位置;

根据所述目标预测框对应的预测框宽度调整值,以及所述目标预测框的宽度,确定所述一个车道中心点对应的车道宽度。

4.根据权利要求1所述的方法,其特征在于,所述根据所述图像中的各个车道中心点的位置以及对应的车道宽度,确定所述图像中的车道线,包括:针对所述图像中的每个车道中心点,将所述车道中心点位置的横向坐标减去对应的车道宽度的一半,得到所述车道中心点对应的左侧车道线边界点的位置;

将所述车道中心点位置的横向坐标加上对应的车道宽度的一半,得到所述车道中心点对应的右侧车道线边界点的位置;

根据各个车道中心点对应的左侧车道线边界点的位置以及右侧车道线边界点的位置,确定所述图像中的车道线。

5.根据权利要求1所述的方法,其特征在于,所述目标检测模型包括:卷积部分和回归部分;

所述卷积部分,用于获取所述图像不同深度的底层特征,对不同深度的底层特征进行降维、反卷积以及联合卷积操作,得到所述图像对应的特征图;所述特征图中包括:所述图像中各个网格对应的特征点;

所述回归部分用于结合图像以及对应的特征图获取各个网格的检测信息。

6.根据权利要求1所述的方法,其特征在于,所述将所述图像输入预设的目标检测模型,获取所述图像中各个网格的检测信息之前,还包括:获取训练数据,所述训练数据包括:大于预设数量的图像,以及图像中各个真实车道线边界点的位置;

采用所述训练数据对初始的目标检测模型进行训练,直至所述目标检测模型的损失函数满足预设条件;所述损失函数根据图像中各个真实车道线边界点的位置以及图像中各个网格的检测信息确定。

7.一种车道线检测装置,其特征在于,包括:

获取模块,用于获取待检测的图像;

输入模块,用于将所述图像输入预设的目标检测模型,获取所述图像中各个网格的检测信息,所述检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值;

处理模块,用于对所述各个网格的检测信息进行非极大值抑制处理,获取所述图像中的各个车道中心点的位置以及对应的车道宽度;

确定模块,用于根据所述图像中的各个车道中心点的位置以及对应的车道宽度,确定所述图像中的车道线。

8.根据权利要求7所述的装置,其特征在于,所述处理模块具体用于,针对所述图像中的每个网格,将所述网格中对应的车道中心点分数最大的预测框确定为所述网格对应的最优预测框;

针对每行网格,每隔预设步长选出分数大于阈值的最优预测框作为目标预测框;

针对每个目标预测框,根据所述目标预测框对应的车道中心点横向偏移以及预测框宽度调整值,确定所述图像中的一个车道中心点的位置以及对应的车道宽度。

9.根据权利要求8所述的装置,其特征在于,所述处理模块具体用于,针对每个目标预测框,根据所述目标预测框对应的车道中心点横向偏移,以及所述目标预测框所属的网格的坐标,确定所述图像中的一个车道中心点的位置;

根据所述目标预测框对应的预测框宽度调整值,以及所述目标预测框的宽度,确定所述一个车道中心点对应的车道宽度。

10.根据权利要求7所述的装置,其特征在于,所述确定模块具体用于,针对所述图像中的每个车道中心点,将所述车道中心点位置的横向坐标减去对应的车道宽度的一半,得到所述车道中心点对应的左侧车道线边界点的位置;

将所述车道中心点位置的横向坐标加上对应的车道宽度的一半,得到所述车道中心点对应的右侧车道线边界点的位置;

根据各个车道中心点对应的左侧车道线边界点的位置以及右侧车道线边界点的位置,确定所述图像中的车道线。

11.根据权利要求7所述的装置,其特征在于,所述目标检测模型包括:卷积部分和回归部分;

所述卷积部分,用于获取所述图像不同深度的底层特征,对不同深度的底层特征进行降维、反卷积以及联合卷积操作,得到所述图像对应的特征图;所述特征图中包括:所述图像中各个网格对应的特征点;

所述回归部分用于结合图像以及对应的特征图获取各个网格的检测信息。

12.根据权利要求7所述的装置,其特征在于,还包括:训练模块;

所述获取模块,还用于获取训练数据,所述训练数据包括:大于预设数量的图像,以及图像中各个真实车道线边界点的位置;

所述训练模块,用于采用所述训练数据对初始的目标检测模型进行训练,直至所述目标检测模型的损失函数满足预设条件;所述损失函数根据图像中各个真实车道线边界点的位置以及图像中各个网格的检测信息确定。

13.一种电子设备,其特征在于,包括:

存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序,其特征在于,所述处理器执行所述程序时实现如权利要求1-6任一所述的车道线检测方法。

14.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,该程序被处理器执行时实现如权利要求1-6任一所述的车道线检测方法。

15.一种计算机程序产品,当所述计算机程序产品中的指令处理器执行时,实现如权利要求1-6任一所述的车道线检测方法。

说明书 :

车道线检测方法、装置、电子设备及存储介质

技术领域

[0001] 本申请涉及计算机应用技术领域,尤其涉及一种车道线检测方法、装置、电子设备及存储介质。

背景技术

[0002] 目前的车道线检测方法主要有两种。一种是提取车道线颜色、亮度等特征,进而进行边缘检测,利用霍夫变换提取直线,该方法容易受到其他颜色和亮度的干扰,且对噪声比较敏感,例如车道线模糊,光线暗等,鲁棒性差,检测准确度差。
[0003] 另一种方法是利用神经网络进行车道线分割,提取分割后的车道线区域的中心线作为车道线检测结果。该方法在过曝、阴雨天气、车道线边界磨损情况下难以进行有效分割,且容易将箭头、刹车痕等干扰误检为车道线,准确度差。

发明内容

[0004] 本申请提出的车道线检测方法、装置、电子设备及存储介质,用于解决相关技术中的车道线检测方法,不仅容易受到噪声干扰,鲁棒性差,而且容易产生误检,准确度差的问题。
[0005] 本申请一方面实施例提出的车道线检测方法,包括:获取待检测的图像;将所述图像输入预设的目标检测模型,获取所述图像中各个网格的检测信息,所述检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值;对所述各个网格的检测信息进行非极大值抑制处理,获取所述图像中的各个车道中心点的位置以及对应的车道宽度;根据所述图像中的各个车道中心点的位置以及对应的车道宽度,确定所述图像中的车道线。
[0006] 本申请另一方面实施例提出的车道线检测装置,包括:获取模块,用于获取待检测的图像;输入模块,用于将所述图像输入预设的目标检测模型,获取所述图像中各个网格的检测信息,所述检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值;处理模块,用于对所述各个网格的检测信息进行非极大值抑制处理,获取所述图像中的各个车道中心点的位置以及对应的车道宽度;确定模块,用于根据所述图像中的各个车道中心点的位置以及对应的车道宽度,确定所述图像中的车道线。
[0007] 本申请再一方面实施例提出的电子设备,其包括:存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序,其特征在于,所述处理器执行所述程序时实现如前所述的车道线检测方法。
[0008] 本申请再一方面实施例提出的计算机可读存储介质,其上存储有计算机程序,其特征在于,所述程序被处理器执行时实现如前所述的车道线检测方法。
[0009] 本申请又一方面实施例提出的计算机程序,该程序被处理器执行时,以实现本申请实施例所述的车道线检测方法。
[0010] 本申请实施例提供的车道线检测方法、装置、电子设备、计算机可读存储介质及计算机程序,可以将获取的待检测的图像输入预设的目标检测模型,以获取图像中各个网格的检测信息,检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值,并对各个网格的检测信息进行非极大值抑制处理,获取图像中的各个车道中心点的位置以及对应的车道宽度,进而根据图像中的各个车道中心点的位置以及对应的车道宽度,确定图像中的车道线。由此,通过将待检测的图像划分为多个网格,并利用训练好的目标检测模型检测每个网格中包括的车道中心点及车道宽度,之后即可根据检测的多个车道中心点及车道宽度,确定出图像中的车道线,从而降低了噪声对车道线检测的干扰,提高了车道线检测的鲁棒性和准确度。
[0011] 本申请附加的方面和优点将在下面的描述中部分给出,部分将从下面的描述中变得明显,或通过本申请的实践了解到。

附图说明

[0012] 本申请上述的和/或附加的方面和优点从下面结合附图对实施例的描述中将变得明显和容易理解,其中:
[0013] 图1为本申请实施例所提供的一种车道线检测方法的流程示意图;
[0014] 图2为本申请实施例所提供的另一种车道线检测方法的流程示意图;
[0015] 图3为本申请实施例所提供的一种车道线检测装置的结构示意图;
[0016] 图4为本申请实施例所提供的电子设备的结构示意图。

具体实施方式

[0017] 下面详细描述本申请的实施例,所述实施例的示例在附图中示出,其中自始至终相同或类似的标号表示相同或类似的要素。下面通过参考附图描述的实施例是示例性的,旨在用于解释本申请,而不能理解为对本申请的限制。
[0018] 本申请实施例针对相关技术中的车道线检测方法,不仅容易受到噪声干扰,鲁棒性差,而且容易产生误检,准确度差的问题,提出一种车道线检测方法。
[0019] 本申请实施例提供的车道线检测方法,可以将获取的待检测的图像输入预设的目标检测模型,以获取图像中各个网格的检测信息,检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值,并对各个网格的检测信息进行非极大值抑制处理,获取图像中的各个车道中心点的位置以及对应的车道宽度,进而根据图像中的各个车道中心点的位置以及对应的车道宽度,确定图像中的车道线。由此,通过将待检测的图像划分为多个网格,并利用训练好的目标检测模型检测每个网格中包括的车道中心点及车道宽度,之后即可根据检测的多个车道中心点及车道宽度,确定出图像中的车道线,从而降低了噪声对车道线检测的干扰,提高了车道线检测的鲁棒性和准确度。
[0020] 下面参考附图对本申请提供的车道线检测方法、装置、电子设备、存储介质及计算机程序进行详细描述。
[0021] 图1为本申请实施例所提供的一种车道线检测方法的流程示意图。
[0022] 如图1所示,该车道线检测方法,包括以下步骤:
[0023] 步骤101,获取待检测的图像。
[0024] 需要说明的是,本申请实施例的车道线检测方法,可以由本申请所提供的车道线检测装置执行。实际使用时,本申请实施例所提供的车道线检测方法可以应用于自动驾驶领域,为自动驾驶车辆提供路况信息,从而本申请实施例的车道线检测装置可以配置在任意车辆中,以执行本申请的车道线检测方法。
[0025] 在本申请实施例中,可以根据具体的应用场景,确定待检测的图像的获取方式。比如,当本申请实施例的车道线检测装置应用在自动驾驶车辆中时,可以获取自动驾驶车辆中的摄像头采集的车辆前方的路面信息,作为待检测的图像。具体的,车道线检测装置可以直接与摄像头建立通信连接,以直接获取摄像头采集的实时图像;或者,摄像头可以将采集的图像存储在车辆的存储设备中,从而车道线检测装置也可以从车辆的存储设备中获取待检测的图像。
[0026] 步骤102,将所述图像输入预设的目标检测模型,获取所述图像中各个网格的检测信息,所述检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值。
[0027] 其中,预设的目标检测模型,可以是提前训练完成的,比如可以是You Only Look Once:Unified,Real-Time Object Detection V2(Yolo V2)算法模型、Single Shot MultiBox Detector(SSD)算法模型等one-stage目标检测模型,但不仅限于此。
[0028] 其中,预测框,是预设的目标检测模型中定义的,其具有一定的尺寸和位置,与待检测的图像和图像中的网格没有直接关联,其是对图像进行目标检测的工具。实际使用时,预测框的数量及初始尺寸和位置等,可以根据所需的预测精度、计算复杂度等实际需要预设,本申请实施例对此做限定,比如,预测框的数量可以是5个。
[0029] 车道中心点横向偏移,是指车道中心点与其所在网格的左上角坐标之间的横向偏移;车道中心点分数,是指预测框对应的车道中心点的置信度,可以反映出预测框对应的车道中心点的可靠性;预测框宽度调整值,用于对预测框的宽度进行调整,以得到预测框当前的宽度值。
[0030] 优选的,由于在本申请实施例的目标检测模型用于对车道中心点以及车道宽度进行检测,因此可以将预测框定义为具有一定位置和宽度的线段,从而检测信息中可以仅包括预测框的宽度调整值,以对预测框的宽度进行调整。
[0031] 在本申请实施例中,可以首先将待检测的图像划分为多个网格,并将待检测的图像输入预设的目标检测模型,通过预设的目标检测模型的卷积部分获得图像的特征图,其中,特征图中的每个点对应图像中的一个网格。之后根据获得的特征图以及待检测图像,通过预设的目标检测模型的回归部分,获取图像中各个网格的检测信息。
[0032] 需要说明的是,图像中的每个网格用于预测中心在该网格中的目标。实际使用时,网格的尺寸可以根据实际需要预设,本申请实施例对此不做限定。比如,待检测的图像的尺寸为1920×640像素,将待检测图像划分为多个尺寸为16×16像素的网格,即获得的特征图的尺寸为120×40像素。
[0033] 步骤103,对所述各个网格的检测信息进行非极大值抑制处理,获取所述图像中的各个车道中心点的位置以及对应的车道宽度。
[0034] 在本申请实施例中,通过预先设置多个预测框对图像中每个网格中的目标(即车道中心点)进行检测,以保证车道线检测的准确性。而由于多个预测框的尺寸不同,使得每个预测框对应的检测信息的准确度不同,从而可以根据各个网格的检测信息,确定出每个网格对应的准确度最高的预测框,进而根据每个网格对应的准确度最高的预测框对应的车道中心点横向偏移、预测框宽度调整值等,确定出各个网格中分别包括的车道中心点的位置以及每个车道中心点的位置对应的车道宽度,即图像中的各个车道中心点的位置以及对应的车道宽度。
[0035] 具体的,可以通过对各个网格的检测信息进行非极大值抑制处理,确定每个网格对应的准确度最高的预测框,从而确定出图像中的各个车道中心点的位置以及对应的车道宽度。即在本申请实施例一种可能的实现形式中,上述步骤103,可以包括:
[0036] 针对所述图像中的每个网格,将所述网格中对应的车道中心点分数最大的预测框确定为所述网格对应的最优预测框;
[0037] 针对每行网格,每隔预设步长选出分数大于阈值的最优预测框作为目标预测框;
[0038] 针对每个目标预测框,根据所述目标预测框对应的车道中心点横向偏移以及预测框宽度调整值,确定所述图像中的一个车道中心点的位置以及对应的车道宽度。
[0039] 在本申请实施例中,预测框对应的车道线中心点分数,可以反映出预测框对车道中心点预测的准确度,从而可以根据各个网格对应的各预测框对应的车道中心点分数,确定各个网格对应的最优预测框。具体的,由于预测框对应的车道线中心点分数越大,则预测框对应的车道中心点横向偏移越准确,从而可以将每个网格中对应的车道中心点分数最大的预测框,确定为每个网格对应的最优预测框。
[0040] 在确定出图像中的每个网格对应的最优预测框之后,可以根据预设的步长从每行网格对应的最优预测框中,选取出每行网络对应的目标预测框。具体的,可以每隔预设步长,将对应的车道中心点分数大于阈值的最优预测框,确定为目标预测框。
[0041] 举例来说,待检测的图像的尺寸为1920×640像素,每个网格的尺寸为16×16像素,即待检测的图像包括120×40个网格,预设步长为160像素,即在每行网格中,每160像素判断一次这160像素对应的网格对应的最优预测框中是否包含车道中心点分数大于阈值的最优预测框,即每10个网格判断一次这10个网格对应的最优预测框中是否包含车道中心点分数大于阈值的最优预测框,若包含,则将车道中心点分数大于阈值的最优预测框确定为目标预测框。
[0042] 需要说明的是,上述举例仅为示例性的,不能视为对本申请的限制。实际使用时,可以根据实际需要预设步长与阈值,本申请实施例对此不做限定。
[0043] 在本申请实施例中,确定出目标预测框之后,则可以根据每个目标预测框对应的车道中心点横向偏移以及预测框宽度调整值,确定出每个目标预测框对应的车道中心的位置以及对应的车道宽度,从而确定出图像中所有的车道中心点的位置以及对应的车道宽度,即每个目标预测框对应图像中的一个车道中心点。
[0044] 具体的,根据目标预测框对应的车道中心点横向偏移以及预测框宽度调整值,确定图像中的一个车道中心点的位置以及对应的车道宽度,包括以下步骤:
[0045] 针对每个目标预测框,根据所述目标预测框对应的车道中心点横向偏移,以及所述目标预测框所属的网格的坐标,确定所述图像中的一个车道中心点的位置;
[0046] 根据所述目标预测框对应的预测框宽度调整值,以及所述目标预测框的宽度,确定所述一个车道中心点对应的车道宽度。
[0047] 在本申请实施例中,由于预测框对应的车道中心点横向偏移,是指车道中心点的横坐标相对于预测框所属网格左上角横坐标之间的差值,从而可以首先根据目标预测框的位置,确定出其所属的网格的左上角的坐标,进而根据目标预测框对应的车道中心点横向偏移,确定出目标预测框对应的车道中心点的坐标,即图像中一个车道中心点的位置。
[0048] 需要说明的是,在对预设的目标检测模型进行训练时,可以将训练数据中的车道宽度作为预测框的宽度,从而在对车道线进行检测时,可以将预测框的当前宽度确定为车道宽度。因此,可以根据目标预测框对应的预测框宽度调整值,以及目标预测框的宽度,确定目标预测框对应的车道中心点对应的车道宽度。具体的,可以将目标预测框的宽度与目标预测框对应的预测框宽度调整值之和,确定为目标预测框对应的一个车道中心点对应的车道宽度。
[0049] 步骤104,根据所述图像中的各个车道中心点的位置以及对应的车道宽度,确定所述图像中的车道线。
[0050] 在本申请实施例中,确定出图像中包括的各个车道中心点的位置以及各个车道中心点对应的车道宽度之后,则可以根据各个车道中心点的位置以及对应的车道宽度,确定出车道线上包括的各个点,从而确定出图像中的车道线。
[0051] 具体的,对一条车道通常包括左侧车道线和右侧车道线,从而可以根据各车道中心点的位置及其对应的车道宽度,分别确定出左右两侧的车道线。即在本申请实施例一种可能的实现形式中,上述步骤104,可以包括:
[0052] 针对所述图像中的每个车道中心点,将所述车道中心点位置的横向坐标减去对应的车道宽度的一半,得到所述车道中心点对应的左侧车道线边界点的位置;
[0053] 将所述车道中心点位置的横向坐标加上对应的车道宽度的一半,得到所述车道中心点对应的右侧车道线边界点的位置;
[0054] 根据各个车道中心点对应的左侧车道线边界点的位置以及右侧车道线边界点的位置,确定所述图像中的车道线。
[0055] 可以理解的是,图像中的各个车道中心点与其对应的车道线之间的距离为车道宽度的一半,即竖向坐标与车道中心点位置的竖向坐标相同、且横向坐标与车道中心点位置的横向坐标的差值为车道宽度的一半的点,位于车道中心点对应的车道线上。
[0056] 在本申请实施例中,可以将车道中心点位置的横向坐标减去其对应的车道宽度的一半,确定出车道中心点对应的左侧车道线边界点的横向坐标,并将车道中心点位置的竖向坐标作为车道纵向对应的左侧车道线边界点的竖向坐标,从而确定出车道中心点对应的左侧车道线边界点的位置;相应的,可以将车道中心点位置的横向坐标加上其对应的车道宽度的一半,确定出车道中心点对应的右侧车道线边界点的横向坐标,并将车道中心点位置的竖向坐标作为车道纵向对应的右侧车道线边界点的竖向坐标,从而确定出车道中心点对应的右侧车道线边界点的位置。
[0057] 可以理解的是,各个车道中心点对应的左侧车道线边界点的连线即为车道的左侧车道线,各个车道中心点对应的右侧车道线边界点的连线即为车道的右侧车道线。从而在确定出各个车道中心点对应的左侧车道线边界点的位置以及右侧车道线边界点的位置之后,可以根据各个左侧车道线边界点的位置,确定出各个左侧车道线边界点所在的直线,即左侧车道线;根据各个右侧车道线边界点的位置,确定出各个右侧车道线边界点所在的直线,即右侧车道线,从而确定出图像中的车道线。
[0058] 本申请实施例提供的车道线检测方法,可以将获取的待检测的图像输入预设的目标检测模型,以获取图像中各个网格的检测信息,检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值,并对各个网格的检测信息进行非极大值抑制处理,获取图像中的各个车道中心点的位置以及对应的车道宽度,进而根据图像中的各个车道中心点的位置以及对应的车道宽度,确定图像中的车道线。由此,通过将待检测的图像划分为多个网格,并利用训练好的目标检测模型检测每个网格中包括的车道中心点及车道宽度,之后即可根据检测的多个车道中心点及车道宽度,确定出图像中的车道线,从而降低了噪声对车道线检测的干扰,提高了车道线检测的鲁棒性和准确度。
[0059] 在本申请一种可能的实现形式中,预设的目标检测模型可以是通过大量训练数据进行训练得到,并通过损失函数不断优化目标检测模型的性能,以使得预设的目标检测模型的性能符合实际的应用需求。
[0060] 下面结合图2,对本申请实施例提供的车道线检测方法进行进一步说明。
[0061] 图2为本申请实施例所提供的另一种车道线检测方法的流程示意图。
[0062] 如图2所示,该车道线检测方法,包括以下步骤:
[0063] 步骤201,获取训练数据,所述训练数据包括:大于预设数量的图像,以及图像中各个真实车道线边界点的位置。
[0064] 其中,训练数据,通过可以包括大量图像数据以及对每个图像数据的标注信息。需要说明的是,训练数据中包括的图像数据以及对图像数据的标注信息,与目标检测模型的具体用途有关。比如,若目标检测模型的用途为人脸检测,则训练数据中可以包括大量包含人脸的图像,以及对图像中人脸的标注信息;又如本申请实施例的目标检测模型的用途为车道线检测,则训练数据中可以包括大量包含车道线的图像,以及对图像中真实车道线边界点的位置的标注信息。
[0065] 需要说明的是,为保证最终获得的目标检测模型的精确性,训练数据需要具有一定规模,从而可以提前预设训练数据中包括的图像数量,在获取训练数据时,训练数据中包括的图像数量必须大于预设数量,以保证目标检测模型的性能。实际使用时,训练数据中包括的图像数量,可以根据实际需要预设,本申请实施例对此不做限定。
[0066] 在本申请实施例中,获取训练数据的途径有多种,比如,可以从网络中搜集包含车道线的图像,或者可以将在实际的应用场景中(比如自动驾驶场景)采集的图像数据作为训练数据,并在获取到图像数据之后对图像数据进行标注,以获得图像中各个真实车道线边界点的位置。
[0067] 步骤202,采用所述训练数据对初始的目标检测模型进行训练,直至所述目标检测模型的损失函数满足预设条件;所述损失函数根据图像中各个真实车道线边界点的位置以及图像中各个网格的检测信息确定。
[0068] 在本申请实施例中,可以采用训练数据对初始的目标检测模型进行训练,即将训练数据中的图像数据依次输入初始的目标检测模型,以得到每个图像数据对应的检测信息,进而根据每个图像数据对应的检测信息以及每个图像数据对应的各个真实车道线边界点的位置,损失函数的当前取值,若损失函数的当前取值满足预设条件,则可以确定目标检测模型当前的性能满足要求,从而可以结束对目标检测模型的训练;若损失函数的当前取值不满足预设条件,则可以确定目标检测模型当前的性能不满足要求,从而可以对目标检测模型的参数进行优化,并继续利用训练数据对参数优化后的目标检测模型进行训练,直至目标检测模型的损失函数满足预设条件。
[0069] 需要说明的是,损失函数的取值越小,则说明目标检测模型输出的检测信息与真实的车道线边界点的位置越接近,即目标检测模型的性能越好,因此,目标检测模型的损失函数需要满足的预设条件,可以是损失函数的取值小于预设阈值。实际使用时,损失函数需要满足的预设条件,可以根据实际需要预设,本申请实施例对此不做限定。
[0070] 优选的,在本申请实施例中,对目标检测模型进行训练时,可以对车道中心点横向偏移、车道中心点分数以及车道宽度三个部分进行回归,即目标检测模型的损失函数可以分为三个部分,以分别对车道中心点横向偏移、车道中心点分数以及车道宽度三个部分的损失分别进行惩罚,从而可以进一步提高最终获得的目标检测模型的精确性。可选的,可以采用L2范数损失函数对车道中心点横向偏移、车道中心点分数进行回归,采用L1smooth损失函数对车道宽度进行回归。实际使用时,可以根据实际需要选择各部分对应的损失函数,本申请实施例对此不做限定。
[0071] 需要说明的是,在目标检测模型的损失函数分为三个部分时,可以在损失函数的三个部分分别满足预设条件时,完成对目标检测模型的训练;或者,也可以在损失函数的三个部分的取值之和满足预设条件时,完成对目标检测模型的训练,本申请实施例对此不做限定。
[0072] 步骤203,获取待检测的图像,并将所述图像输入预设的目标检测模型,获取所述图像中各个网格的检测信息,所述检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值。
[0073] 在本申请实施例中,预设的目标检测模型,可以包括卷积部分和回归部分,将待检测的图像输入预设的目标检测模型,可以通过预设的目标检测模型的卷积部分获得图像的特征图,其中,特征图中的每个点对应图像中的一个网格。之后根据获得的特征图以及待检测图像,通过预设的目标检测模型的回归部分,获取图像中各个网格的检测信息。
[0074] 进一步的,本申请实施例的目标检测模型可以将图像的浅层特征和深层特征结合,以提取更有效的结构特征,从而提高了目标检测模型的准确度。即在本申请实施例一种可能的实现形式中,上述卷积部分,用于获取所述图像不同深度的底层特征,对不同深度的底层特征进行降维、反卷积以及联合卷积操作,得到所述图像对应的特征图;所述特征图中包括:所述图像中各个网格对应的特征点;
[0075] 上述回归部分用于结合图像以及对应的特征图获取各个网格的检测信息。
[0076] 需要说明的是,本申请实施例的目标检测模型所使用的神经网络模型,可以包括多个卷积层,从而可以通过卷积部分的多个卷积层对图像进行不同深度的卷积操作,以获得图像对应的图像对应的不同深度的底层特征,其中,底层特征的深度不同,其对应的特征图的尺寸也不同。比如,底层特征conv5_5的特征图的尺寸为图像的1/32,底层特征conv6_5的特征图的尺寸为图像的1/64,底层特征conv7_5的特征图的尺寸为图像的1/128。
[0077] 在获取到图像对应的不同深度的底层特征之后,可以对不同深度的底层特征进行降维,比如可以通过1×1卷积核对不同深度的底层特征进行卷积操作,从而获得对不同深度的底层特征进行降维之后的特征图,之后对降维后的不同深度的底层特征进行不同深度的反卷积操作,以使得降维后的不同深度的底层特征具有相同的尺寸,即使得降维后的不同深度的底层特征的尺寸与图像中包括的网格数量相同。比如,图像中网格的尺寸为16×16,则对降维后的不同深度的底层特征进行不同深度的反卷积操作后,获得的特征图的尺寸均为图像的1/16,。最后对进行不同深度的反卷积操作之后的特征图进行联合卷积操作,从而获得图像对应的特征图,且特征图中的每个特征点与图像中的一个网格对应。
[0078] 步骤204,对所述各个网格的检测信息进行非极大值抑制处理,获取所述图像中的各个车道中心点的位置以及对应的车道宽度。
[0079] 步骤205,根据所述图像中的各个车道中心点的位置以及对应的车道宽度,确定所述图像中的车道线。
[0080] 上述步骤204-205的具体实现过程及原理,可以参照上述实施例的详细描述,此处不再赘述。
[0081] 本申请实施例提供的车道线检测方法,可以采用获取的训练数据对初始的目标检测模型进行训练,直至目标检测模型的损失函数满足预设条件,其中,损失函数根据图像中各个真实车道线边界点的位置以及图像中各个网格的检测信息确定,并将获取的待检测的图像输入预设的目标检测模型,以获取图像中各个网格的检测信息,检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值,以及对各个网格的检测信息进行非极大值抑制处理,获取图像中的各个车道中心点的位置以及对应的车道宽度,进而根据图像中的各个车道中心点的位置以及对应的车道宽度,确定图像中的车道线。由此,大量训练数据对初始的目标检测模型进行训练,并利用训练好的目标检测模型检测图像中每个网格包括的车道中心点及车道宽度,从而不仅降低了噪声对车道线检测的干扰,提高了车道线检测的鲁棒性和准确度,而且进一步优化了目标检测模型的性能。
[0082] 为了实现上述实施例,本申请还提出一种车道线检测装置。
[0083] 图3为本申请实施例提供的一种车道线检测装置的结构示意图。
[0084] 如图3所示,该车道线检测装置30,包括:
[0085] 获取模块31,用于获取待检测的图像;
[0086] 输入模块32,用于将所述图像输入预设的目标检测模型,获取所述图像中各个网格的检测信息,所述检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值;
[0087] 处理模块33,用于对所述各个网格的检测信息进行非极大值抑制处理,获取所述图像中的各个车道中心点的位置以及对应的车道宽度;
[0088] 确定模块34,用于根据所述图像中的各个车道中心点的位置以及对应的车道宽度,确定所述图像中的车道线。
[0089] 在实际使用时,本申请实施例提供的车道线检测装置,可以被配置在任意电子设备中,以执行前述车道线检测方法。
[0090] 本申请实施例提供的车道线检测装置,可以将获取的待检测的图像输入预设的目标检测模型,以获取图像中各个网格的检测信息,检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值,并对各个网格的检测信息进行非极大值抑制处理,获取图像中的各个车道中心点的位置以及对应的车道宽度,进而根据图像中的各个车道中心点的位置以及对应的车道宽度,确定图像中的车道线。由此,通过将待检测的图像划分为多个网格,并利用训练好的目标检测模型检测每个网格中包括的车道中心点及车道宽度,之后即可根据检测的多个车道中心点及车道宽度,确定出图像中的车道线,从而降低了噪声对车道线检测的干扰,提高了车道线检测的鲁棒性和准确度。
[0091] 在本申请一种可能的实现形式中,上述处理模块33,具体用于:
[0092] 针对所述图像中的每个网格,将所述网格中对应的车道中心点分数最大的预测框确定为所述网格对应的最优预测框;
[0093] 针对每行网格,每隔预设步长选出分数大于阈值的最优预测框作为目标预测框;
[0094] 针对每个目标预测框,根据所述目标预测框对应的车道中心点横向偏移以及预测框宽度调整值,确定所述图像中的一个车道中心点的位置以及对应的车道宽度。
[0095] 进一步的,在本申请另一种可能的实现形式中,上述处理模块33,还用于:
[0096] 针对每个目标预测框,根据所述目标预测框对应的车道中心点横向偏移,以及所述目标预测框所属的网格的坐标,确定所述图像中的一个车道中心点的位置;
[0097] 根据所述目标预测框对应的预测框宽度调整值,以及所述目标预测框的宽度,确定所述一个车道中心点对应的车道宽度。
[0098] 在本申请一种可能的实现形式中,上述确定模块34,具体用于:
[0099] 针对所述图像中的每个车道中心点,将所述车道中心点位置的横向坐标减去对应的车道宽度的一半,得到所述车道中心点对应的左侧车道线边界点的位置;
[0100] 将所述车道中心点位置的横向坐标加上对应的车道宽度的一半,得到所述车道中心点对应的右侧车道线边界点的位置;
[0101] 根据各个车道中心点对应的左侧车道线边界点的位置以及右侧车道线边界点的位置,确定所述图像中的车道线。
[0102] 在本申请一种可能的实现形式中,上述目标检测模型包括:卷积部分和回归部分;
[0103] 所述卷积部分,用于获取所述图像不同深度的底层特征,对不同深度的底层特征进行降维、反卷积以及联合卷积操作,得到所述图像对应的特征图;所述特征图中包括:所述图像中各个网格对应的特征点;
[0104] 所述回归部分用于结合图像以及对应的特征图获取各个网格的检测信息。
[0105] 进一步的,在本申请另一种可能的实现形式中,上述车道线检测装置30,还包括:训练模块;
[0106] 相应的,上述获取模块31,还用于获取训练数据,所述训练数据包括:大于预设数量的图像,以及图像中各个真实车道线边界点的位置;
[0107] 上述训练模块,具体用于采用所述训练数据对初始的目标检测模型进行训练,直至所述目标检测模型的损失函数满足预设条件;所述损失函数根据图像中各个真实车道线边界点的位置以及图像中各个网格的检测信息确定。
[0108] 需要说明的是,前述对图1、图2所示的车道线检测方法实施例的解释说明也适用于该实施例的车道线检测装置30,此处不再赘述。
[0109] 本申请实施例提供的车道线检测装置,可以采用获取的训练数据对初始的目标检测模型进行训练,直至目标检测模型的损失函数满足预设条件,其中,损失函数根据图像中各个真实车道线边界点的位置以及图像中各个网格的检测信息确定,并将获取的待检测的图像输入预设的目标检测模型,以获取图像中各个网格的检测信息,检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值,以及对各个网格的检测信息进行非极大值抑制处理,获取图像中的各个车道中心点的位置以及对应的车道宽度,进而根据图像中的各个车道中心点的位置以及对应的车道宽度,确定图像中的车道线。由此,大量训练数据对初始的目标检测模型进行训练,并利用训练好的目标检测模型检测图像中每个网格包括的车道中心点及车道宽度,从而不仅降低了噪声对车道线检测的干扰,提高了车道线检测的鲁棒性和准确度,而且进一步优化了目标检测模型的性能。
[0110] 为了实现上述实施例,本申请还提出一种电子设备。
[0111] 图4为本申请一个实施例的电子设备的结构示意图。
[0112] 如图4所示,上述电子设备200包括:
[0113] 存储器210及处理器220,连接不同组件(包括存储器210和处理器220)的总线230,存储器210存储有计算机程序,当处理器220执行所述程序时实现本申请实施例所述的车道线检测方法。
[0114] 总线230表示几类总线结构中的一种或多种,包括存储器总线或者存储器控制器,外围总线,图形加速端口,处理器或者使用多种总线结构中的任意总线结构的局域总线。举例来说,这些体系结构包括但不限于工业标准体系结构(ISA)总线,微通道体系结构(MAC)总线,增强型ISA总线、视频电子标准协会(VESA)局域总线以及外围组件互连(PCI)总线。
[0115] 电子设备200典型地包括多种电子设备可读介质。这些介质可以是任何能够被电子设备200访问的可用介质,包括易失性和非易失性介质,可移动的和不可移动的介质。
[0116] 存储器210还可以包括易失性存储器形式的计算机系统可读介质,例如随机存取存储器(RAM)240和/或高速缓存存储器250。电子设备200可以进一步包括其它可移动/不可移动的、易失性/非易失性计算机系统存储介质。仅作为举例,存储系统260可以用于读写不可移动的、非易失性磁介质(图4未显示,通常称为“硬盘驱动器”)。尽管图4中未示出,可以提供用于对可移动非易失性磁盘(例如“软盘”)读写的磁盘驱动器,以及对可移动非易失性光盘(例如CD-ROM,DVD-ROM或者其它光介质)读写的光盘驱动器。在这些情况下,每个驱动器可以通过一个或者多个数据介质接口与总线230相连。存储器210可以包括至少一个程序产品,该程序产品具有一组(例如至少一个)程序模块,这些程序模块被配置以执行本申请各实施例的功能。
[0117] 具有一组(至少一个)程序模块270的程序/实用工具280,可以存储在例如存储器210中,这样的程序模块270包括——但不限于——操作系统、一个或者多个应用程序、其它程序模块以及程序数据,这些示例中的每一个或某种组合中可能包括网络环境的实现。程序模块270通常执行本申请所描述的实施例中的功能和/或方法。
[0118] 电子设备200也可以与一个或多个外部设备290(例如键盘、指向设备、显示器291等)通信,还可与一个或者多个使得用户能与该电子设备200交互的设备通信,和/或与使得该电子设备200能与一个或多个其它计算设备进行通信的任何设备(例如网卡,调制解调器等等)通信。这种通信可以通过输入/输出(I/O)接口292进行。并且,电子设备200还可以通过网络适配器293与一个或者多个网络(例如局域网(LAN),广域网(WAN)和/或公共网络,例如因特网)通信。如图所示,网络适配器293通过总线230与电子设备200的其它模块通信。应当明白,尽管图中未示出,可以结合电子设备200使用其它硬件和/或软件模块,包括但不限于:微代码、设备驱动器、冗余处理单元、外部磁盘驱动阵列、RAID系统、磁带驱动器以及数据备份存储系统等。
[0119] 处理器220通过运行存储在存储器210中的程序,从而执行各种功能应用以及数据处理。
[0120] 需要说明的是,本实施例的电子设备的实施过程和技术原理参见前述对本申请实施例的车道线检测方法的解释说明,此处不再赘述。
[0121] 本申请实施例提供的电子设备,可以执行如前所述的车道线检测方法,将获取的待检测的图像输入预设的目标检测模型,以获取图像中各个网格的检测信息,检测信息包括:各个预测框对应的车道中心点横向偏移、车道中心点分数、预测框宽度调整值,并对各个网格的检测信息进行非极大值抑制处理,获取图像中的各个车道中心点的位置以及对应的车道宽度,进而根据图像中的各个车道中心点的位置以及对应的车道宽度,确定图像中的车道线。由此,通过将待检测的图像划分为多个网格,并利用训练好的目标检测模型检测每个网格中包括的车道中心点及车道宽度,之后即可根据检测的多个车道中心点及车道宽度,确定出图像中的车道线,从而降低了噪声对车道线检测的干扰,提高了车道线检测的鲁棒性和准确度。
[0122] 为了实现上述实施例,本申请还提出一种计算机可读存储介质。
[0123] 其中,该计算机可读存储介质,其上存储有计算机程序,该程序被处理器执行时,以实现本申请实施例所述的车道线检测方法。
[0124] 为了实现上述实施例,本申请再一方面实施例提供一种计算机程序,该程序被处理器执行时,以实现本申请实施例所述的车道线检测方法。
[0125] 一种可选实现形式中,本实施例可以采用一个或多个计算机可读的介质的任意组合。计算机可读介质可以是计算机可读信号介质或者计算机可读存储介质。计算机可读存储介质例如可以是——但不限于——电、磁、光、电磁、红外线、或半导体的系统、装置或器件,或者任意以上的组合。计算机可读存储介质的更具体的例子(非穷举的列表)包括:具有一个或多个导线的电连接、便携式计算机磁盘、硬盘、随机存取存储器(RAM)、只读存储器(ROM)、可擦式可编程只读存储器(EPROM或闪存)、光纤、便携式紧凑磁盘只读存储器(CD-ROM)、光存储器件、磁存储器件、或者上述的任意合适的组合。在本文件中,计算机可读存储介质可以是任何包含或存储程序的有形介质,该程序可以被指令执行系统、装置或者器件使用或者与其结合使用。
[0126] 计算机可读的信号介质可以包括在基带中或者作为载波一部分传播的数据信号,其中承载了计算机可读的程序代码。这种传播的数据信号可以采用多种形式,包括——但不限于——电磁信号、光信号或上述的任意合适的组合。计算机可读的信号介质还可以是计算机可读存储介质以外的任何计算机可读介质,该计算机可读介质可以发送、传播或者传输用于由指令执行系统、装置或者器件使用或者与其结合使用的程序。
[0127] 计算机可读介质上包含的程序代码可以用任何适当的介质传输,包括——但不限于——无线、电线、光缆、RF等等,或者上述的任意合适的组合。
[0128] 可以以一种或多种程序设计语言或其组合来编写用于执行本申请操作的计算机程序代码,所述程序设计语言包括面向对象的程序设计语言—诸如Java、Smalltalk、C++,还包括常规的过程式程序设计语言—诸如“C”语言或类似的程序设计语言。程序代码可以完全地在用户电子设备上执行、部分地在用户电子设备上执行、作为一个独立的软件包执行、部分在用户电子设备上部分在远程电子设备上执行、或者完全在远程电子设备或服务器上执行。在涉及远程电子设备的情形中,远程电子设备可以通过任意种类的网络——包括局域网(LAN)或广域网(WAN)—连接到用户电子设备,或者,可以连接到外部电子设备(例如利用因特网服务提供商来通过因特网连接)。
[0129] 本领域技术人员在考虑说明书及实践这里申请的发明后,将容易想到本申请的其它实施方案。本申请旨在涵盖本申请的任何变型、用途或者适应性变化,这些变型、用途或者适应性变化遵循本申请的一般性原理并包括本申请未发明的本技术领域中的公知常识或惯用技术手段。说明书和实施例仅被视为示例性的,本申请的真正范围和精神由权利要求指出。
[0130] 应当理解的是,本申请并不局限于上面已经描述并在附图中示出的精确结构,并且可以在不脱离其范围进行各种修改和改变。本申请的范围仅由所附的权利要求来限制。