[发明专利]一种道路交通标志识别方法及装置

说明书

本申请提供一种道路交通标志识别方法及装置，在获取到彩色二维行车图像后；依据预设道路交通标志检测模型，从所述彩色二维行车图像中检测道路交通标志，并确定所述道路交通标志在所述彩色二维行车图像中的坐标位置，所述道路交通标志包括：警告标志、禁令标志、指示标志、指路标志、旅游区标志以及辅助标志；然后依据禁令标识分类模型，根据检测到的禁令标识的内容确定所述道路交通标志的类别。通过本申请可以提高道路交通标志的检测精度和实时性。

技术领域

本申请涉及图像识别技术领域，特别的，尤其涉及一种道路交通标志识别方法及装置。

背景技术

交通标识识别是智能交通系统的关键技术，在车辆控制、交通监控和智能驾驶等系统中都有着广泛的应用。近年来，无人驾驶汽车因具有安全、高效等优点受到社会的广泛关注，Google的无人驾驶车队已经在多个州进行了试运行，但在试运行期间发生了多起交通事故。由此可以看出，道路交通标志的检测与识别对于改善自动驾驶系统的安全性和可靠性、确保安全驾驶等方面具有重要意义。

目前的无人驾驶技术存在极大的安全隐患，并且考虑到交通标志通常处于室外复杂的环境条件下，识别的过程中容易受环境光照、方向旋转的影响，在不同环境中对道路交通标志的检测效果和实时性有较大差异，从而导致交通事故，所以需要一种有效的方法来提高道路交通标志的检测精度和实时性。

发明内容

鉴于上述内容中的问题，本申请提供了一种道路交通标志识别方法及装置，用以提高道路交通标志的检测精度和实时性。

为了实现上述目的，本申请提供了以下技术方案：

一种道路交通标志识别方法，包括：

获取彩色二维行车图像；

依据预设道路交通标志检测模型，从所述彩色二维行车图像中检测道路交通标志，并确定所述道路交通标志在所述彩色二维行车图像中的坐标位置，所述道路交通标志包括：警告标志、禁令标志、指示标志、指路标志、旅游区标志以及辅助标志；

依据禁令标识分类模型，根据检测到的禁令标识的内容确定所述道路交通标志的类别。

进一步的，所述依据预设道路交通标志检测模型，从所述彩色二维行车图像中检测道路交通标志，并确定所述道路交通标志在所述彩色二维行车图像中的坐标位置，包括：

将所述彩色二维行车图像作为输入，输入至所述预设道路交通标志检测模型中，根据道路交通标志的形状特征和颜色特征，确定所述彩色二维行车图像中的道路交通标志；

利用所述预设道路交通标志检测模型使用所述彩色二维行车图像进行前向推理，得到所述道路交通标志在所述彩色二维行车图像中的坐标位置。

进一步的，所述预设道路交通标志检测模型的构建方法，包括：

获取构建所述预设道路交通标志检测模型的样本数据，所述样本数据为进行图像标注后的彩色二维行车图像；

根据所述样本数据确定所述道路交通标志在所述标注后的彩色二维行车图像中的坐标位置；

将所述样本数据和所述坐标位置作为训练数据，基于轻量化的深度学习模型进行模型训练，得到所述预设道路交通标志检测模型。

进一步的，所述禁令标识分类模型的构建方法，包括：

获取构建所述禁令标识分类模型的样本数据，所述样本数据为将所述彩色二维行车图像中的禁令标志类别裁剪出来的彩色二维行车图像；

针对所述样本数据进行数据增强操作，得到增强操作后的图像数据；

将所述增强操作后的图像数据作为训练数据，基于轻量化分类模型进行模型训练，得到所述禁令标识分类模型。

一种道路交通标志识别装置，包括：