-
题名基于坐标卷积和最优传输分配的交通标志检测模型
- 1
-
-
作者
熊昌镇
李熙宇
王庞伟
-
机构
北方工业大学电气与控制工程学院
-
出处
《计算机工程与科学》
北大核心
2025年第6期1071-1078,共8页
-
基金
北京市自然科学基金(4212034)。
-
文摘
针对因尺度小、高速移动和天气等因素导致交通标志检测失败的问题,提出了一种基于坐标卷积和最优传输标签分配的交通标志检测模型。为兼顾嵌入式平台部署和检测速度,以YOLOv5s检测模型为基础。首先,使用具有额外坐标通道的坐标卷积来感知空间信息,增强小尺度交通标志的特征表征能力。其次,通过最优传输分配方法寻找全局最优的目标标签分配方法,减少模糊框的个数,提高训练数据的利用率。最后,使用包含角度损失的SIo U损失函数,提升预测框的收敛速度并增强检测能力。在CCTSDB和TSRD 2个交通标志数据集上测试本文模型的表现,实验结果表明,本文模型的表现较原YOLOv5s模型有较大的提升。与YOLOv7模型相比,本文模型的m AP_0.5和m AP_0.5:0.95在TSRD数据集上分别提升了2.35%和1.45%,在CCTSDB数据集上与YOLOv7的持平,同时本文模型在2个数据集上的检测速度是YOLOv7的2.5倍多,因此本文模型具有更出色的检测精度和速度。
-
关键词
交通标志
坐标卷积
目标检测
最优传输分配
-
Keywords
traffic sign
coordinate convolution
object detection
optimal transport assignment
-
分类号
TP391.4
[自动化与计算机技术—计算机应用技术]
-
