A Semi-Supervised Learning Framework Combining CNN and Multi-scale Transformer for Traffic Sign Detection and Recognition

Siyun Chen,Zhenxin Zhang,Liqiang Zhang, Rixing He,Zhen Li, Mengbing Xu, Hao Ma

IEEE Internet of Things Journal(2024)

引用 0|浏览0
暂无评分
摘要
The accurate extraction of traffic signs is of great significance to the digitization of traffic information and the fine management of traffic. The paper introduces an innovative approach to address the challenges associated with recognizing and detecting traffic signs, considering their vulnerability to complex backgrounds, variations in illumination, and motion blur. The proposed method utilizes a semi-supervised learning strategy, combining Convolutional Neural Networks (CNNs) with a transformer encoder-decoder architecture, to extract traffic sign features from vehicle panoramic images. To enhance feature extraction, a hierarchical sampling method (HSM) is introduced, which facilitates the extraction of multi-scale self-attention features in the transformer encoder-decoder structure. Additionally, a network module called Local and Global Information Aggregator (LGIA) is designed based on HSM, enabling the incorporation of both local and global context information. Furthermore, a Semi-Supervised Learning (SSL) strategy is adopted to simultaneously train our model using both labeled and unlabeled data samples. This strategy aims to improve the extraction of traffic signs by capitalizing on the broader dataset available through unlabeled data. Experimental results demonstrate the effectiveness and robustness of the proposed method in improving the detection and recognition of traffic signs. The approach showcases significant improvements in overcoming the challenges posed by complex backgrounds, variations in illumination, and motion blur. Our approach achieved a 0.9% improvement in the F1-score evaluation over the current classical object detection algorithm on the public dataset Tsinghua-Tencent 100K and a 1.1% improvement on the SSW dataset.
更多
查看译文
关键词
Traffic sign recognition and detection,vehicle panoramic image,semi-supervised learning,Transformer,CNN
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要