基于Inception卷积模块的国内交通标志分类方法技术

本发明专利技术公开了基于Inception卷积模块的国内交通标志分类方法，构建国内交通标志数据集；然后构建基于Inception卷积模块的卷积神经网络；将训练集图像数据输入卷积神经网络，经训练得到模型；最后将测试集图像数据输入训练后模型中，得到预测结果。本发明专利技术采用卷积层与池化层组合替代平铺层和全链接层，并使用简化Inception卷积模块，能够有效减少卷积神经网络中参数数量，减少训练次数。

Domestic Traffic Sign Classification Method Based on Inception Convolution Module

【技术实现步骤摘要】
基于Inception卷积模块的国内交通标志分类方法
本专利技术涉及图像识别
，具体涉及一种基于Inception卷积模块的国内交通标志分类方法。
技术介绍
近几年随着人工智能的火爆，自动驾驶技术也越来越多的被学者、专家所重视，如何准确地获取自动驾驶所必须的路况信息成为一个热点。交通标志是路况信息中不可缺少的一环，其所包含信息决定了自动驾驶系统应以何种模式运行。交通标志牌中含有大量信息，如限行、限速、限高以及危险路况的警示与预告等等，给道路安全提供了重要的辅助作用。交通标志牌的识别算法大致上可分为两类：模板匹配法和神经网络法。神经网络法的高识别率和自学习功能成为它的极大优势。传统神经网络的最后几层通常使用全链接层传输给Softmax层，这种方式使得处理分类较多的网络结构时，最后几层全链接层会带来极多的参数，导致训练速度较慢。
技术实现思路
本专利技术的目的在于一种基于Inception卷积模块的国内交通标志分类方法，以克服上述现有技术存在的缺陷，本专利技术简化传统卷积神经网络的结构，使用卷积层与池化层结合代替全链接层和平铺层作用并将参数传递给Softmax层，并采用简化的Softmax卷积模块，简化整体卷积神经网络结构，减少参数数量，在保证较高准确率的情况下加快训练速度。为达到上述目的，本专利技术采用如下技术方案：基于Inception卷积模块的国内交通标志分类方法，包括以下步骤：步骤1：构建国内交通标志数据集，并根据该数据集构建训练集和测试集；步骤2：构建基于Inception卷积模块的卷积神经网络；步骤3：将训练集图像数据输入步骤2构建的卷积神经网络，经训练得到训练后的模型；步骤4：将测试集图像数据输入训练后模型中，得到预测结果。进一步地，步骤1中的国内交通标志数据集中图像尺寸为64×64×3的RGB彩色图像，其训练集与测试集中数据个数比为8:2。进一步地，步骤2中所述的卷积神经网络中，包含3个卷积层、2个Inception卷积模块、4个池化层，其中：第一层为卷积层Conv1，其卷积核大小为3×3，卷积核数目为9，卷积步长为1，填充方式使用SAME填充方式，经过Conv1即得到64×64×9的特征图；第二层为Inception卷积模块Inception_Conv2，其有三个分支，分别为Branch2_0、Branch2_1、Branch2_2，其中Branch2_0卷积核大小为1×1，卷积核数目为24，卷积步长为1，填充方式使用SAME填充方式；Branch2_1使用两个级联的卷积核，分别为16个1×1的卷积核和24个5×5的卷积核，卷积步长为1，填充方式使用SAME填充方式；Branch2_2使用三个级联的卷积核，分别为32个1×1、64个3×3和64个3×3的卷积核，卷积步长为1，填充方式使用SAME填充方式，经过Inception_Conv2即得到64×64×112的特征图；第三层为池化层Pool3，池化方式为最大值池化，池化尺寸为2×2，池化步长为2，经过Pool3即得到32×32×112的特征图；第四层为Inception卷积模块Inception_Conv4，其有三个分支，分别为Branch4_0、Branch4_1、Branch4_2，其中Branch4_0卷积核大小为1×1，卷积核数目为64，卷积步长为1，填充方式使用SAME填充方式；Branch4_1使用两个级联的卷积核，分别为48个1×1的卷积核和64个5×5的卷积核，卷积步长为1，填充方式使用SAME填充方式；Branch4_2使用三个级联的卷积核，分别为64个1×1、96个3×3和96个3×3的卷积核，卷积步长为1，填充方式使用SAME填充方式，经过Inception_Conv4即得到32×32×224的特征图；第五层为池化层Pool5，池化方式为最大值池化，池化尺寸为2×2，池化步长为2，经过Pool5即得到16×16×224的特征图；第六层为卷积层Conv6，其卷积核大小为5×5，卷积核数目为112，卷积步长为1，填充方式使用VALID填充方式，经过Conv6即得到12×12×112的特征图；第七层为池化层Pool7，池化方式为最大值池化，池化尺寸为2×2，池化步长为2，经过Pool7即得到6×6×112的特征图；第八层为卷积层Conv8，其卷积核大小为5×5，卷积核数目为64，卷积步长为1，填充方式使用VALID填充方式，经过Conv6即得到2×2×64的特征图；第九层为池化层Pool9，池化方式为最大值池化，池化尺寸为2×2，池化步长为2，经过Pool9即得到1×1×112的特征向量，并输入到Softmax层进行分类输出。进一步地，步骤2中卷积神经网络的每一层都使用BN层对输入数据进行规范化处理。进一步地，步骤3中对卷积神经网络的训练的超参数设置如下：批量样本数量Mini_batch＝128，迭代次数Train_num＝500次，初始学习率Learn_rate＝0.05，使用指数衰减法对学习率进行衰减，衰减因子Deacy＝0.96，BN层的权重衰减项设为Weight_Deacy＝0.005。进一步地，步骤3中卷积神经网络使用小批量梯度下降法进行训练。与现有技术相比，本专利技术具有以下有益的技术效果：本专利技术采用简化的Inception卷积模块，原Inception卷积模块使用不同大小的卷积核对特征图进行卷积，使不同尺度的特征融合在一起，且由网络自身决定是否需要这些参数，使得网络在特征提取和目标分类上由显著优势。本专利技术使用简化后的Inception卷积模块，减少原模块中无用的参数和运算量，并使用卷积层与池化层结合替代全链接层和平铺层作用，减少网络参数的同时极大精简了网络结构并减少训练时间，在保证较高准确率的情况下加快训练速度。附图说明图1是本专利技术的流程示意图；图2是卷积过程示意图；图3是最大值池化和平均池化过程的池化示意图；图4是梯度下降示意图；图5是简化Inception卷积模块结构示意图；图6是本专利技术网络结构示意图；图7是本专利技术卷积神经网络中第二层的可视化特征图；图8是本专利技术卷积神经网络中第三层的可视化特征图；图9是本专利技术卷积神经网络中第四层的可视化特征图；在图7、图8、图9中，(a)、(b)、(c)、(d)分别为同一输入特征图下不同层的特征可视化图像；图10是本专利技术的训练集损失函数曲线；图11是本专利技术的训练集学习衰减率曲线；图12是本专利技术的训练集正确率曲线。具体实施方式下面结合附图对本专利技术作进一步详细描述：参见图1，本专利技术提供一种基于Inception卷积模块的国内交通标志分类方法，设计了一种基于Inception卷积模块的卷积神经网络。具体思路是，使用卷积层与池化层结合代替全链接层和平铺层作用并将参数传递给Softmax层，并采用简化的Softmax卷积模块，简化整体卷积神经网络结构，减少参数数量，在保证较高准确率的情况下加快训练速度。参见图2，其中为卷积层中使用卷积核对特征图像滤波的过程示意图；参见图3，其中为池化层中对特征图像分别使用最大值池化和平均值池化的过程示意图；参见图4，为使用梯度下降法对训练参数优化的过程示意图，其图中曲面为训练过程中所使用的损失函数，图中曲线为被训练参数沿其梯度方向进行训练更新的路径本文档来自技高网...

【技术保护点】
1.基于Inception卷积模块的国内交通标志分类方法，其特征在于，包括以下步骤：步骤1：构建国内交通标志数据集，并根据该数据集构建训练集和测试集；步骤2：构建基于Inception卷积模块的卷积神经网络；步骤3：将训练集图像数据输入步骤2构建的卷积神经网络，经训练得到训练后的模型；步骤4：将测试集图像数据输入训练后模型中，得到预测结果。

【技术特征摘要】
1.基于Inception卷积模块的国内交通标志分类方法，其特征在于，包括以下步骤：步骤1：构建国内交通标志数据集，并根据该数据集构建训练集和测试集；步骤2：构建基于Inception卷积模块的卷积神经网络；步骤3：将训练集图像数据输入步骤2构建的卷积神经网络，经训练得到训练后的模型；步骤4：将测试集图像数据输入训练后模型中，得到预测结果。2.根据权利要求1所述的基于Inception卷积模块的国内交通标志分类方法，其特征在于，步骤1中的国内交通标志数据集中图像尺寸为64×64×3的RGB彩色图像，其训练集与测试集中数据个数比为8:2。3.根据权利要求1所述的基于Inception卷积模块的国内交通标志分类方法，其特征在于，步骤2中所述的卷积神经网络中，包含3个卷积层、2个Inception卷积模块、4个池化层，其中：第一层为卷积层Conv1，其卷积核大小为3×3，卷积核数目为9，卷积步长为1，填充方式使用SAME填充方式，经过Conv1即得到64×64×9的特征图；第二层为Inception卷积模块Inception_Conv2，其有三个分支，分别为Branch2_0、Branch2_1、Branch2_2，其中Branch2_0卷积核大小为1×1，卷积核数目为24，卷积步长为1，填充方式使用SAME填充方式；Branch2_1使用两个级联的卷积核，分别为16个1×1的卷积核和24个5×5的卷积核，卷积步长为1，填充方式使用SAME填充方式；Branch2_2使用三个级联的卷积核，分别为32个1×1、64个3×3和64个3×3的卷积核，卷积步长为1，填充方式使用SAME填充方式，经过Inception_Conv2即得到64×64×112的特征图；第三层为池化层Pool3，池化方式为最大值池化，池化尺寸为2×2，池化步长为2，经过Pool3即得到32×32×112的特征图；第四层为Inception卷积模块Inception_Conv4，其有三个分支，分别为Branch4_0、Branch4_1、Branch4_2，其中Branch4_0卷积核大小为1×1，卷积核数目为...

【专利技术属性】
技术研发人员：黄鹤，郭璐，许哲，茹锋，胡凯益，王会峰，汪贵平，黄莺，惠晓滨，
申请(专利权)人：长安大学，
类型：发明
国别省市：陕西,61