Laser & Optoelectronics Progress, Volume. 58, Issue 4, 0415009(2021)
Classification Method of Crossing Vehicle Based on Improved Residual Network
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.
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
Category: Machine Vision
Received: Sep. 2, 2020
Accepted: Nov. 5, 2020
Published Online: Feb. 22, 2021
The Author Email: Yang Fan (commanderjy@163.com)