YOLOv3-A: a traffic sign detection network based on attention mechanism

被引：0
|
作者：
Guo F. [1 ]
Zhang Y. [1 ]
Tang J. [1 ]
Li W. [1 ]
机构：
[1] School of Automation, Central South University, Changsha
来源：
Tongxin Xuebao/Journal on Communications | 2021年 / 42卷 / 01期
基金：
中国国家自然科学基金;
关键词：
Attention mechanism; Semantic segmentation; Small target detection; Traffic sign detection;
D O I：
10.11959/j.issn.1000-436x.2021031
中图分类号：
学科分类号：
摘要：
To solve the problem that the existing YOLOv3 algorithm had more false detections and missed detections for traffic sign detection task with small target problems and complex background, based on the YOLOv3, a channel attention method for target detection and a spatial attention method based on semantic segmentation guidance were proposed to form the YOLOv3-A (attention) algorithm. The detection features in the channel and spatial dimensions were recalibrated, allowing the network to focus and enhance the effective features, and suppress interference features, which greatly improved the detection performance. Experiments on the TT100K traffic sign data set show that the algorithm improves the detection performance of small targets, and the accuracy and recall rate of the YOLOv3 are improved by 1.9% and 2.8% respectively. © 2021, Editorial Board of Journal on Communications. All right reserved.
引用
收藏
页码：87 / 99
页数：12
相关论文
共 28 条
  • [21] HE K, GKIOXARI G, DOLLAR P, Et al., Mask R-CNN, IEEE Transactions on Pattern Analysis and Machine Intelligence, 42, 2, pp. 386-397, (2020)
  • [22] SZEGEDY C, LIU W, JIA Y, Et al., Going deeper with convolutions, 2015 IEEE Conference on Computer Vision and Pattern Recognition, pp. 1-9, (2015)
  • [23] CHEN L C, PAPANDREOU G, SCHROFF F, Et al., Rethinking atrous convolution for semantic image segmentation, (2017)
  • [24] UIJLINGS J R R, VAN D S K E A, GEVERS T, Et al., Selective search for object recognition, International Journal of Computer Vision, 104, 2, pp. 154-171, (2013)
  • [25] ZHU Z, LIANG D, ZHANG S, Et al., Traffic-sign detection and classification in the wild, 2016 IEEE Conference on Computer Vision and Pattern Recognition, pp. 2110-2118, (2016)
  • [26] LARSSON F, FELSBERG M, FORSSEN P E., Correlating Fourier descriptors of local patches for road sign recognition, IET Computer Vision, 5, 4, pp. 244-254, (2011)
  • [27] MOGELMOSE A, TRIVEDI M M, MOESLUND T B., Vision based traffic sign detection and analysis for intelligent driver assistance systems: perspectives and survey, IEEE Transactions on Intelligent Transportation Systems, 13, 4, pp. 1484-1497, (2012)
  • [28] SERMANET P, EIGEN D, ZHANG X, Et al., Overfeat: integrated recognition, localization and detection using convolutional networks, (2013)
123