基于改进Yolov8的交通标志检测算法
Improving Yolov8 for Traffic Sign Detection Algorithm
摘要: 针对在自动驾驶行驶中交通标志背景复杂和检测目标小等问题,设计了一种基于yolov8改进的交通标志识别算法。首先在yolov8的DarkNet53中使用动态蛇形卷积,通过引入动态蛇形卷积自适应调整卷积核形状和大小,可以更好地适应小目标尺寸,从而提高准确性,而且可通过目标形状的和轮廓,局部调整卷积核,使在卷积过程中更加关注目标特征,计算复杂度降低,提高检测精度;其次引入局部敏感卷积(LSKA)对SPPF进行改进,通过金字塔池化和频率金字塔来获得多尺度和多频率的特征表示,再利用LSKA来提取局部位置不变性,让其具有更好的鲁棒性和精度,提高网络对交通标志识别的准确性。实验结果表明,改进后的算法在精确率、召回率、平均精确率上相比原算法分别提升2.8%、2.7%、3.3%,检测速度满足实时性要求。
Abstract: A traffic sign recognition algorithm based on Yolov8 improvement is designed to address issues such as complex backgrounds of traffic signs and small target detection in autonomous driving. Firstly, dynamic snake convolution is introduced into DarkNet53 of Yolov8. By dynamically adjusting the shape and size of convolution kernels, adaptive adjustments are made to better accommodate small target sizes, thereby improving accuracy. Moreover, by adjusting convolution kernels locally based on target shapes and contours, attention to target features during the convolution process is enhanced, reducing computational complexity and improving detection accuracy. Secondly, Local Sensitivity Kernel Aggregation (LSKA) is introduced to improve the Spatial Pyramid Pooling Fusion (SPPF). By utilizing pyramid pooling and frequency pyramids to obtain multi-scale and multi-frequency feature representations, and then using LSKA to extract local position invariance, the algorithm exhibits better robustness and accuracy, enhancing the network’s accuracy in traffic sign recognition. Experimental results show that the improved algorithm achieves a 2.8% increase in precision, a 2.7% increase in recall, and a 3.3% increase in average precision compared to the original algorithm, while meeting the requirements for practical detection speed.
文章引用:苟皓, 唐岚, 孟忠伟. 基于改进Yolov8的交通标志检测算法[J]. 图像与信号处理, 2024, 13(3): 328-337. https://doi.org/10.12677/jisp.2024.133028

参考文献

[1] 陈晗晗, 王俊英, 任肖月. 交通标志的检测与识别方法研究综述[J]. 信息技术与信息化, 2024(3): 77-82.
[2] 陈红, 王相超, 陈志琳. 自然场景下的交通标志检测与识别[J]. 电子测量技术, 2021, 44(12): 102-109.
[3] 赵坤, 刘立, 孟宇, 等. 弱光照条件下交通标志检测与识别[J]. 工程科学学报, 2020, 42(8): 1074-1084.
[4] Hu, P., Lu, J., Cui, Y., Hu, B. and Liang, F. (2023). Insulator Defect Detection Based on Faster R-CNN and Yolov3 Algorithm. 2023 IEEE 4th China International Youth Conference on Electrical Engineering (CIYCEE), Chengdu, 8-10 December 2023, 1-4.[CrossRef
[5] Wang, Y., Pan, F., Li, Z., Xin, X. and Li, W. (2023). CoT-YOLOv8: Improved YOLOv8 for Aerial Images Small Target Detection. 2023 China Automation Congress (CAC), Chongqing, 17-19 November 2023, 4943-4948.[CrossRef
[6] Bochkovskiy, A., Wang, C.Y. and Liao, H.Y.M. (2020) Yolov4: Optimal Speed and Accuracy of Object Detection. arXiv: 2004.10934.
[7] 顾峰, 四兵锋. 基于多源特征增强的交通标志识别方法[J]. 北京交通大学学报, 2023, 47(4): 73-80.
[8] Chu, J., Zhang, C., Yan, M., Zhang, H. and Ge, T. (2023) TRD-YOLO: A Real-Time, High-Performance Small Traffic Sign Detection Algorithm. Sensors, 23, Article 3871. [Google Scholar] [CrossRef] [PubMed]
[9] 石镇岳, 侯婷, 苏勇东. 改进YOLOv7的交通标志检测算法[J]. 计算机系统应用, 2023, 32(10): 157-165.
[10] Gao, X., Chen, L., Wang, K., Xiong, X., Wang, H. and Li, Y. (2022) Improved Traffic Sign Detection Algorithm Based on Faster R-CNN. Applied Sciences, 12, Article 8948. [Google Scholar] [CrossRef
[11] 王琳毅, 白静, 李文静, 等. YOLO系列目标检测算法研究进展[J]. 计算机工程与应用, 2023, 59(14): 15-29.
[12] Qi, Y., He, Y., Qi, X., Zhang, Y. and Yang, G. (2023). Dynamic Snake Convolution Based on Topological Geometric Constraints for Tubular Structure Segmentation. 2023 IEEE/CVF International Conference on Computer Vision (ICCV), Paris, 1-6 October 2023, 6047-6056.[CrossRef
[13] Lau, K.W., Po, L. and Rehman, Y.A.U. (2024) Large Separable Kernel Attention: Rethinking the Large Kernel Attention Design in CNN. Expert Systems with Applications, 236, Article ID: 121352. [Google Scholar] [CrossRef
[14] Zhang, J., Zou, X., Kuang, L.D., et al. (2022) CCTSDB 2021: A More Comprehensive Traffic Sign Detection Benchmark. Human-Centric Computing and Information Sciences, 12, Article 23.
[15] 史涛, 崔杰, 李松. 优化改进YOLOv8实现实时无人机车辆检测的算法[J]. 计算机工程与应用, 2024, 60(9): 79-89.