基于轻量型卷积神经网络的交通标志识别方法
作者:
基金项目:

黑龙江省自然科学基金(LH2019F042);大庆市科技计划(zd-2019-08);东北石油大学创新基金(2017PYZL-06,2018YDL-22)


Traffic Sign Recognition Method Based on Lightweight Convolutional Neural Network
Author:
  • 摘要
  • | |
  • 访问统计
  • |
  • 参考文献 [16]
  • |
  • 相似文献 [20]
  • | | |
  • 文章评论
    摘要:

    交通标志识别设备的功耗和硬件性能较低,而现有卷积神经网络模型内存占用高、训练速度慢、计算开销大,无法应用于识别设备.针对此问题,为降低模型存储,提升训练速度,引入深度可分离卷积和混洗分组卷积并与极限学习机相结合,提出两种轻量型卷积神经网络模型:DSC-ELM模型和SGC-ELM模型.模型使用轻量化卷积神经网络提取特征后,将特征送入极限学习机进行分类,解决了卷积神经网络全连接层参数训练慢的问题.新模型结合了轻量型卷积神经网络模型内存占用低、提取特征质量好以及ELM的泛化性好、训练速度快的优点.实验结果表明.与其他模型相比,该混合模型能够更加快速准确地完成交通标志识别任务.

    Abstract:

    Traffic sign recognition equipment has low power consumption and hardware performance, while the existing convolutional neural network model has high memory footprint, slow training speed, and high computational overhead, which cannot be applied to the recognition equipment. To solve this problem, in order to reduce model storage and improve training speed, deep separation convolution and mixed wash grouping convolution are introduced and combined with the ultimate learning machine. Two lightweight convolutional neural network models are proposed:DSC-ELM model and SGC-ELM model. The proposed models use the lightweight convolutional neural network to extract the features, and then send the features to the extreme learning machine for classification, which solve the problem of slow parameter training in the full connection layer of the convolutional neural network. The new models combine the advantages of lightweight convolutional neural network model with low memory footprint, good feature extraction quality, good generalization of ELM, and fast training and classification. Experimental results show that compared with other models, the hybrid model can accomplish traffic sign recognition tasks more quickly and accurately.

    参考文献
    [1] Gudigar A, Chokkadi S, Raghavendra U. A review on automatic detection and recognition of traffic sign. Multimedia Tools and Applications, 2016, 75(1):333-364.[doi:10.1007/s11042-014-2293-7
    [2] Jin J Q, Fu K, Zhang CS. Traffic sign recognition with hinge loss trained convolutional neural networks. IEEE Transactions on Intelligent Transportation Systems, 2014, 15(5):1991-2000.[doi:10.1109/TITS.2014.2308281
    [3] Wang GY, Ren GH, Wu ZL, et al. A hierarchical method for traffic sign classification with support vector machines. Proceedings of 2013 International Joint Conference on Neural Networks. Dallas, TX, USA. 2013. 1-6.
    [4] Zhang Q, He N, Chi Y, et al. Traffic sign detection and recognition based on the improved SIFT algorithm. Journal of Beijing Union University, 2017, 31(2):59-65
    [5] Guo ZH, Zhang L, Zhang D. A completed modeling of local binary pattern operator for texture classification. IEEE Transactions on Image Processing, 2010, 19(6):1657-1663.[doi:10.1109/TIP.2010.2044957
    [6] Krizhevsky A, Sutskever I, Hinton GE. ImageNet classification with deep convolutional neural networks. Communications of the ACM, 2017, 60(6):84-90.[doi:10.1145/3065386
    [7] Szegedy C, Liu W, Jia YQ, et al. Going deeper with convolutions. Proceedings of 2015 IEEE Conference on Computer Vision and Pattern Recognition. Boston, MA, USA. 2015. 1-9.
    [8] Simonyan K, Zisserman A. Very deep convolutional networks for large-scale image recognition. arXiv:1409.1556, 2014.
    [9] He KM, Zhang XY, Ren SQ, et al. Deep residual learning for image recognition. Proceedings of 2016 IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas, NV, USA. 2016.
    [10] Howard AG, Zhu ML, Chen B, et al. MobileNets:Efficient convolutional neural networks for mobile vision applications. arXiv:1704.04861, 2017.
    [11] Zhang XY, Zhou XY, Lin MX, et al. ShuffleNet:An extremely efficient convolutional neural network for mobile devices. Proceedings of 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Salt Lake City, UT, USA. 2018. 6848-6856.
    [12] Huang GB, Zhou HM, Ding XJ, et al. Extreme learning machine for regression and multiclass classification. IEEE Transactions on Systems, Man, and Cybernetics, Part B (Cybernetics), 2012, 42(2):513-529.[doi:10.1109/TSMCB.2011.2168604
    [13] Stallkamp J, Schlipsing M, Salmen J, et al. The German traffic sign recognition benchmark:A multi-class classification competition. Proceedings of 2011 International Joint Conference on Neural Networks. San Jose, CA, USA. 2011. 1453-1460.
    [14] Mathias M, Timofte R, Benenson R, et al. Traffic sign recognition-How far are we from the solution? Proceedings of 2013 International Joint Conference on Neural Networks. Dallas, TX, USA. 2013. 1-8.
    [15] 刘志刚, 李盼池, 许少华. 基于ELM-PNN算法的第24周太阳黑子预测预报. 控制与决策, 2017, 32(4):642-646
    [16] 闫河, 王鹏, 董莺艳, 罗成, 等. 一种CNN与ELM相结合的船舶分类识别方法. 重庆理工大学学报(自然科学), 2019, 33(1):53-57
    引证文献
    网友评论
    网友评论
    分享到微博
    发 布
引用本文

程越,刘志刚.基于轻量型卷积神经网络的交通标志识别方法.计算机系统应用,2020,29(2):198-204

复制
分享
文章指标
  • 点击次数:1497
  • 下载次数: 2940
  • HTML阅读次数: 1830
  • 引用次数: 0
历史
  • 收稿日期:2019-07-09
  • 最后修改日期:2019-07-26
  • 在线发布日期: 2020-01-16
  • 出版日期: 2020-02-15
文章二维码
您是第12820687位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京海淀区中关村南四街4号 中科院软件园区 7号楼305房间,邮政编码:100190
电话:010-62661041 传真: Email:csa (a) iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号