Improved YOLOv5 network for real-time multi-scale traffic sign detection

by   Junfan Wang, et al.

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 of the targets changes greatly, which will have a certain impact on the detection accuracy. Feature pyramid is widely used to solve this problem but it might break the feature consistency across different scales of traffic signs. Moreover, in practical application, it is difficult for common methods to improve the detection accuracy of multi-scale traffic signs while ensuring real-time detection. In this paper, we propose an improved feature pyramid model, named AF-FPN, which utilizes the adaptive attention module (AAM) and feature enhancement module (FEM) to reduce the information loss in the process of feature map generation and enhance the representation ability of the feature pyramid. We replaced the original feature pyramid network in YOLOv5 with AF-FPN, which improves the detection performance for multi-scale targets of the YOLOv5 network under the premise of ensuring real-time detection. Furthermore, a new automatic learning data augmentation method is proposed to enrich the dataset and improve the robustness of the model to make it more suitable for practical scenarios. Extensive experimental results on the Tsinghua-Tencent 100K (TT100K) dataset demonstrate the effectiveness and superiority of the proposed method when compared with several state-of-the-art methods.



There are no comments yet.


page 1

page 3

page 4

page 5

page 7

page 9


Localized Traffic Sign Detection with Multi-scale Deconvolution Networks

Autonomous driving is becoming a future practical lifestyle greatly driv...

BAF-Detector: An Efficient CNN-Based Detector for Photovoltaic Solar Cell Defect Detection

The multi-scale defect detection for solar cell electroluminescence (EL)...

Deep Dual Pyramid Network for Barcode Segmentation using Barcode-30k Database

Digital signs(such as barcode or QR code) are widely used in our daily l...

Research on Mosaic Image Data Enhancement for Overlapping Ship Targets

The problem of overlapping occlusion in target recognition has been a di...

The Research of the Real-time Detection and Recognition of Targets in Streetscape Videos

This study proposes a method for the real-time detection and recognition...

CenterAtt: Fast 2-stage Center Attention Network

In this technical report, we introduce the methods of HIKVISION_LiDAR_De...

Objects Detection from Digitized Herbarium Specimen based on Improved YOLO V3

Automatic measurement of functional trait data from digitized herbarium ...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.