• Laser & Optoelectronics Progress
  • Vol. 58, Issue 4, 0415009 (2021)
Yuxin Li, Fan Yang*, Zhao Liu, and Yazhong Si
Author Affiliations
  • School of Electronic and Information Engineering, Hebei University of Technology, Tianjin 300401, China
  • show less
    DOI: 10.3788/LOP202158.0415009 Cite this Article Set citation alerts
    Yuxin Li, Fan Yang, Zhao Liu, Yazhong Si. Classification Method of Crossing Vehicle Based on Improved Residual Network[J]. Laser & Optoelectronics Progress, 2021, 58(4): 0415009 Copy Citation Text show less

    Abstract

    To improve the feature extraction capability and recognition capability of models for vehicle images in crossing environments, a vehicle classification method based on an improved residual network is proposed. First, the residual network is used as the basic model, the position of the activation function on the residual block is improved, and the normal convolution in the residual block is replaced with a group convolution. An attention mechanism is then added in the residual block. Finally, the focal loss function replaces the cross-entropy loss function. In the experiment, the Stanford Cars public dataset is used for pretraining and a self-built crossing vehicle dataset is used for migration learning. The results show that the classification accuracy of the proposed model is better than several classical deep learning models in both datasets.
    Yuxin Li, Fan Yang, Zhao Liu, Yazhong Si. Classification Method of Crossing Vehicle Based on Improved Residual Network[J]. Laser & Optoelectronics Progress, 2021, 58(4): 0415009
    Download Citation