摘要
针对卷积神经网络在交通标志识别实时性不好,对设备硬件要求过高的缺点,提出了一种具有实时性,高精度的基于轻量型卷积神经网络的改进网络。一方面引入深度可分离卷积和激活函数Mish,加快网络的训练和识别速度,降低对硬件设备的要求;另一方面通过对网络架构及层次的改进,同时合理改变卷积核的大小和数目,加强图片特征的表达与传递。在BelgiumTSC交通标志数据集上的实验结果表明,改进后网络明显提高了网络训练速度,同时识别精度也略高于原网络,验证了改进方法的有效性。通过与其他模型相比,该模型能够更快速准确完成交通标志识别任务,验证了该方法的可行性。
Aiming at the shortcomings of convolutional neural network in the recognition of traffic signs that the real-time performance is not good,and the equipment hardware requirements are too high,a real-time and high-precision improved network based on lightweight convolutional neural network is proposed.Separate convolution and activation function Mish,speed up the network training and recognition speed,reduce the requirements for hardware equipment;on the other hand,through the improvement of the network architecture and level,while reasonably changing the size and number of convolution kernels,the expression of image features and transfer.The experimental results on the BelgiumTSC traffic sign dataset show that the improved network significantly increases the network training speed,and the recognition accuracy is slightly higher than the original network,which verifies the effectiveness of the method in this paper.Compared with other models,this model can complete the task of traffic sign recognition more quickly and accurately,which verifies the feasibility of this method.
作者
龙曼仪
李茂军
张辉
刘芾
LONG Man-yi;LI Mao-jun;ZhANG Hui;LIU Fu(College of Electrical and Information Engineering,Changsha University of Science and Technology,Changsha,Hunan 410114,China;College of Robotics,Hunan University,Changsha,Hunan 410114,China)
出处
《计算技术与自动化》
2020年第4期112-118,共7页
Computing Technology and Automation
基金
国家自然科学基金资助项目(61971071)。
关键词
卷积神经网络
交通标识
图像增强
深度可分离卷积
激活函数
convolutional neural network
traffic sign
image processing
depth separable convolution
activation function