基于改进RT-DETR的有遮挡交通标志检测算法

Blocked traffic sign detection algorithm based on improved RT-DETR

  • 摘要: 针对交通标志检测中目标尺寸小、检测精度低等问题,尤其是在远距离拍摄、遮挡严重的情况下,传统检测算法往往难以准确识别交通标志。本文提出了一种基于改进RT-DETR的交通标志检测算法。首先。考虑到当前交通标志被遮挡情况下数据集的匮乏,本文自建了一个遮挡条件下的交通标志数据集。然后,在反向残差移动块中引入膨胀重参数块,构建了一个轻量级的复合膨胀残差块来替换原始主干提取网络中的BasicBlock,增强了模型的特征提取能力。最后,对RT-DETR模型的损失函数进行了优化,提出了DS-IoU联合损失函数加快收模型敛速度。实验结果表明,改进后的算法在自制数据集上的mAP为94.2%,相比于原始算法分别提升了4.7%,在公开数据集TT100K和CCTSDB2021的mAP分别为92.8%和91.7%,相比于原始算法分别提升了3.1%和2.4%,Params和FLOPs相比于原始的算法分别降低了26.0%和12.5%。证明本文提出的改进方法极大减少了计算量和参数数量,有效提升了遮挡情况下的交通标志的检测精度。

     

    Abstract: Aiming at the problems of small target size and low detection accuracy in traffic sign detection, especially in the case of long-distance shooting and serious occlusion, traditional detection algorithms are often difficult to accurately identify traffic signs. In this paper, a traffic sign detection algorithm based on improved RT-DETR is proposed. First. Considering the scarcity of current datasets in the case of occluded traffic signs, this paper builds a self-constructed dataset of traffic signs under occluded conditions. Then, a lightweight composite inflated residual block is constructed to replace the BasicBlock in the original backbone extraction network by introducing an inflated reparameter block in the inverse residual shifted block, which enhances the feature extraction capability of the model. Finally, the loss function of the RT-DETR model is optimized, and the Inner-MPDIoU joint loss function is proposed to accelerate the convergence speed of the model. The experimental results show that the improved algorithm has a mAP of 94.2% on the homemade dataset, which is improved by 4.7% compared to the original algorithm, respectively, and the mAPs on the publicly available datasets, TT100K and CCTSDB2021, are 92.8% and 91.7%, which are improved by 3.1% and 2.4% compared to the original algorithm, and the Params and FLOPs are improved by 26.0% compared to the original algorithm by 26.0% and 12.5%, respectively. It is proved that the improved method proposed in this paper greatly reduces the amount of computation and the number of parameters, and effectively improves the detection accuracy of traffic signs under the occlusion situation

     

/

返回文章
返回