计算机工程与科学

作者：熊昌镇, 李熙宇, 王庞伟

单位：北方工业大学电气与控制工程学院，北京 100144

关键词：交通标志,坐标卷积,目标检测,最优传输分配,

基金：北京市自然科学基金（4212034）

针对因尺度小、高速移动和天气等因素导致交通标志检测失败的问题，提出了一种基于坐标卷积和最优传输标签分配的交通标志检测模型。为兼顾嵌入式平台部署和检测速度，以YOLOv5s检测模型为基础。首先，使用具有额外坐标通道的坐标卷积来感知空间信息，增强小尺度交通标志的特征表征能力。其次，通过最优传输分配方法寻找全局最优的目标标签分配方法，减少模糊框的个数，提高训练数据的利用率。最后，使用包含角度损失的SIoU损失函数，提升预测框的收敛速度并增强检测能力。在CCTSDB和TSRD 2个交通标志数据集上测试本文模型的表现，实验结果表明，本文模型的表现较原YOLOv5s模型有较大的提升。与YOLOv7模型相比，本文模型的mAP_0.5和mAP_0.5:0.95在TSRD数据集上分别提升了2.35%和1.45%，在CCTSDB数据集上与YOLOv7的持平，同时本文模型在2个数据集上的检测速度是YOLOv7的2.5倍多，因此本文模型具有更出色的检测精度和速度。

来源：2025年第06期

《计算机工程与科学》期刊编辑部

查看计算机工程与科学杂志2025年第06期