Laser Journal, Volume. 45, Issue 1, 166(2024)
Improved road vehicle classification network based on RepVGG-A0
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.
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
Category:
Received: Jun. 9, 2023
Accepted: --
Published Online: Aug. 6, 2024
The Author Email: Chenghan REN (1214939728@qq.com)