Education, Science, Technology, Innovation and Life
Open Access
Sign In

Research on CEL-YOLO Algorithm for Lightweight Detection of Traffic Signs

Download as PDF

DOI: 10.23977/autml.2025.060108 | Downloads: 18 | Views: 538

Author(s)

Fuzhaojin Huang 1, Zijian Dong 1, Lingsheng Liang 1, Xinrui Chen 1

Affiliation(s)

1 School of Electronic Engineering, Jiangsu Ocean University, Lianyungang, Jiangsu, 222000, China

Corresponding Author

Zijian Dong

ABSTRACT

Nowadays, in the era of rapid development in the field of intelligent transportation and large information reserves, in order to better adapt to embedded devices and improve the real-time and robustness of intelligent vehicle perception, this paper proposes a traffic sign small target detection algorithm based on YOLOv8 model. First, on the basis of C2f-STAR module, StarNet convolution StarsBlock is added to build C2f-Starsblock, which replaces C2f module in Backbone part of YOLOv8 network to improve the feature representation capability and detection performance of the model. Secondly, based on the BottleNeck of the residual module Faster_Block in FasterNet, the C2f module in YOLOv8 network is replaced, and EMA attention mechanism is added to the C2F-FASTER module to construct the C2f-Faster-EMA module. Improve the ability of C2f module to capture multi-scale feature information; Thirdly, the SPPF module is combined with the large separable Kernel Attention (LSKA) module to construct the SPPF-LSKA module to enhance the feature extraction capability of the model. Finally, a Light Weight Shared Convolutional Detection (LSCD) is added. It can be seen in the CCTSDB2021 traffic sign dataset, finally the improvement of the traffic sign this kind of small target detection accuracy and robustness of the model. To verify the effectiveness of CEL-YOLO, mAP-50 achieves 97.1% in traffic sign detection tasks. When the total number of parameters and calculation amount are reduced by 54.5% and 44.5% respectively, the accuracy remains the same as the original model. The verification results show that compared with the benchmark model, the model is significantly lighter in volume and computation, and is more suitable for small target detection.

KEYWORDS

Traffic sign detection, Lightweight improvement, LSCD, YOLOv8n, Detection accuracy, TT100K

CITE THIS PAPER

Fuzhaojin Huang, Zijian Dong, Lingsheng Liang, Xinrui Chen, Research on CEL-YOLO Algorithm for Lightweight Detection of Traffic Signs. Automation and Machine Learning (2025) Vol. 6: 67-77. DOI: http://dx.doi.org/10.23977/autml.2025.060108.

REFERENCES

[1] Benallal M, Meunier J.Real-time color segmentation of road signs[C].Canadian Conference on Electrical and Computer Engineering. Toward a Caring and Humane Technology.IEEE,2003, 3:1823-1826.
[2] Chakraborty S, Deb K. Bangladeshi road sign detection based on YCbCr color model and DtBs vectorf[C].2015 International Conference on Computer and Information Engineering (ICCIE).IEEE, 2015:158-161.
[3] Wang G, Ren G, Jiang L, et al. Hole-based traffic sign detection method for traffic signs with red rim[J]. The Visual Computer, 2014, 30:539-551.
[4] Redmon J, Divvala S, Girshick R, et al. You only look once: unified, real-time object detection[C]. IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, 2016: 779-788.
[5] Hong Yan, Wang Lei, Su Jingming, et al. Detection of foreign matter in coal mine conveyor belt based on improved YOLOv8 [J]. Industrial and Mine Automation, 2024, 50(06):61-69.
[6] Ye Hanyu, Guo Laide, Li Yuexian, et al. Design of intelligent household waste classification system based on improved YOLOv5 algorithm [J/OL]. Environmental Engineering, 2024, 8(30): 1-12.
[7] Zheng Wenxuan, Yang Ying. Target detection method of ripe fragrant pear based on frequency domain data enhancement and lightweight YOLO v7 model [J]. Transactions of the Chinese Society for Agricultural Machinery, 2019, 55(05):244-253.
[8] Wang Hai, Wang Kuan, CAI Yingfeng, et al. Traffic sign recognition based on improved cascaded convolutional neural networks [J]. Automotive Engineering, 2019, 42(09):256-1262+1269.
[9] WEI Tiancheng, Chen Xiaofeng, Yin Yuanliang. Research on Road traffic sign recognition based on multi-scale Convolutional neural network [J]. Journal of Northwestern Polytechnical University, 2021.39(04):891-900.
[10] Cheng P, Liu W, et al. LOCO: local context based faster R-CNN for small traffic sign Detection[C]. MultiMedia Modeling: 24th International Conference, 2018:329-341.
[11] Daliang Ouyang, Su He, Guozhong Zhang, Mingzhu Luo.Efficient Multi-Scale Attention Module with Cross-Spatial Learning[J],Computer Vision and Pattern Recognition, 2023, 133: 108667.
[12] Liu Y, Sun P, Wergeles N, et al. A survey and performance evaluation of deep learning methods for small object detection[J]. Expert Systems with Applications, 2021, 172: 114602. 
[13] ZengJ, ZhongH.YOLOv8-PD: an improved road damage detection algorithm based on YOLOv8n model[I.Scientific reports, 2024,14(1): 12052.
[14] Daquan Zhou, Qibin Hou, Yunpeng Chen, Jiashi Feng, and Shuicheng Yan. Rethinking bottleneck structure for efficient mobile network design. In ECCV, 2020.06.
[15] GUO M H,LU C Z,LIU Z N, et al. Visual attention network[J].Computational Visual Media, 2023,9(4):733-752. 
[16] Ma X, Guo FM, Niu W, Lin X, Tang J, Ma K, Ren B, Wang Y. Pconv: The missing but desirable sparsity in dnn weight pruning for real-time execution on mobile devices. InProceedings of the AAAI conference on artificial intelligence 2020 Apr 3 (Vol. 34, No. 04, pp. 5117-5124). 
[17] Wu Y, He K. Group normalization[C]//Proceedings of the European conference on computer vision (ECCV). 2018: 3-19.
[18] Tian Z, Shen C, Chen H, et al. FCOS: A simple and strong anchor-free object detector[J]. IEEE transactions on pattern analysis and machine intelligence, 2020, 44(4): 1922-1933. 
[19] Zhao Huipeng, Cao Jingsheng, Pan Dijing et al. Traffic Sign small target detection with improved YOLOv8 algorithm [J]. Modern Electronic Technique. 2024, 47 (20).

Downloads: 3722
Visits: 166907

Sponsors, Associates, and Links


All published work is licensed under a Creative Commons Attribution 4.0 International License.

Copyright © 2016 - 2031 Clausius Scientific Press Inc. All Rights Reserved.