Pyramid Transformer for Traffic Sign Detection
Traffic sign detection is a vital task in the visual system of self-driving cars and the automated driving system. Recently, novel Transformer-based models have achieved encouraging results for various computer vision tasks. We still observed that vanilla ViT could not yield satisfactory results in traffic sign detection because the overall size of the datasets is very small and the class distribution of traffic signs is extremely unbalanced. To overcome this problem, a novel Pyramid Transformer with locality mechanisms is proposed in this paper. Specifically, Pyramid Transformer has several spatial pyramid reduction layers to shrink and embed the input image into tokens with rich multi-scale context by using atrous convolutions. Moreover, it inherits an intrinsic scale invariance inductive bias and is able to learn local feature representation for objects at various scales, thereby enhancing the network robustness against the size discrepancy of traffic signs. The experiments are conducted on the German Traffic Sign Detection Benchmark (GTSDB). The results demonstrate the superiority of the proposed model in the traffic sign detection tasks. More specifically, Pyramid Transformer achieves 77.8% mAP on GTSDB when applied to the Cascade RCNN as the backbone, which surpasses most well-known and widely-used state-of-the-art models.
Code (0)
등록된 구현이 없습니다.
Tasks
Inductive BiasSelf-Driving CarsTraffic Sign DetectionMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Improved YOLOv5 network for real-time multi-scale traffic sign detection
Traffic sign detection is a challenging task for the unmanned driving system, especially for the detection of multi-scale targets and the real-time problem of detection. In the traffic sign detection process, the scale o…
Data AugmentationTraffic Sign DetectionExtended Feature Pyramid Network for Small Object Detection
Small object detection remains an unsolved challenge because it is hard to extract information of small objects with only a few pixels. While scale-level corresponding detection in feature pyramid network alleviates this…
Objectobject-detectionObject DetectionSmall Object DetectionPeri-midFormer: Periodic Pyramid Transformer for Time Series Analysis
Time series analysis finds wide applications in fields such as weather forecasting, anomaly detection, and behavior recognition. Previous methods attempted to model temporal variations directly using 1D time series. Howe…
Anomaly DetectionClassificationImputationTime Series+3RePFormer: Refinement Pyramid Transformer for Robust Facial Landmark Detection
This paper presents a Refinement Pyramid Transformer (RePFormer) for robust facial landmark detection. Most facial landmark detectors focus on learning representative image features. However, these CNN-based feature repr…
Facial Landmark DetectionExplainable Graph Pyramid Autoformer for Long-Term Traffic Forecasting
Accurate traffic forecasting is vital to an intelligent transportation system. Although many deep learning models have achieved state-of-art performance for short-term traffic forecasting of up to 1 hour, long-term traff…
Graph Neural NetworkTemporal Sequences