Fine-grained Vehicle Recognition Using Lightweight Convolutional Neural Network with Combined Learning Strategy - Details

Author：

Indexed by：

EI Scopus

Abstract：

Fine-grained　vehicle　recognition　plays　an　important　part　in　applications,　such　as　urban　traffic　management,　public　security,　and　criminal　investigation.　It　has　great　chanllengs　due　to　the　subtle　differences　among　numerous　subcategories.　In　this　paper,　a　fine-grained　vehicle　recognition　method　using　lightweight　convolutional　neural　network　with　combined　learning　strategy　is　proposed.　Firstly,　a　lightweight　Convolutional　Neural　Network　(LWCNN)　is　designed　specially　for　the　fine-grained　vehicle　recognition　task.　Then,　a　combined　training　strategy,　including　pre-training,　fine-tuning　training　and　transfer　training,　is　proposed　to　optimize　the　LWCNN　parameters.　In　the　pre-training　phase,　ILSVRC-2012　dataset　is　adopted　to　train　the　VGG16-Net,　generating　an　initial　model.　Then,　in　the　fine-tuning　phase,　the　vehicle　dataset　is　used　for　fine-tuning　the　pre-trained　model　to　avoid　learning　parameters　from　scratch.　Finally,　in　the　transfer　training　phase,　appropriate　initialization　parameters　of　LWCNN　are　obtained　through　the　analysis　of　the　fine-tuned　network　parameters.　LWCNN　is　then　trained　using　the　vehicle　dataset　to　obtain　the　highly　accurate　and　robust　classification　model.　Compared　with　the　state-of-the-art　methods,　the　proposed　method　can　effectively　decrease　the　computational　complexity　while　maintaining　the　recognition　performance.　©　2018　IEEE.

Keyword：

Traffic control Convolutional neural networks Vehicles Convolution Deep learning Deep neural networks Big data Learning systems Transfer learning Classification (of information)

Author Community：

[ 1 ] [Zhang, Qiang]Beijing University of Technology, Beijing Key Laboratory of Computational Intelligence and Intelligent System, China
[ 2 ] [Zhang, Qiang]College of Microelectronics, Faculty of Information Technology, Beijing University of Technology, Beijing, China
[ 3 ] [Zhuo, Li]Beijing University of Technology, Beijing Key Laboratory of Computational Intelligence and Intelligent System, China
[ 4 ] [Zhuo, Li]College of Microelectronics, Faculty of Information Technology, Beijing University of Technology, Beijing, China
[ 5 ] [Zhang, Shiyu]Beijing University of Technology, Beijing Key Laboratory of Computational Intelligence and Intelligent System, China
[ 6 ] [Zhang, Shiyu]College of Microelectronics, Faculty of Information Technology, Beijing University of Technology, Beijing, China
[ 7 ] [Li, Jiafeng]Beijing University of Technology, Beijing Key Laboratory of Computational Intelligence and Intelligent System, China
[ 8 ] [Li, Jiafeng]College of Microelectronics, Faculty of Information Technology, Beijing University of Technology, Beijing, China
[ 9 ] [Zhang, Hui]Beijing University of Technology, Beijing Key Laboratory of Computational Intelligence and Intelligent System, China
[ 10 ] [Zhang, Hui]College of Microelectronics, Faculty of Information Technology, Beijing University of Technology, Beijing, China
[ 11 ] [Li, Xiaoguang]Beijing University of Technology, Beijing Key Laboratory of Computational Intelligence and Intelligent System, China
[ 12 ] [Li, Xiaoguang]College of Microelectronics, Faculty of Information Technology, Beijing University of Technology, Beijing, China