一种基于双阶段卷积神经网络的数字类交通标志检测识别方法技术

本发明专利技术公开了一种基于双阶段卷积神经网络的数字类交通标志检测识别方法，包括：(1)采集图像数据并标注；(2)设计第一阶段目标检测模型，用于定位和识别大类别；(3)将图像输入到第一阶段目标检测模型中进行训练；(4)将图像输入到第一阶段目标检测模型中，通过框的融合后处理算法获得预测框的坐标以及大类别，并将预测框对应的区域扣除出来，标注扣除图像中的各个数字的位置、类别以及对应的大类别；(5)设计第二阶段网络，用于扣除图像中数字的识别以及大类别的重分类；(6)将扣除出来的图像输入到第二阶段网络中进行训练；(7)训练好的两个阶段的模型用于整体的预测和推理。本发明专利技术的模型结构简单、易于部署，识别精确度高。识别精确度高。识别精确度高。

【技术实现步骤摘要】
一种基于双阶段卷积神经网络的数字类交通标志检测识别方法


[0001]本专利技术属于目标检测领域，尤其是涉及一种基于双阶段卷积神经网络的数字类交通标志检测识别方法。

技术介绍

[0002]自动的数字类交通标志的检测与识别对于无人驾驶以及辅助驾驶系统是非常重要的。随着深度学习的发展，基于卷积神经网络的目标检测网络例如SSD，YoloV3等已经在各个目标检测场景中取得了非常好的效果。
[0003]如公开号为CN111444847A的中国专利文献公开了一种交通标志检测识别方法，包括获取交通标志图像；对YOLOv3卷积神经网络模型的残差组件进行处理得到DFF
‑
YOLOv3卷积神经网络模型；利用DFF
‑
YOLOv3卷积神经网络模型对获取的交通标志图像进行检测识别。该方法通过对现有的基于YOLOv3的卷积神经网络模型的残差组件进行改进处理得到改进的DFF
‑
YOLOv3卷积神经网络模型，利用改进的DFF
‑
YOLOv3卷积神经网络模型对获取的小目标交通标志图像进行检测识别。但本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于双阶段卷积神经网络的数字类交通标志检测识别方法，其特征在于，包括以下步骤：(1)采集交通场景下的图像数据，并在图像数据中标注出数字类交通标志的类别以及位置；(2)设计第一阶段目标检测模型，用于交通标志的定位和识别交通标志属于哪一个大类别；(3)将标注后的图像数据输入到第一阶段目标检测模型中，进行不少于100个Epoch的迭代训练，同时在随着训练的深入依次减少学习率；监视模型的损失函数，直到模型收敛；(4)将原始图像数据输入到训练好的第一阶段目标检测模型中，通过框的融合后处理算法获得预测框的坐标以及大类别的预测结果，并将预测框对应的区域从原始图像中扣除出来，同时标注扣除图像中的各个数字的位置、类别以及其对应的大类别；(5)设计第二阶段网络，用于扣除图像中数字的识别以及大类别的重分类；(6)将步骤(4)扣除出来的图像输入到第二阶段网络中，进行不少于100个Epoch的迭代训练，同时在随着训练的深入依次减少学习率；监视模型的损失函数，直到模型收敛；(7)在应用过程中，将待识别图像首先输入到训练好的第一阶段目标检测模型中获得交通标志的大类别以及其具体的位置，然后再将对应的图像区域扣除出来后送到训练好的第二阶段网络中进行数字的识别以及大类别的重分类；最后将第一阶段目标检测模型中获得的大类别与第二阶段网络得到的重分类的大类别进行融合，得到最终的大类别。2.根据权利要求1所述的基于双阶段卷积神经网络的数字类交通标志检测识别方法，其特征在于，步骤(1)中，在图像数据中标注出的数字类交通标志的类别包括大类别以及对应的具体数值，所述的大类别包括最高限速、最低限速、解除限速、限高和限重。3.根据权利要求1所述的基于双阶段卷积神经网络的数字类交通标志检测识别方法，其特征在于，步骤(2)中，所述的第一阶段目标检测模型采用轻量化的YoloV3网络，包括用于提取特征的卷积...

【专利技术属性】
技术研发人员：谢磊，李至善，
申请(专利权)人：浙江大学，
类型：发明
国别省市：