Abstract

Vehicle classification is widely used in intelligent transportation and smart cities. However, the good performance of vehicle classification often depends on a large number of labeling data, which leads to the high cost of manual labeling. How to use a small number of labeled samples with rich features to achieve good classification performance is a challenge in vehicle classification. To solve these problems, we propose a deep active learning framework called Feature Fusion Spatial Pyramid Pooling and Re-parameterization Visual Geometry Group (FFSPP-RepVGG). The framework is mainly divided into two parts: query strategy and feature extraction module. Specifically, the query strategy uses Feature Fusion (FF) to calculate the loss, thus defining uncertainty and being able to select more valuable samples for labeling training. The feature extraction module uses Spatial Pyramid Pooling and Re-parameterization Visual Geometry Group (SPP-RepVGG) model, which can extract more reliable image features. It can reduce the cost of data labeling and have better performance at the same time. The experimental results on the BIT-Vehicle data set and the car-10 data set show that the FFSPP-RepVGG framework has superior performance than that of the comparison models.

Talk to us

Join us for a 30 min session where you can share your feedback and ask us any queries you have

Schedule a call

Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.