Laser & Optoelectronics Progress, Volume. 58, Issue 4, 0415009(2021)

Classification Method of Crossing Vehicle Based on Improved Residual Network

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

    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.

    Tools

    Get Citation

    Copy Citation Text

    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

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

    Category: Machine Vision

    Received: Sep. 2, 2020

    Accepted: Nov. 5, 2020

    Published Online: Feb. 22, 2021

    The Author Email: Yang Fan (commanderjy@163.com)

    DOI:10.3788/LOP202158.0415009

    Topics