CN111199217A - 一种基于卷积神经网络的交通标志识别方法和系统 - Google Patents

一种基于卷积神经网络的交通标志识别方法和系统 Download PDF

Info

Publication number
CN111199217A
CN111199217A CN202010022391.2A CN202010022391A CN111199217A CN 111199217 A CN111199217 A CN 111199217A CN 202010022391 A CN202010022391 A CN 202010022391A CN 111199217 A CN111199217 A CN 111199217A
Authority
CN
China
Prior art keywords
traffic sign
layer
neural network
convolutional neural
convolution
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Granted
Application number
CN202010022391.2A
Other languages
English (en)
Other versions
CN111199217B (zh
Inventor
张珂
侯捷
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Shanghai Institute of Technology
Original Assignee
Shanghai Institute of Technology
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by Shanghai Institute of Technology filed Critical Shanghai Institute of Technology
Priority to CN202010022391.2A priority Critical patent/CN111199217B/zh
Publication of CN111199217A publication Critical patent/CN111199217A/zh
Application granted granted Critical
Publication of CN111199217B publication Critical patent/CN111199217B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Images

Classifications

    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V20/00—Scenes; Scene-specific elements
    • G06V20/50—Context or environment of the image
    • G06V20/56—Context or environment of the image exterior to a vehicle by using sensors mounted on the vehicle
    • G06V20/58—Recognition of moving objects or obstacles, e.g. vehicles or pedestrians; Recognition of traffic objects, e.g. traffic signs, traffic lights or roads
    • G06V20/582—Recognition of moving objects or obstacles, e.g. vehicles or pedestrians; Recognition of traffic objects, e.g. traffic signs, traffic lights or roads of traffic signs
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06F—ELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00—Pattern recognition
    • G06F18/20—Analysing
    • G06F18/24—Classification techniques
    • G06F18/241—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06F—ELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00—Pattern recognition
    • G06F18/20—Analysing
    • G06F18/24—Classification techniques
    • G06F18/241—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches
    • G06F18/2415—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches based on parametric or probabilistic models, e.g. based on likelihood ratio or false acceptance rate versus a false rejection rate
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00—Computing arrangements based on biological models
    • G06N3/02—Neural networks
    • G06N3/04—Architecture, e.g. interconnection topology
    • G06N3/045—Combinations of networks
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00—Computing arrangements based on biological models
    • G06N3/02—Neural networks
    • G06N3/08—Learning methods
    • G06N3/084—Backpropagation, e.g. using gradient descent
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V2201/00—Indexing scheme relating to image or video recognition or understanding
    • G06V2201/09—Recognition of logos

Abstract

本发明提供了一种基于卷积神经网络的交通标志识别方法和系统，所述方法包括步骤：S1：获取若干交通标志图像，各交通标志图像预设所属类别；S2：对若干由所述交通标志图像组成的训练数据集和测试数据集进行预处理；S3：构建卷积神经网络；S4：将所述训练数据集输入到已构建的卷积神经网络中，通过反向传播算法进行连续多次的迭代训练，以便生成交通标志识别模型；S5：将所述测试数据集输入到所述交通标志识别模型中，输出所述交通标示图像识别分类结果。本发明对车载环境中取得的交通标志图像进行识别，能够取得较高的准确率，同时网络模型的训练时间相对较短，有效的解决了识别时间长、过拟合的难题。

Description

一种基于卷积神经网络的交通标志识别方法和系统
技术领域
本发明涉及机器视觉领域，具体涉及一种基于卷积神经网络的交通标志识别方法和系统。
背景技术
随着人们生活的进步，车辆已经是人们不可或缺的交通工具，但是由于汽车的保有量急剧增加，导致交通堵塞，事故频繁的发生，因此提高汽车智能化的行驶安全性能已是自动驾驶车辆的首要任务。随着车辆智能化技术发展，无人驾驶车辆技术成为了自动驾驶系统的重要研究领域。
自动驾驶车辆和驾驶员辅助系统普遍是依靠车载摄像头摄取的图像进行检测识别，再通过系统的一系列运算反馈给车辆做出正确的动作。摄像头收集图片时，车辆都是在运动状态下完成的，再加上室外交通环境复杂，识别图像中的物体要比一般静止的图片难度更大。主要困难体现在运动模糊、光照多变复杂、硬件问题、角度倾斜、外物遮掩等。
近年来随着卷积神经网络理论及应用的开发，诸多图像分类方法都已不再需要人工进行提取，而是把图像传输入神经网络中，让神经网络自主学习，自行提取图像的多个特征，之后在通过分类器进行图像的识别分类。
发明内容
本申请提供了一种基于卷积神经网络的交通标志识别方法和系统，应用于环境复杂的交通标志识别，针对交通标志图像所处环境的多变，拍摄角度不同，运动模糊，光照强度多变等情况下无法进行快速、准确的分类问题。
为达到以上目的，本发明采取的技术方案是：
一种基于卷积神经网络的交通标志识别方法，其包括如下步骤：
S1：获取若干交通标志图像，各所述交通标志图像预设所属类别；
S2：对若干由所述交通标志图像组成的训练数据集和测试数据集进行预处理；
S3：构建卷积神经网络；
S4：将所述训练数据集输入到已构建的卷积神经网络中，通过反向传播算法、改进网络内部参数，进行连续多次的迭代训练，以便生成交通标志识别模型；
S5：将所述测试数据集输入到所述交通标志识别模型中，输出所述交通标示图像识别分类结果。
进一步地，所述S2中，对所述交通标志图像进行预处理，包括步骤：
S21：对所述交通标志图像进行图像裁剪；
S22：将裁剪后的所述交通标志图像进行标准化处理；
S23：将标准化处理后的所述交通标志图像进行尺寸归一化处理；
S24：输出预处理后用作所述卷积神经网络输入图像的所述交通标志图像。
进一步地，在所述S22中，所述交通标志图像的标准化处理的运算公式采用：
Figure BDA0002361275030000021
其中，μ为图像的均值，x为图像矩阵，σ为标准方差，P为图像的像素数。
进一步地，在所述S3中，所述卷积神经网络中包括用于模型训练的依次连接的输入层、第一卷积层、第二池化层、第三卷积模块组、第四池化层、第五卷积层、第六池化层、第七卷积层、第八全连接层、输出层；
其中，所述第三卷积模块组包括三个分支，包括第一分支、第二分支以及第三分支；
所述第一分支包括第一分支第一卷积层、第一分支第二卷积层；
所述第二分支包括第二分支第一卷积层、第二分支第二卷积层；
所述第三分支包括第三分支卷积层；
通过预设的卷积参数，进行卷积处理的各层在获取上层的图像进行卷积处理后，得到的特征图像的数量呈梯度上升趋势；
通过预设的池化参数，进行池化处理的各层在获取上层的图像进行池化处理后，得到的特征图像的尺寸呈梯度下降趋势；
所述第八全连接层预设有若干神经元，各所述神经元连接所述输出层，用于输出分类结果。
进一步地，所述卷积神经网络中用于池化处理的各层采用最大池化方式进行池化处理。
进一步地，所述卷积神经网络中预设有多种卷积处理方式，其中，
所述第五卷积层、第七卷积层采用VALID填充方式进行卷积运算；
所述第一卷积层、第三卷积模块组采用SAME填充方式进行卷积运算。
进一步地，所述卷积神经网络中用于卷积处理的各层中预设有激活函数；所述激活函数采用ReLU激活函数，所述ReLU激活函数的表达式为：f(x)＝max(0,x)，其中，x表示各卷积层的输出。
进一步地，所述卷积神经网络中进行卷积处理的各层的卷积核初始化方法采用高斯初始化方法，其中，所述高斯初始化方法的表达式为：
Figure BDA0002361275030000031
其中，a为ReLU激活函数的负半轴斜率，nl为输入的维数，即nl＝卷积核边长2*通道数。
进一步地，所述卷积神经网络中的所述输出层通过Softmax函数获取图像所属类别的概率，并选择概率最大值的分类为最后分类结果，所述Softmax函数的表达式为：
Figure BDA0002361275030000032
其中，N表示图像分类的个数，ck为输出属于第k类。
为达到以上目的，本发明采取的另一技术方案是：
一种基于卷积神经网络的交通标志识别系统，所述系统包括：采集模块、预处理模块、构建模块、训练模块、识别模块，
所述采集模块用于获取若干交通标志图像，各所述交通标志图像预设所属类别；
所述预处理模块用于对若干由所述交通标志图像组成的训练数据集和测试数据集进行预处理；
所述构建模块用于构建卷积神经网络；
所述训练模块用于将所述训练数据集输入到已构建的卷积神经网络中，通过反向传播算法、改进网络内部参数，进行连续多次的迭代训练，以便生成交通标志识别模型；
所述识别模块用于将所述测试数据集输入到所述交通标志识别模型中，输出所述交通标示图像识别分类结果。
一种交通标志识别装置，包括处理器、存储器，所述处理器在执行存储于存储器上的计算机程序时，用于实现如前文所述的基于卷积神经网络的交通标志识别方法。
一种可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时，用于实现如如前文所述的基于卷积神经网络的交通标志识别方法。
相比现有技术，本发明具有的有益效果：
本发明针对车载环境中取得的交通标志图像进行识别，能够取得较高的准确率，同时网络模型的训练时间相对较短，有效的解决了识别时间长、过拟合的难题。
附图说明
图1为本发明的一种基于卷积神经网络训练和测试过程的示意图。；
图2为本发明的一种交通标志图像的预处理过程示意图；
图3为本发明的一种基于卷积神经网络结构示意图。
具体实施方式
为了便于理解本发明，下面通过具体实施方式结合附图对本发明作进一步详细说明。
附图中给出了本发明的较佳实施方式。图1为本发明的一种基于卷积神经网络训练和测试过程的示意图；图2为本发明的一种交通标志图像的预处理过程示意图；图3为本发明的一种基于卷积神经网络结构示意图。除非另有定义，本文所使用的所有的技术和科学术语与所属于本发明的技术领域的技术人员通常理解的含义相同。
参照附图，本申请提供了一种基于卷积神经网络的交通标志识别方法，该方法应用于环境复杂的交通标志识别，针对交通标志图像所处环境的多变性，拍摄角度不同，运动模糊，光照强度多变等情况下进行快速、准确的分类。
为实现上述目的，参考图1，本次发明所设计的一种基于卷积神经网络的交通标志识别方法，该方法包括步骤：
S1：获取若干交通标志图像；
S2：对若干由交通标志图像组成的训练数据集和测试数据集进行预处理；
S3：构建卷积神经网络；
S4：将训练数据集输入到已构建的卷积神经网络中，通过反向传播算法、改进网络内部参数，进行连续多次的迭代训练，以获取交通标志识别模型；
S5：将测试数据集输入到交通标志识别模型中，输出交通标示图像识别分类结果。
在步骤S1中，获取若干交通标志图像，包括模拟在自然环境下行驶车辆采集的交通标志图像，在一种实施例中，包括在各个环境中、不同拍摄角度、各种行驶状态下、各种光照强度下的交通标志图像等，并且根据不同情况进行快速准确的分类。
在步骤S2中，对若干由交通标志图像组成的训练数据集和测试数据集进行预处理；进一步地，包括对训练数据集和测试数据集中的各个交通标志图像进行预处理；根据不同情况，本实施例中，可根据不同情况将交通标志图像划分为包括多类的交通标志图像的训练数据集和测试数据集。进一步地，对训练数据集和测试数据集中的各个交通标志图像进行预处理，使训练数据集和测试数据集中的各个交通标志图像处理为尺寸统一的三通道彩色图像。
本实施例中，参考图2，对各个交通标志图像进行预处理，包括步骤：
S21：对交通标志图像进行图像裁剪；
S22：将裁剪后的交通标志图像进行标准化处理；
S23：将标准化处理后的交通标志图像进行尺寸归一化处理；
S24：输出预处理后用作卷积神经网络输入图像的交通标志图像。
其中，在步骤S22中，各交通标志图像的标准化处理的运算公式采用：
Figure BDA0002361275030000051
其中，μ为图像的均值，x为图像矩阵，σ为标准方差，P为图像的像素数。
进一步地，
Figure BDA0002361275030000052
其中，adjusted_stddev为调整样本标准偏差。
在步骤S3中，卷积神经网络中包括用于模型训练的依次连接的输入层、第一卷积层、第二池化层、第三卷积模块组、第四池化层、第五卷积层、第六池化层、第七卷积层、第八全连接层、输出层。
其中，第三卷积模块组包括三个分支，包括第一分支、第二分支以及第三分支；第一分支包括第一分支第一卷积层、第一分支第二卷积层；第二分支包括第二分支第一卷积层、第二分支第二卷积层；第三分支包括第三分支卷积层。
其中，通过预设的卷积参数，进行卷积处理的各层在获取上层的图像进行卷积处理后，得到的特征图像的数量呈梯度上升趋势。通过预设侧池化参数，进行池化处理的各层在获取上层的图像进行池化处理后，得到的特征图像的尺寸呈梯度下降趋势。第八全连接层预设有若干神经元，各神经元连接输出层，用于输出分类结果。
本实施例中的卷积神经网络中用于池化处理的各层采用最大池化方式进行池化处理，进一步地，第二池化层、第四池化层以及第六池化层采用最大池化方式处理。
本实施例中的卷积神经网络中预设有多种卷积处理方式，其中，第五卷积层、第七卷积层采用VALID填充方式进行卷积运算；第一卷积层、第三卷积模块组采用SAME填充方式进行卷积运算，进一步地，第一卷积层、第一分支第一卷积层、第一分支第二卷积层、第二分支第一卷积层、第二分支第二卷积层、第三分支卷积层采用SAME填充方式进行卷积运算。
卷积神经网络中用于卷积处理的各层中预设有激活函数，常见的激活函数包括sigmod激活函数、tanh激活函数、ELU激活函数、ReLU激活函数等。本实施例中的激活函数采用ReLU激活函数，其中，ReLU激活函数的表达式为：f(x)＝max(0,x)，其中，x表示各卷积层的输出。
进一步地，sigmod激活函数、tanh激活函数、ELU激活函数等函数，计算激活函数时，计算量大，反向传播求误差梯度时，求导涉及除法，计算量相对大，而采用Relu激活函数，Relu激活函数会使一部分神经元的输出为0，这样就造成了网络的稀疏性，并且减少了参数的相互依存关系，缓解了过拟合问题的发生，整个过程的计算量节省很多。
卷积神经网络中进行卷积处理的各层的卷积核初始化方法采用高斯初始化方法，本实施例中的高斯初始化方法的表达式为：
Figure BDA0002361275030000061
其中，a为ReLU的负半轴斜率，nl为输入的维数，即nl＝卷积核边长2*通道数。
在神经网络的训练中如果将权重全部初始化为0，则第一遍前向传播过程中，所有隐藏层神经元的激活函数值都相同，导致深层神经元可有可无，造成对称权重现象，因此，本实施例中，采用最简单常用的高斯分布初始化权重能够达到训练效果。
卷积神经网络中的输出层通过Softmax函数获取图像所属类别的概率，并选择概率最大值的分类为最后分类结果。本实施例中的Softmax函数的表达式为：
Figure BDA0002361275030000062
其中，N表示图像分类的个数，表示输出属于第k类。
参考图3，以下通过设置具体参数，进行通过卷积神经网络的模型训练的详细描述。
卷积神经网络通过进行模型训练获取交通标志识别模型，卷积神经网络中用于模型训练的各个处理层，包括依次连接的输入层、第一卷积层、第二池化层、第三卷积模块组、第四池化层、第五卷积层、第六池化层、第七卷积层、第八全连接层、输出层；其中第三卷积模块组包括三个分支，包括第一分支、第二分支以及第三分支，第一分支包括第一分支第一卷积层、第一分支第二卷积层；第二分支包括第二分支第一卷积层、第二分支第二卷积层；第三分支包括第三分支卷积层。
预设参数，在一种实施例中，将各个交通标志图像处理为像素值为48*48的RGB三通道的彩色图像，使得各个交通标志图像为可进行卷积神经网络图像输入的48*48的RGB三通道的彩色图像。其中，本实施例中，包括N类交通标志图像M张，进一步地，由交通标志图像组成的训练数据集X张、由交通标志图像组成的测试数据集Y张，X<M，Y<M。
本实施例中的卷积神经网络的模型训练的步骤包括：
步骤1：将由交通标志图像组成的训练数据集和测试数据集中的各个交通标志图像输入到输入层中，即输入的交通标志图像为48*48*3；
步骤2：第一卷积层获取输入层输入的各个交通标志图像，对各个交通标志图像进行特征提取。第一卷积层Conv1中的卷积核尺寸为3*3，卷积核数量为16，经过卷积处理后各交通标志图像获取16个尺寸为48*48的交通标志图像的特征图像，即输出的特征图像为Conv1:3*3*16。
步骤3：第二池化层获取第一卷积层提取的特征图片，包括16个尺寸为48*48的交通标志图像的特征图片，第二池化层MaxPool1采用最大池化方式对特征图片进行池化处理，池化处理的尺寸为2*2，步长为2，经过池化处理后，提取出16个尺寸为24*24的交通标志图像的特征图片，即输出的特征图像MaxPool1:2*2。
步骤3：第三卷积模块组包括三个分支，从步骤3中获取的16个尺寸为24*24的交通标志图像的特征图片分别经过第一分支、第二分支以及第三分支进行卷积处理。进一步地：
A、第一分支第一卷积层获取第二池化层提取的特征图片，包括16个24*24的交通标志图像的特征图片，第一分支第一卷积层Conv2_1中的卷积核尺寸为3*3，卷积核数量为32个，即输出的特征图像Conv2_1:3*3*32；第一分支第二卷积层Conv2_2中的卷积核尺寸为3*3，卷积核数量为64，即输出的特征图像Conv2_2:3*3*64。依次经过第一分支第一卷积层、第一分支第二卷积层卷积处理后，提取出64个尺寸为24*24的交通标志图像的特征图片。
B、第二分支第一卷积层获取第二池化层提取的特征图片，包括16个24*24的交通标志图像的特征图片，第二分支第一卷积层Conv3_1中的卷积核尺寸为3*3，卷积核数量为32个，即输出的特征图像Conv3_1:3*3*32；第一分支第二卷积层Conv3_2中的卷积核尺寸为1*1，卷积核数量为16，即输出的特征图像Conv3_2:1*1*16，依次经过第二分支第一卷积层、第二分支第二卷积层卷积处理后，提取出16个尺寸为24*24的交通标志图像的特征图片。
C、第三分支卷积层获取第二池化层提取的特征图片，包括16个24*24的交通标志图像的特征图片，第三分支卷积层Conv4中的卷积核尺寸为1*1，卷积核数量为32，经过卷积处理后处32个尺寸为24*24的交通标志图像的特征图片，即输出的特征图像Conv4:1*1*32。
D、经过A、B、C三个分支，进行三个分支的维度堆叠，运算提取出112个尺寸为24*24的交通标志图像的特征图片，即输出的特征图像Conv5:1*1*112。
步骤4：第四池化层获取第三卷积模块组提取的特征图片，包括112个尺寸为24*24的交通标志图像的特征图片，第四池化层采用最大池化方式对特征图片进行池化处理，池化处理的尺寸为2*2，步长为2，经过池化处理后，提取出112个尺寸为12*12的交通标志图像的特征图片，即输出的特征图像MaxPool2:2*2。
步骤5：第五卷积层获取第四池化层提取的特征图片，包括112个尺寸为12*12的交通标志图像的特征图片，第五卷积层中的卷积核尺寸为3*3，卷积核数量为256，经过卷积处理后各交通标志图像提取出256个尺寸为10*10的交通标志图像的特征图像，即输出的特征图像Conv6:3*3*256。
步骤6：第六池化层获取第五卷积层提取的特征图片，包括256个尺寸为10*10的交通标志图像的特征图片，第六池化层采用最大池化方式对特征图片进行池化处理，池化处理的尺寸为2*2，步长为2，经过池化处理后，提取出256个尺寸为5*5的交通标志图像的特征图片，即输出的特征图像MaxPool3:2*2。
步骤7：第七卷积层获取第六池化层提取的特征图片，包括256个尺寸为5*5的交通标志图像的特征图片，第七卷积层中的卷积核尺寸为3*3，卷积核数量为512，经过卷积处理后各交通标志图像提取出512个尺寸为3*3的交通标志图像的特征图像，即输出的特征图像Conv7:3*3*512。
步骤8：第八全连接层获取第七卷积层提取的特征图片，包括512个尺寸为3*3的交通标志图像的特征图像。第八全连接成中包含120个神经元，即FC1:120，各个神经元的节点连接到输出层。
步骤9：输出层Softmax将各交通标志图像的特征图像输出分类结果。
本申请还提供了一种基于卷积神经网络的交通标志识别系统，该系统包括：采集模块、预处理模块、构建模块、训练模块、识别模块，采集模块用于获取若干交通标志图像；所述预处理模块用于对若干由所述交通标志图像组成的训练数据集和测试数据集进行预处理；所述构建模块用于构建卷积神经网络；所述训练模块用于将所述训练数据集输入到已构建的卷积神经网络中，通过反向传播算法、改进网络内部参数，进行连续多次的迭代训练，以获取交通标志识别模型；所述识别模块用于将所述测试数据集输入到所述交通标志识别模型中，输出所述交通标示图像识别分类结果。
作为本发明的一种实施例，提供了一种交通标志识别装置，交通标志识别装置包括：存储器，存储有计算机程序的程序指令；计算机程序的程序指令用于被处理器加载并执行以实现如前文的基于卷积神经网络的交通标志识别方法的步骤；处理器，用于加载并执行存储器上的程序指令，以实现前文的基于卷积神经网络的交通标志识别方法的步骤；其中，存储器可以包括易失性存储器(英文：volatile memory)，例如随机存取存储器(英文：random-access memory，缩写：RAM)；存储器也可以包括非易失性存储器(英文：non-volatile memory)，例如快闪存储器(英文：flash memory)，硬盘(英文：hard disk drive，缩写：HDD)或固态硬盘(英文：solid-state drive，缩写：SSD)；存储器还可以包括上述种类的存储器的组合。其中，处理器可以是中央处理器(英文：central processing unit，缩写：CPU)，网络处理器(英文：network processor，缩写：NP)或者CPU和NP的组合。其中，处理器还可以进一步包括硬件芯片。上述硬件芯片可以是专用集成电路(英文：application-specificintegrated circuit，缩写：ASIC)，可编程逻辑器件(英文：programmable logicdevice，缩写：PLD)或其组合。上述PLD可以是复杂可编程逻辑器件(英文：complexprogrammablelogic device，缩写：CPLD)，现场可编程逻辑门阵列(英文：field-programmable gatearray，缩写：FPGA)，通用阵列逻辑(英文：generic array logic,缩写：GAL)或其任意组合。
作为本发明的另一种实施例提供了一种可读存储介质，可读存储介质中存储有计算机程序的程序指令，计算机程序的程序指令用于被处理器加载并执行以实现如前文的基于卷积神经网络的交通标志识别方法的步骤。
以上应用了具体个例对本发明进行阐述，只是用于帮助理解本发明，并不用以限制本发明。对于本发明所属技术领域的技术人员，依据本发明的思想，还可以做出若干简单推演、变形或替换。

Claims (12)

1.一种基于卷积神经网络的交通标志识别方法，其特征在于，所述方法包括步骤：
S1：获取若干交通标志图像，各所述交通标志图像预设所属类别；
S2：对若干由所述交通标志图像组成的训练数据集和测试数据集进行预处理；
S3：构建卷积神经网络；
S4：将所述训练数据集输入到已构建的卷积神经网络中，通过反向传播算法、改进网络内部参数，进行连续多次的迭代训练，以便生成交通标志识别模型；
S5：将所述测试数据集输入到所述交通标志识别模型中，输出所述交通标示图像识别分类结果。
2.如权利要求1所述的基于卷积神经网络的交通标志识别方法，其特征在于，所述S2中，对所述交通标志图像进行预处理，包括步骤：
S21：对所述交通标志图像进行图像裁剪；
S22：将裁剪后的所述交通标志图像进行标准化处理；
S23：将标准化处理后的所述交通标志图像进行尺寸归一化处理；
S24：输出预处理后用作所述卷积神经网络输入图像的所述交通标志图像。
3.如权利要求2所述的基于卷积神经网络的交通标志识别方法，其特征在于，在所述S22中，所述交通标志图像的标准化处理的运算公式采用：
Figure FDA0002361275020000011
其中，μ为图像的均值，x为图像矩阵，σ为标准方差，P为图像的像素数。
4.如权利要求1所述的基于卷积神经网络的交通标志识别方法，其特征在于，在所述S3中，所述卷积神经网络中包括用于模型训练的依次连接的输入层、第一卷积层、第二池化层、第三卷积模块组、第四池化层、第五卷积层、第六池化层、第七卷积层、第八全连接层、输出层；
其中，所述第三卷积模块组包括三个分支，包括第一分支、第二分支以及第三分支；
所述第一分支包括第一分支第一卷积层、第一分支第二卷积层；
所述第二分支包括第二分支第一卷积层、第二分支第二卷积层；
所述第三分支包括第三分支卷积层；
通过预设的卷积参数，进行卷积处理的各层在获取上层的图像进行卷积处理后，得到的特征图像的数量呈梯度上升趋势；
通过预设的池化参数，进行池化处理的各层在获取上层的图像进行池化处理后，得到的特征图像的尺寸呈梯度下降趋势；
所述第八全连接层预设有若干神经元，各所述神经元连接所述输出层，用于输出分类结果。
5.如权利要求4所述的基于卷积神经网络的交通标志识别方法，其特征在于，所述卷积神经网络中用于池化处理的各层采用最大池化方式进行池化处理。
6.如权利要求4所述的基于卷积神经网络的交通标志识别方法，其特征在于，所述卷积神经网络中预设有多种卷积处理方式，其中，
所述第五卷积层、第七卷积层采用VALID填充方式进行卷积运算；
所述第一卷积层、第三卷积模块组采用SAME填充方式进行卷积运算。
7.如权利要求4所述的基于卷积神经网络的交通标志识别方法，其特征在于，所述卷积神经网络中用于卷积处理的各层中预设有激活函数；所述激活函数采用ReLU激活函数，所述ReLU激活函数的表达式为：f(x)＝max(0,x)，其中，x表示各卷积层的输出。
8.如权利要求4所述的基于卷积神经网络的交通标志识别方法，其特征在于，所述卷积神经网络中进行卷积处理的各层的卷积核初始化方法采用高斯初始化方法，其中，所述高斯初始化方法的表达式为：
Figure FDA0002361275020000021
其中，a为ReLU激活函数的负半轴斜率，nl为输入的维数，即nl＝卷积核边长2*通道数。
9.如权利要求4所述的基于卷积神经网络的交通标志识别方法，其特征在于，所述卷积神经网络中的所述输出层通过Softmax函数获取图像所属类别的概率，并选择概率最大值的分类为最后分类结果，所述Softmax函数的表达式为：
Figure FDA0002361275020000022
其中，N表示图像分类的个数，ck为输出属于第k类。
10.一种基于卷积神经网络的交通标志识别系统，其特征在于，所述系统包括：采集模块、预处理模块、构建模块、训练模块、识别模块，
所述采集模块用于获取若干交通标志图像，各所述交通标志图像预设所属类别；
所述预处理模块用于对若干由所述交通标志图像组成的训练数据集和测试数据集进行预处理；
所述构建模块用于构建卷积神经网络；
所述训练模块用于将所述训练数据集输入到已构建的卷积神经网络中，通过反向传播算法、改进网络内部参数，进行连续多次的迭代训练，以便生成交通标志识别模型；
所述识别模块用于将所述测试数据集输入到所述交通标志识别模型中，输出所述交通标示图像识别分类结果。
11.一种交通标志识别装置，包括处理器、存储器，其特征在于，所述处理器在执行存储于存储器上的计算机程序时，用于实现如权利要求1至9中任一项所述的基于卷积神经网络的交通标志识别方法。
12.一种可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时，用于实现如权利要求1至9中任一项所述的基于卷积神经网络的交通标志识别方法。
CN202010022391.2A 2020-01-09 2020-01-09 一种基于卷积神经网络的交通标志识别方法和系统 Active CN111199217B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202010022391.2A CN111199217B (zh) 2020-01-09 2020-01-09 一种基于卷积神经网络的交通标志识别方法和系统

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202010022391.2A CN111199217B (zh) 2020-01-09 2020-01-09 一种基于卷积神经网络的交通标志识别方法和系统

Publications (2)

Publication Number Publication Date
CN111199217A true CN111199217A (zh) 2020-05-26
CN111199217B CN111199217B (zh) 2023-03-28

Family

ID=70747172

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202010022391.2A Active CN111199217B (zh) 2020-01-09 2020-01-09 一种基于卷积神经网络的交通标志识别方法和系统

Country Status (1)

Country Link
CN (1) CN111199217B (zh)

Cited By (5)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN112712126A (zh) * 2021-01-05 2021-04-27 南京大学 一种图片识别方法
CN112906808A (zh) * 2021-03-05 2021-06-04 华南师范大学 基于卷积神经网络的图像分类方法、系统、装置及介质
CN113536943A (zh) * 2021-06-21 2021-10-22 上海赫千电子科技有限公司 一种基于图像增强的道路交通标志识别方法
CN114882469A (zh) * 2022-04-07 2022-08-09 淮阴工学院 一种基于dl-ssd模型的交通标志检测方法及系统
CN116704476A (zh) * 2023-06-12 2023-09-05 郑州轻工业大学 一种基于改进Yolov4-tiny算法的交通标志检测方法

Citations (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
WO2016155371A1 (zh) * 2015-03-31 2016-10-06 百度在线网络技术（北京）有限公司 识别交通标志的方法和装置
CN107292869A (zh) * 2017-06-07 2017-10-24 西安电子科技大学 基于各向异性高斯核和梯度搜索的图像斑点检测方法
CN109635784A (zh) * 2019-01-10 2019-04-16 重庆邮电大学 基于改进的卷积神经网络的交通标志识别方法
US20190251446A1 (en) * 2018-02-15 2019-08-15 Adobe Inc. Generating visually-aware item recommendations using a personalized preference ranking network

Patent Citations (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
WO2016155371A1 (zh) * 2015-03-31 2016-10-06 百度在线网络技术（北京）有限公司 识别交通标志的方法和装置
CN107292869A (zh) * 2017-06-07 2017-10-24 西安电子科技大学 基于各向异性高斯核和梯度搜索的图像斑点检测方法
US20190251446A1 (en) * 2018-02-15 2019-08-15 Adobe Inc. Generating visually-aware item recommendations using a personalized preference ranking network
CN109635784A (zh) * 2019-01-10 2019-04-16 重庆邮电大学 基于改进的卷积神经网络的交通标志识别方法

Non-Patent Citations (1)

* Cited by examiner, † Cited by third party
Title
张邯等: "基于优化的卷积神经网络在交通标志识别中的应用", 《现代电子技术》 *

Cited By (8)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN112712126A (zh) * 2021-01-05 2021-04-27 南京大学 一种图片识别方法
CN112712126B (zh) * 2021-01-05 2024-03-19 南京大学 一种图片识别方法
CN112906808A (zh) * 2021-03-05 2021-06-04 华南师范大学 基于卷积神经网络的图像分类方法、系统、装置及介质
CN113536943A (zh) * 2021-06-21 2021-10-22 上海赫千电子科技有限公司 一种基于图像增强的道路交通标志识别方法
CN113536943B (zh) * 2021-06-21 2024-04-12 上海赫千电子科技有限公司 一种基于图像增强的道路交通标志识别方法
CN114882469A (zh) * 2022-04-07 2022-08-09 淮阴工学院 一种基于dl-ssd模型的交通标志检测方法及系统
CN114882469B (zh) * 2022-04-07 2023-08-25 淮阴工学院 一种基于dl-ssd模型的交通标志检测方法及系统
CN116704476A (zh) * 2023-06-12 2023-09-05 郑州轻工业大学 一种基于改进Yolov4-tiny算法的交通标志检测方法

Also Published As

Publication number Publication date
CN111199217B (zh) 2023-03-28

Similar Documents

Publication Publication Date Title
CN111199217B (zh) 一种基于卷积神经网络的交通标志识别方法和系统
US10282589B2 (en) Method and system for detection and classification of cells using convolutional neural networks
CN108615010B (zh) 基于平行卷积神经网络特征图融合的人脸表情识别方法
Shih et al. Real-time object detection with reduced region proposal network via multi-feature concatenation
CN109583322B (zh) 一种人脸识别深度网络训练方法和系统
Mayr et al. Weakly supervised segmentation of cracks on solar cells using normalized L p norm
US10713563B2 (en) Object recognition using a convolutional neural network trained by principal component analysis and repeated spectral clustering
US20190236411A1 (en) Method and system for multi-scale cell image segmentation using multiple parallel convolutional neural networks
CN110321967B (zh) 基于卷积神经网络的图像分类改进方法
US20170124409A1 (en) Cascaded neural network with scale dependent pooling for object detection
CN109711264B (zh) 一种公交车道占道检测方法及装置
CN112446388A (zh) 一种基于轻量化二阶段检测模型的多类别蔬菜幼苗识别方法及系统
CN111898621B (zh) 一种轮廓形状识别方法
CN106407986A (zh) 一种基于深度模型的合成孔径雷达图像目标识别方法
CN108416270B (zh) 一种基于多属性联合特征的交通标志识别方法
Daood et al. Sequential recognition of pollen grain Z-stacks by combining CNN and RNN
Lin et al. Determination of the varieties of rice kernels based on machine vision and deep learning technology
CN111832580B (zh) 结合少样本学习与目标属性特征的sar目标识别方法
CN112364705A (zh) 基于多层次特征融合的轻量型cnn的表情识别方法
CN112784754A (zh) 一种车辆再识别方法、装置、设备及存储介质
CN112347936A (zh) 一种基于深度可分离卷积的快速目标检测方法
CN114492634B (zh) 一种细粒度装备图片分类识别方法及系统
Uthaib et al. Multiclassification of license plate based on deep convolution neural networks
CN111310837A (zh) 车辆改装识别方法、装置、系统、介质和设备
Yuxin et al. A highway entrance vehicle logo recognition system based on convolutional neural network

Legal Events

Date Code Title Description
PB01 Publication
PB01 Publication
SE01 Entry into force of request for substantive examination
SE01 Entry into force of request for substantive examination
GR01 Patent grant
GR01 Patent grant