[发明专利]基于多尺度信息和残差网络的小目标交通标志检测方法

说明书

本发明涉及一种基于多尺度信息和残差网络的小目标交通标志检测方法，包括下列步骤：准备包含不同种类交通标志的数据集；网络搭建，包括特征提取网络、区域生成网络、兴趣区域池化层和全连接层；训练所搭建的网络，得到网络模型：使用gradual warmup的方法优化学习率，在训练的过程中，学习率从0开始逐渐增大到设置值；在训练过程中，每次迭代，将随机选取的训练集图片输入所搭建的网络，得到图片中交通标志的类别预测值和边界框预测值，计算预测值和json文件中真实值的误差得到损失函数，采用随机梯度下降SGD更新网络参数，不断重复此过程直至误差不再减小，从而得到用于检测小目标交通标志的网络模型。

技术领域

本方法涉及目标检测领域，特别涉及自动驾驶汽车中的辅助驾驶系统。

背景技术

伴随着汽车的大范围普及，我国交通安全事故发生率也随之逐年上升。使车辆能够自动检测并识别交通标志、获取路面情况可以在很大程度上减轻驾驶员的精神负担、缓解疲劳，保障安全驾驶。

交通标志检测是指根据实时图片中交通标志的颜色、形状等特征，定位交通标志的位置，并且对图片中的交通标志进行分类识别，获取交通标志的含义，规范车辆的行驶。交通标志检测是辅助驾驶系统中的核心任务，对小目标交通标志检测的研究具有重要的现实意义。

由于国内外大部分交通标志具有规则的形状和易于区分的颜色，所以早期国内外学者多采用图像处理的方法来解决交通标志检测问题。后来随着机器学习的兴起，HOG特征结合SVM分类器等交通标志检测算法不断涌现。而近年来随着神经网络的不断研究和创新，深度学习的方法越来越多地应用到交通标志检测领域，并占据了重要地位。在近年来公布的众多论文、专利中，交通标志检测算法基本都基于深度神经网络。Zang等[1]提出了一种基于级联卷积神经网络(CNN)的交通标志检测方法。首先，结合特征检测器和AdaBoost分类器提取感兴趣区域；其次，采用级联卷积神经网络来筛选感兴趣区域，实现交通标志检测。Zhu等[2]提出了一个包含深度学习的新框架，整个框架包含两个深度神经网络，一个是全卷积网络(FCN)，用以产生交通标志的候选区域；另一个是卷积神经网络，用于分类。ZhangJ等[3]提出了一种基于卷积神经网络的浅层网络结构，该网络仅由三个卷积层进行特征提取，并采用逆向优化的方法进行学习，使用激活函数ReLU来提高计算效率，该方法在德国交通标志识别数据集(GTSRB)上获得了高准确率。(CN201910365006.1)提出一种基于注意力机制的交通标志检测方法，此方法通过贪心策略算法对待检测图像进行切片，得到目标检测区域，通过检测模型获取与每个目标检测区域对应的预测结果，提高了交通标志检测的精度值和效率。(CN201910440872.2)将交通标志图像进行模糊化处理，虚化交通标志因为光线、提取范围等原因而产生的不光滑的轮廓表面，然后建立卷积神经网络识别图像，在一定程度上减少了噪声的干扰。(CN201910552184.5)设计了一种改进的神经网络模型，模型使用PW卷积、DW卷积以降低参数量、提升检测速度，并改进交叉熵损失，在损失函数中对易分类样本进行抑制，使模型在训练过程中更加注重难分类样本。(CN201910474058.2)提出了一种基于F-RCNN的远距离交通标志检测识别方法，利用F-RCNN和极限学习机网络，实现了较高的识别准确率。

现有的目标检测方法，主要是针对通用目标数据集来设计的解决方案，应用于交通标志检测中时，对于小目标交通标志来说，检测效果不是很理想。此外，小目标本身包含信息少，且经过多次卷积之后信息容易被丢失，给检测带来困难。