基于改进RT-DETR的公路路面交通标识检测算法
DOI:
CSTR:
作者:
作者单位:

1.内蒙古工业大学信息工程学院 呼和浩特 010080;2.内蒙古灵奕高科技(集团)有限责任公司 呼和浩特 010010

作者简介:

通讯作者:

中图分类号:

TP391.4;TN919.8

基金项目:


Highway traffic sign detection algorithm based on improved RT-DETR
Author:
Affiliation:

1.School of Information Engineering, Inner Mongolia University of Technology, Hohhot 010080, China;2.Inner Mongolia Lingyi High Tech (Group) Co., Ltd.,Hohhot 010010, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为了降低复杂天气、光照变化和交通标识污损等环境因素干扰和算法自身复杂度高导致模型性能的下降,提出一种以RT-DETR为基准模型的公路路面交通标识检测算法。首先,使用设计的轻量化模块CSP-PMSFA作为算法的主干网络,降低模型的计算量和参数量,提高模型的表达能力。然后,针对原有算法尺度内特征交互模块计算量大、模型容量小且计算效率有限等问题,引入了级联分组注意力CGA进行改进。最后,设计了跨尺度特征融合模块EMBSFPN,用于解决感受野适应性不足,特征信息处理有限的问题;采用EUCB上采样模块的解码机制在保证准确度的情况下保留和融合特征信息,优化性能的同时提高模型的鲁棒性。实验结果表明,改进算法在ROAD MARK路面标识检测数据集上与原有算法相比mAP50提高了2.6%、FPS提高了13.6帧,计算量和参数量分别下降了29.7%和39.6%,整体优于其他改进算法,在进行轻量化的同时提高了检测精度和速度,具有实用性。

    Abstract:

    In order to reduce the interference of environmental factors such as complex weather, lighting changes, and traffic sign fouling, as well as the degradation of model performance caused by the high complexity of the algorithm itself, a highway road traffic sign detection algorithm based on RT-DETR as the benchmark model is proposed. Firstly, the designed lightweight module CSP-PMSFA is used as the backbone network of the algorithm to reduce the computational and parameter complexity of the model and improve its expressive power. Then, to address the issues of high computational complexity, small model capacity, and limited computational efficiency in the feature interaction module within the original algorithm scale, cascaded group attention CGA was introduced for improvement. Finally, a cross scale feature fusion module EMBSFPN was designed to address the issues of insufficient receptive field adaptability and limited feature information processing; the decoding mechanism using EUCB upsampling module preserves and fuses feature information while ensuring accuracy, optimizing performance and improving the robustness of the model. The experimental results show that the improved algorithm has improved mAP50 by 2.6% and FPS by 13.6 frames compared to the original algorithm on the ROAD MARK road marking detection dataset. The computational and parameter requirements have decreased by 29.7% and 39.6%, respectively. Overall, it outperforms other improved algorithms and improves detection accuracy and speed while being lightweight, demonstrating practicality.

    参考文献
    相似文献
    引证文献
引用本文

孙海青,杨传颖,敖乐根.基于改进RT-DETR的公路路面交通标识检测算法[J].电子测量技术,2025,48(8):187-195

复制
相关视频

分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2025-05-23
  • 出版日期:
文章二维码