用轻量级孪生网络提升复杂路况下交通标志识别准确率
Road Traffic Sign Recognition method using Siamese network Combining Efficient-CNN based Encoder
- 采用高效卷积编码器+孪生结构,通过对比损失增强抗模糊与遮挡能力
- 在清华-腾讯100K和德国交通标志数据集上达到86.1%准确率,模型仅2.9M
- 推理时可关闭模板分支,处理速度达每帧0.1秒,比现有方法快1.5倍
交通标志识别(TSR)在智能驾驶与交通系统中至关重要。复杂环境中的运动模糊和遮挡问题严重影响实时高精度识别。本文提出IECES网络,包含三阶段:基于Efficient-CNN的编码器、孪生主干网络与全连接层。首先利用卷积编码器提取增强样本与标准图像的特征;随后设计基于Efficient-CNN的孪生网络,结合对比损失函数,通过计算输入与模板间的距离提升对运动模糊和遮挡样本的鲁棒性;训练后可停止模板分支以加速推理,降低计算资源消耗。最后通过特征码与SoftMax分类器完成类别识别。在清华-腾讯100K与德国交通标志识别基准数据集上的实验表明,该方法在模糊与遮挡场景下平均精度、召回率与准确率分别为88.1%、86.43%与86.1%,模型规模仅2.9M,单帧处理时间0.1秒,较现有方法提速1.5倍。
原文摘要 · Abstract (English)
Traffic signs recognition (TSR) plays an essential role in assistant driving and intelligent transportation system. However, the noise of complex environment may lead to motion-blur or occlusion problems, which raise the tough challenge to real-time recognition with high accuracy and robust. In this article, we propose IECES-network which with improved encoders and Siamese net. The three-stage approach of our method includes Efficient-CNN based encoders, Siamese backbone and the fully-connected layers. We firstly use convolutional encoders to extract and encode the traffic sign features of augmented training samples and standard images. Then, we design the Siamese neural network with Efficient-CNN based encoder and contrastive loss function, which can be trained to improve the robustness of TSR problem when facing the samples of motion-blur and occlusion by computing the distance between inputs and templates. Additionally, the template branch of the proposed network can be stopped when executing the recognition tasks after training to raise the process speed of our real-time model, and alleviate the computational resource and parameter scale. Finally, we recombined the feature code and a fully-connected layer with SoftMax function to classify the codes of samples and recognize the category of traffic signs. The results of experiments on the Tsinghua-Tencent 100K dataset and the German Traffic Sign Recognition Benchmark dataset demonstrate the performance of the proposed IECESnetwork. Compared with other state-of-the-art methods, in the case of motion-blur and occluded environment, the proposed method achieves competitive performance precision-recall and accuracy metric average is 88.1%, 86.43% and 86.1% with a 2.9M lightweight scale, respectively. Moreover, processing time of our model is 0.1s per frame, of which the speed is increased by 1.5 times compared with existing methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。