A vehicle classification model based on deep active learning.

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.