• Laser Journal
  • Vol. 45, Issue 1, 166 (2024)
REN Chenghan* and HUANG Jun
Author Affiliations
  • [in Chinese]
  • show less
    DOI: 10.14016/j.cnki.jgzz.2024.1.166 Cite this Article
    REN Chenghan, HUANG Jun. Improved road vehicle classification network based on RepVGG-A0[J]. Laser Journal, 2024, 45(1): 166 Copy Citation Text show less

    Abstract

    Aiming at the problem that it is difficult to balance the detection accuracy and real-time in the current vehicle type recognition process, an improved road vehicle type recognition network based on RepVGG-A0 is proposed,which uses the idea of structural re-parameterization to fuse the multi-branch network to improve the network reasoning speed. The mixed void convolution is used to replace the traditional convolution, which strengthens the recognition ability of the model for large targets. Integrating the residual structure coordinate attention (RES-CA) module into the network backbone improves the network's ability to extract effective feature information, and avoids the impact of gradient disappearance and gradient degradation. In addition, the label smoothing regularization method is used to improve the loss function, reduce the impact of model overfitting on the detection results, and improve the generalization of the model. After verification, the recognition accuracy of the method in this paper on the road vehicle data set BIT-Vehicle has reached 97. 17%, which is 2. 67% higher than the original model, and is superior to the existing ResNet-18, VGG and other network models, while ensuring the detection speed of the model.