-
题名改进YOLOv8算法的交通标志小目标检测
- 1
-
-
作者
赵会鹏
曹景胜
潘迪敬
李刚
-
机构
辽宁工业大学汽车与交通工程学院
-
出处
《现代电子技术》
北大核心
2024年第20期141-147,共7页
-
基金
国家自然科学基金项目(51675257)
国家自然科学基金青年基金项目(51305190)
+1 种基金
辽宁省教育厅基本科研项目(面上项目)(LJKMZ20220976)
辽宁省自然科学基金指导计划项目(20180550020)。
-
文摘
针对传统网络模型在交通标志检测方面存在的小目标识别不准确的问题,提出一种改进的Ghost-YOLOv8交通标志检测模型。首先,使用GhostConv代替了全部Conv,并设计全新的GhostC2f模块来替代全部的C2f,使模型轻量化;其次,将上采样算法替换为CARAFE,以更好地保留图像的细节信息;然后,在Neck部分引入了GAM注意力机制模块,以增强特征中的语义信息和位置信息;最后,为了解决检测小目标时尺度不一致导致的语义信息丢失问题,添加了小目标检测层,以增强深层和浅层语义信息的融合。实验结果证明,该模型在中国交通标志检测数据集TT100K中的召回率、mAP@0.5、mAP@0.5:0.95指标分别提高了6.8%、4.0%、3.6%,并且模型的参数量及模型大小分别降低了1.069×10^(6)、1.9 MB。综合说明,所提出的模型在精准度不变的前提下,减少了模型的参数量及大小,并能更好地检测到原模型检测不到的小目标;同时,比对比算法具有更好的性能表现,并且适用于边缘计算设备,具有实际应用价值。
-
关键词
YOLOv8
交通标志
小目标检测
GhostNet
CARAFE
ghostc2f
GAM注意力机制
-
Keywords
YOLOv8
traffic sign
small target detection
GhostNet
CARAFE
ghostc2f
GAM attention mechanism
-
分类号
TN911.23-34
[电子电信—通信与信息系统]
TP3
[自动化与计算机技术—计算机科学与技术]
-