Detection and Classification of Small Traffic Signs Based on Cascade Network
-
Graphical Abstract
-
Abstract
Research on the traffic sign detection is significant for driverless technology, which provides useful navigation information. Existing object detection methods are only applicable to large-size objects or small-scale specific types of traffic signs, and the performance of detecting traffic signs in street views is not adequate. In this regard, we propose a method to detect and classify small traffic signs by constructing a cascaded network. Specifically, the RetinaNet network is adopted firstly to integrate multi-layer information to identify small traffic signs in traffic scene images. The focal loss function is used to balance the biased distribution of traffic sign categories. Then, a two-class network is cascaded after the RetinaNet, which helps identify valid traffic signs from the first-stage prediction results. Experiments show that our cascaded network structure could achieve the balance of different categories of predictions and an improvement in precision and recall.
-
-