-
题名基于改进YOLOv5的复杂场景下交通标志识别方法
- 1
-
-
作者
胡瑛
刘狄昆
刘拯
李铸成
乔汇东
-
机构
湖南工程学院信息科学与工程学院
-
出处
《湖南工程学院学报(自然科学版)》
2024年第2期31-38,共8页
-
基金
国家级大学生创新创业训练计划项目(202211342017)
湖南省教育厅优秀青年科研项目(22B0735)
+1 种基金
湖南省教育厅重点科研项目(20A112)
湖南省自科基金面上项目(2022JJ30197)。
-
文摘
交通标志识别是智能驾驶不可缺少的重要环节,关系着人们进行智能驾驶的安全问题,本文以复杂环境下的交通标志为研究对象,针对目前交通标志识别难以兼顾实时性和准确率的问题,提出一种改进的YOLOv5交通标志识别算法.首先,对数据集做预处理与数据增强,加强对目标的检测能力;其次,引用PP-LCNet轻量型网络,减少主干网络参数量,实现模型轻量化;最后,在颈部网络融合注意力机制,以增强特征提取能力.实验结果表明,相较于原YOLOv5s模型,本文算法的模型参数量减少了25.9%,检测速度提高了50.08帧/s,平均精度达到97.58%,易于部署且能达到智能驾驶场景中对交通标志识别的实时性和准确率要求.
-
关键词
交通标志识别
数据增强
YOLOv5
PP-LCNet网络
注意力机制
-
Keywords
traffic signs recognition
data enhancement
YOLOv5
PP-LCNet
attention mechanism
-
分类号
TP391.4
[自动化与计算机技术—计算机应用技术]
-