Traffic sign recognition algorithm based on depthwise separable convolutions

被引:1
|
作者
Yang Jin-sheng [1 ]
Yang Yan-nan [1 ]
Li Tian-jiao [1 ]
机构
[1] Tianjin Univ, Sch Microelect, Tianjin 030072, Peoples R China
关键词
image processing; target detection; traffic signs; depth separable convolution; YOLOv3; h-swish;
D O I
10.3788/YJYXS20193412.1191
中图分类号
O7 [晶体学];
学科分类号
0702 ; 070205 ; 0703 ; 080501 ;
摘要
In order to solve the problem of low precision detection for small and medium-sized targets in the real-time traffic sign recognition, a lightweight traffic sign detection network based on YOLOv3-tiny depthwise separable convolution is proposed. In this paper, a deep separable convolution module is constructed by using deep separable convolution instead of the common convolution to construction feature extraction network, the feature information of small and medium-sized targets is better extracted under the premise of ensuring the calculation amount.At the same time, the multi-scale feature fusion network is improved to improve the detection accuracy of small and medium-sized traffic signs. The h-swish activation function is used to reduce the image features lost due to the increase of the number of network layers, and the detection of multiple types of traffic signs is realized.The experimental results show that the algorithm effectively improves the detection of small and medium-sized traffic signs. Warining, mandatory and prohibitory traffic signs are detected on the verification set. The detection accuracy (AP) is 98.57% , 96.03% and 98.04% respectively. The average detection accuracy (mAP) was 97.54% and the detection speed was 201.5 f/s. The average accuracy was 14.01% higher than that of YOLOv3-tiny. The algorithm effectively improves the detection accuracy under the premise of ensuring low calculation of light network and good timeliness of detection.
引用
收藏
页码:1191 / 1201
页数:11
相关论文
共 17 条
  • [1] Chieng H.H., 2018, Int.J. Adv. Intell. Informat., V4, P76, DOI [10.26555/ijain.v4i2.249, DOI 10.26555/IJAIN.V4I2.249]
  • [2] Xception: Deep Learning with Depthwise Separable Convolutions
    Chollet, Francois
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 1800 - 1807
  • [3] Detection and Recognition of Traffic Planar Objects Using Colorized Laser Scan and Perspective Distortion Rectification
    Deng, Zhidong
    Zhou, Lipu
    [J]. IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2018, 19 (05) : 1485 - 1495
  • [4] Everingham M., The PASCAL Visual Object Classes Challenge 2012 (VOC2012) Results
  • [5] Girshick R., 2014, IEEE COMP SOC C COMP, DOI [10.1109/CVPR.2014.81, DOI 10.1109/CVPR.2014.81]
  • [6] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [7] HUANG S Z, 2018, P 2018 IEEE INT C MU
  • [8] Jo K, 2017, IEEE I C SIGNAL IMAG, P507, DOI 10.1109/ICSIPA.2017.8120665
  • [9] ImageNet Classification with Deep Convolutional Neural Networks
    Krizhevsky, Alex
    Sutskever, Ilya
    Hinton, Geoffrey E.
    [J]. COMMUNICATIONS OF THE ACM, 2017, 60 (06) : 84 - 90
  • [10] SSD: Single Shot MultiBox Detector
    Liu, Wei
    Anguelov, Dragomir
    Erhan, Dumitru
    Szegedy, Christian
    Reed, Scott
    Fu, Cheng-Yang
    Berg, Alexander C.
    [J]. COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 : 21 - 37