Laser Journal, Volume. 45, Issue 1, 166(2024)

Improved road vehicle classification network based on RepVGG-A0

REN Chenghan* and HUANG Jun
Author Affiliations
  • [in Chinese]
  • show less

    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.

    Tools

    Get Citation

    Copy Citation Text

    REN Chenghan, HUANG Jun. Improved road vehicle classification network based on RepVGG-A0[J]. Laser Journal, 2024, 45(1): 166

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Jun. 9, 2023

    Accepted: --

    Published Online: Aug. 6, 2024

    The Author Email: Chenghan REN (1214939728@qq.com)

    DOI:10.14016/j.cnki.jgzz.2024.1.166

    Topics