基于RT-DETR的轻量化交通标志检测

Lightweight traffic sign detection based on RT-DETR

  • 摘要: 交通标志检测是智能车辆行驶过程中的关键,对于车辆分析路况有着重大的意义。针对现存交通标志检测算法参数较多、精确率不高的问题,提出一种基于改进RT-DETR(real-time detection transformer)的轻量化交通标志检测算法。首先,将模型主干网络中的ResNet网络替换为VanillaNet网络,减少网络的层数与参数量;其次,使用加权双向特征金字塔模型(bi-directional feature pyramid network, BiFPN)代替RT-DETR特征融合模块中的PAN(path aggregation network)结构,增加模型的融合能力,提取更加丰富的特征信息;最后,在特征融合模块中添加GAM(global attention mechanism)注意力机制,加强模型对全局信息的感知,提升多目标与被遮挡目标的检测性能。该算法在交通标志数据集上进行测试,结果说明改进后的RT-DETR算法的mAP(mean average precision)值达到了87.7%,相比原始算法提高了3.6%,且参数减少21%,满足智能车辆设备的部署需要,证明了改进算法的有效性。

     

    Abstract: Traffic sign detection is the key in the driving process of intelligent vehicles, which is of great significance to vehicle analysis of road conditions. Aiming at the problems of too many parameters and low accuracy of existing traffic sign detection algorithms, a lightweight traffic sign detection algorithm based on improved real-time detection transformer (RT-DETR) was proposed. Firstly, the ResNet network in the backbone network of the model was replaced by VanillaNet network to reduce the number of layers and parameters. Secondly, the bi-directional feature pyramid network (BiFPN) was used to replace the path aggregation network (PAN) structure in the feature fusion module of RT-DETR to increase the fusion ability of the model and extract more abundant feature information. Finally, the global attention mechanism (GAM) was added to the feature fusion module to enhance the model perception of global information and improve the detection performance of multiple targets and obscured targets. The proposed algorithm was tested on traffic sign datasets. The mean average precision (mAP) value of the improved RT-DETR algorithm reaches 87.7%, 3.6% higher than that of the original algorithm, and the parameters are reduced by 21%, meeting the deployment needs of intelligent vehicle equipment, which proves the effectiveness of the improved algorithm.

     

/

返回文章
返回