Zhe LI, Hui-hui ZHANG, Jun-yong DENG. Traffic sign detection algorithm based on improved Faster R-CNN[J]. Chinese journal of liquid crystals and displays, 2021, 36(3): 484-492.
DOI:
Zhe LI, Hui-hui ZHANG, Jun-yong DENG. Traffic sign detection algorithm based on improved Faster R-CNN[J]. Chinese journal of liquid crystals and displays, 2021, 36(3): 484-492. DOI: 10.37188/CJLCD.2020-0195.
Traffic sign detection algorithm based on improved Faster R-CNN
Aiming at the problems of complex background and small traffic sign target in large view traffic scene
an improved Faster R-CNN detection network algorithm is proposed. Firstly
the deep residual network ResNet50 is used as the backbone network to extract the features of traffic signs. Secondly
the strategy of using reasonable scale sliding window on two different level feature maps is designed to generate the target proposal region to enhance the detection ability of multi-scale traffic signs. Finally
the attention mechanism module is introduced into the residual block to strengthen the key information of the image and suppress the image background information. The validity of the algorithm is verified on the Chinese traffic sign dataset
with an average detection accuracy of 98.52% and a detection rate of 0.042 s per image. The detection effect of the improved algorithm is obviously better than the original Faster R-CNN detection method
and is more suitable for traffic sign detection in complex scenes
YANG J S, YANG Y N, LI T J. Traffic sign recognition algorithm based on depthwise separable convolutions[J]. Chinese Journal of Liquid Crystals and Displays , 2019, 34(12): 1191-1201. (in Chinese)
SURINWARANGKOON T, NITSUWAT S, MOORE E J. Traffic sign recognition by color filtering and particle swarm optimization[C]// Proceedings of the 2012 4 th International Conference on Computer Research and Development . Singapore: IACSIT Press, 2012: 58-62.
CHEN Y X, XIE Y, WANG Y L. Detection and recognition of traffic signs based on HSV vision model and shape features[J]. Journal of Computers , 2013, 8(5): 1366-1370.
KHAN J F, BHUIYAN S M A, ADHAMI R R. Image segmentation and shape analysis for road-sign detection[J]. IEEE Transactions on Intelligent Transportation Systems , 2011, 12(1): 83-96.
MEUTER M, NUNN C, GORMER S M, et al . A decision fusion and reasoning module for a traffic sign recognition system[J]. IEEE Transactions on Intelligent Transportation Systems , 2011, 12(4): 1126-1134.
CHEN L, PAN Z M, MAO Q Z, et al . HOG-LBP adaptable fused features based method for forbidden traffic signs detection[J]. Geomatics and Information Science of Wuhan University, 2013, 38(2): 191-194. (in Chinese)
GROLÉAT T, ARZEL M, VATON S. Stretching the edges of SVM traffic classification with FPGA acceleration[J]. IEEE Transactions on Network and Service Management , 2014, 11(3): 278-291.
CHEN T, LU S J. Accurate and efficient traffic sign detection using discriminative AdaBoost and support vector regression[J]. IEEE Transactions on Vehicular Technology , 2016, 65(6): 4006-4015.
HUANG L H, CAO L H, LI N, et al . A state perception method for infrared dim and small targets with deep learning[J]. Chinese Optics , 2020, 13(3): 527-536. (in Chinese)
TONG Y, YANG H C. Real-time traffic sign detection method based on improved convolution neural network[J] . Laser & Optoelectronics Progress , 2019, 56(7): 071003. (in Chinese)
GIRSHICK R, DONAHUE J, DARRELL T, et al . Rich feature hierarchies for accurate object detection and semantic segmentation[C]// Proceedings of 2014 IEEE Conference on Computer Vision and Pattern Recognition . Columbus: IEEE, 2014: 580-587.
GIRSHICK R. Fast R-CNN[C]// Proceedings of 2015 IEEE International Conference on Computer Vision . Santiago: IEEE, 2015: 1440-1448.
REN S Q, HE K M, GIRSHICK R, et al . Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence , 2017, 39(6): 1137-1149.
WANG C Z, AN J S, JIANG X J, et al . Region proposal optimization algorithm based on convolutional neural networks[J]. Chinese Optics , 2019, 12(6): 1348-1361. (in Chinese)
SONG Q S, WANG X L, ZHANG C, et al . A residual SSD model based on window size clustering for traffic sign detection[J]. Journal of Hunan University ( Natural Sciences ), 2019, 46(10): 133-140. (in Chinese)
HE K M, ZHANG X Y, REN S Q, et al . Deep residual learning for image recognition[C]// Proceedings of 2016 IEEE Conference on Computer Vision and Pattern Recognition . Las Vegas: IEEE, 2016: 770-778.
WOO S, PARK J, LEE J Y, et al . CBAM: convolutional block attention module[C]// Proceedings of the 15 th European Conference on Computer Vision . Munich: Springer, 2018: 3-19.
CAI Z W, FAN Q F, FERIS R S, et al . A unified multi-scale deep convolutional neural network for fast object detection[C]// Proceedings of the 14 th European Conference on Computer Vision . Amsterdam: Springer, 2016: 354-370.
HU J, SHEN L, SUN G. Squeeze-and-excitation networks[C]// Proceedings of 2018 IEEE / CVF Conference on Computer Vision and Pattern Recognition . Salt Lake City: IEEE, 2018: 7132-7141.
Chinese traffic sign database[DB/OL]. (2017-02-20)[2020-05-12] . http://www.nlpr.ia.ac.cn/pal/trafficdata/detection.html http://www.nlpr.ia.ac.cn/pal/trafficdata/detection.html .