[发明专利]一种车道箭头旋转目标检测模型构建方法及装置

说明书

本发明涉及一种车道箭头旋转目标检测模型构建方法及装置，首先获取车道箭头原始图像数据集，标注图像数据集中的箭头类别及箭头转向信息；然后，设置网络中箭头类别权重参数，将标注好的箭头图像输入GWD网络模型进行训练，得到车道箭头旋转目标检测模型。本发明可以直接检测车道箭头的种类及角度，弥补了其他目标检测算法只检测箭头类别的不足。

技术领域

本发明涉及自动驾驶技术领域，具体涉及一种车道箭头旋转目标检测模型构建方法及装置。

背景技术

在自动驾驶领域，车道箭头的意义有着直接指导车辆行驶方向的作用，目前的车端目标检测对于箭头的识别是归为一类的，并且标注结果也是水平方向的矩形拉框，没有包含箭头的指导转向。因此在目标检测的基础上加入箭头的转向信息，在自动驾驶领域有重要的作用。

发明内容

本发明针对现有技术中存在的技术问题，提供一种车道箭头旋转目标检测模型构建方法及装置，用以解决现有目标检测无法检测车道箭头转向信息的缺失问题。

本发明解决上述技术问题的技术方案如下：

第一方面，本发明提供一种车道箭头旋转目标检测模型构建方法，包括：

S1，获取车道箭头原始图像数据集，标注图像数据集中的箭头类别及箭头转向信息；

S2，设置网络中箭头类别权重参数，将标注好的箭头图像输入GWD网络模型进行训练，得到车道箭头旋转目标检测模型。

进一步的，所述车道箭头类别包括：直行、左转、右转、掉头、直行或左转、直行或右转、左转或左合流、右转或右合流以及上述8类外的其他类型。

进一步的，在进行箭头转向信息标注时，直行或左转、直行或右转的转向信息按直行箭头方式来标注。

进一步的，所述转向信息的标注方法包括：确定车道箭头的中轴线，沿箭头顶点垂直车道箭头中轴线做直线，并以该直线为矩形框的一条边，绘制车道箭头的最小外包矩形框，所述矩形框的倾斜角度与箭头指向一致。

进一步的，所述箭头类别权重参数根据车道箭头原始图像数据集中各类别车道箭头的样本数量在总样本中的占比进行设置，其中占比越大的某类别车道箭头，其权重越小。

进一步的，在模型训练过程中，利用Wasserstein距离作为损失函数对模型进行验证。

第二方面，本发明提供一种车道箭头旋转目标检测模型构建装置，包括：

标注模块，获取车道箭头原始图像数据集，标注图像数据集中的箭头类别及箭头转向信息；

训练模块，设置网络中箭头类别权重参数，将标注好的箭头图像输入GWD网络模型进行训练，得到车道箭头旋转目标检测模型。

第三方面，本发明提供一种电子设备，包括：

存储器，用于存储计算机软件程序；

处理器，用于读取并执行所述计算机软件程序，进而实现本发明第一方面所述的一种车道箭头旋转目标检测模型构建方法。

第四方面，本发明提供一种非暂态计算机可读存储介质，所述存储介质中存储有用于实现本发明第一方面所述的一种车道箭头旋转目标检测模型构建方法的计算机软件程序。

本发明的有益效果是：本发明首先采集并标注车道箭头图像数据，更改不同箭头种类的权重；接着将准备好的数据送入gwd网络模型中进行训练，训练完成后，可以对车道箭头图像直接检测并分类。通过以上步骤，使得可以直接检测车道箭头的种类及角度，弥补了其他目标检测算法只检测箭头类别的不足。

附图说明

图1为本发明实施例提供的一种车道箭头旋转目标检测模型构建方法流程示意图；