Paper Info
Title
Traffic sign detection algorithm based on improved YOLOv4-Tiny
Abstract
There are three problems in YOLOv4-Tiny when it is used for traffic sign detection: the feature pyramid network fails to fuse high-level and low-level features sufficiently, the importance of low-level features for small object detection is not considered, and the ability to extract the features of small objects in the backbone network is not strong. Focusing on these problems, this paper proposes an improved YOLOv4 -Tiny for real-time traffic sign detection. Firstly, this paper improves YOLOv4-Tiny’s feature fusion method and proposes an adaptive feature pyramid network (AFPN), which aims to adaptively fuse the two feature layers with different scales. Secondly, two receptive field blocks (RFB) are added after the two feature layers of the backbone network. These two RFBs are composed of multi-branch structures and dilated convolution with different dilation rates, which can enhance the feature extraction ability of the backbone network. The CCTSDB and GTSDB datasets are used to evaluate the effectiveness of the improved method. The experimental results show that our proposed network is superior to the original network in the precision, recall rate, and mAP. In addition, compared with other state-of-the-art approaches on traffic sign detection, our proposed network has good comprehensive performance in accuracy and speed. The above results show that our improved method is effective in improving the performance of traffic sign detection.
Year
DOI
Venue
2022
10.1016/j.image.2022.116783
Signal Processing: Image Communication
Keywords
DocType
Volume
Traffic sign detection,YOLOv4-tiny,Small object,Adaptive feature pyramid network,Receptive field block
Journal
107
ISSN
Citations 
PageRank 
0923-5965
0
0.34
References 
Authors
0
5
Name
Order
Citations
PageRank
Yingbiao Yao100.34
Li Han200.34
Chenjie Du311.36
Xin Xu416240.08
Xianyang Jiang500.34