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

权利要求书

1.一种车道箭头旋转目标检测模型构建方法，其特征在于，包括：

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

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

2.根据权利要求1所述的方法，其特征在于，所述车道箭头类别包括：直行、左转、右转、掉头、直行或左转、直行或右转、左转或左合流、右转或右合流以及上述8类外的其他类型。

3.根据权利要求2所述的方法，其特征在于，在进行箭头转向信息标注时，直行或左转、直行或右转的转向信息按直行箭头方式来标注。

4.根据权利要求1所述的方法，其特征在于，所述转向信息的标注方法包括：确定车道箭头的中轴线，沿箭头顶点垂直车道箭头中轴线做直线，并以该直线为矩形框的一条边，绘制车道箭头的最小外包矩形框，所述矩形框的倾斜角度与箭头指向一致。

5.根据权利要求1所述的方法，其特征在于，所述箭头类别权重参数根据车道箭头原始图像数据集中各类别车道箭头的样本数量在总样本中的占比进行设置，其中占比越大的某类别车道箭头，其权重越小。

6.根据权利要求1所述的方法，其特征在于，在模型训练过程中，利用Wasserstein距离作为损失函数对模型进行验证。

7.一种车道箭头旋转目标检测模型构建装置，其特征在于，包括：

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

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

8.一种电子设备，其特征在于，包括：

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

处理器，用于读取并执行所述计算机软件程序，进而实现权利要求1-6任一项所述的一种车道箭头旋转目标检测模型构建方法。

9.一种非暂态计算机可读存储介质，其特征在于，所述存储介质中存储有用于实现权利要求1-6任一项所述的一种车道箭头旋转目标检测模型构建方法的计算机软件程序。