Paper
11 October 2023 Research on traffic sign detection based on improved YOLOv4
Jie Cao, Peng-hui Li, Hong Zhang, Zong-li Liu
Author Affiliations +
Proceedings Volume 12800, Sixth International Conference on Computer Information Science and Application Technology (CISAT 2023); 128002H (2023) https://doi.org/10.1117/12.3003843
Event: 6th International Conference on Computer Information Science and Application Technology (CISAT 2023), 2023, Hangzhou, China
Abstract
Traffic signs contain important traffic information. The traditional traffic sign detection method cannot solve the problem of low detection accuracy caused by the small, occupied area of traffic sign images. Based on this, a traffic sign detection algorithm based on improved YOLOv4 is proposed. Firstly, the 13 × 13 large receptive field detection layer is removed on the YOLOv4 structure, and the 104 × 104 detection layer is added. It obtains more global feature information and improves detection accuracy. The attention mechanism is introduced into the algorithm, that is, the backbone network extracts three feature layers and then adds the scSE module. Make the network focus on the target area and improve the algorithm detection ability. Secondly, in order to speed up the convergence of the network and improve the detection accuracy, a dynamic residual connection is added to the backbone network. It promotes the spread of well-performing signals. And use the decoupled-head detection head to use different branches to calculate classification and positioning tasks. By evaluating the average accuracy of the detection effect of CCTSDB traffic sign data set, mAP reaches 97.68 %, which is 3.78 % higher than YOLOv4. Moreover, network convergence experiments have shown that the improved model converges faster. Compared with other models, the improved model has better detection performance for smaller traffic signs and can better meet the actual needs of high-precision detection.
(2023) Published by SPIE. Downloading of the abstract is permitted for personal use only.
Jie Cao, Peng-hui Li, Hong Zhang, and Zong-li Liu "Research on traffic sign detection based on improved YOLOv4", Proc. SPIE 12800, Sixth International Conference on Computer Information Science and Application Technology (CISAT 2023), 128002H (11 October 2023); https://doi.org/10.1117/12.3003843
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Object detection

Education and training

Head

Deep learning

Feature extraction

Algorithm development

Performance modeling

Back to Top