Computer Engineering and Applications ›› 2023, Vol. 59 ›› Issue (11): 105-111.DOI: 10.3778/j.issn.1002-8331.2201-0408

• Pattern Recognition and Artificial Intelligence • Previous Articles     Next Articles

Traffic Sign Recognition Algorithm Based on Siamese Neural Network with Encoder

LYU Binglue, XI Zhenghao, SHAO Yuchao   

  1. School of Electric and Electrical Engineering, Shanghai University of Engineering Science, Shanghai 201620, China
  • Online:2023-06-01 Published:2023-06-01

孪生神经网络交通标志编码识别算法

吕秉略,奚峥皓,邵宇超   

  1. 上海工程技术大学 电子电气工程学院,上海 201620

Abstract: Traffic sign recognition has been applied to the assistant driving system. However, some factors, such as occlusion, contamination damage and weather can seriously affect the accuracy and robustness of traffic sign recognition function. To solve this problem, a traffic sign encoding and classification method based on the Siamese neural network is developed. The method treats the traffic sign recognition problem as a convolutional feature code recognition problem. Firstly, the method uses convolutional neural network to extract and encode features of training samples and reference samples. Secondly, the method uses Siamese neural network to compare the feature code of training samples and reference samples and trains the encoder with contrastive loss. With the help of a fully connected layer, the method can recombine and classify the convolutional feature code of input in the end. The experimental results show that this method can produce effective and robust feature code of traffic sign under motion blur and occlusion conditions. Compared to other advanced methods, this method has higher accuracy.

Key words: machine vision, traffic sign, contrastive learning, siamese neural network, feature-based similarity measure

摘要: 交通标志识别技术正在被逐步应用到汽车辅助驾驶领域。但是,遮挡、污损、天气环境变化等因素会严重影响交通标志识别的准确性和稳定性。针对该问题,提出了一种基于孪生神经网络的交通标志编码识别模型。该模型将交通标志的识别问题视为交通标志的卷积特征编码识别问题。通过卷积神经网络对交通标志训练样本和基准样本进行特征提取与编码。再利用孪生神经网络进行编码对比,结合对比损失函数对编码器训练调整。通过全连接层对输入通路的标志卷积编码进行重新组合与分类,从而实现交通标识的识别。实验结果表明,所提的基于改进孪生神经网络的编码器模型对存在运动模糊与遮挡的标志图像能生成有效、鲁棒的特征编码,相较于其他先进算法,具有更高的识别准确率。

关键词: 机器视觉, 交通标志, 对比学习, 孪生神经网络, 特征相似性度量