Vehicle classification method based on convolution neural network

Abstract

The invention discloses a vehicle classification method based on a convolution neural network. The vehicle classification method is concretely implemented according to the following steps that step 1: learning samples are acquired, and category tags are marked on the samples; step 2: the acquired learning samples act as training data to train a convolution neural network model so that excellent network model parameters are obtained; step 3: the characteristics of training data are extracted by using the trained convolution neural network model, and tenfold cross training is performed by using a support vector machine classification model constructed by a liblinear classifier so that the support vector machine classification model is obtained; and step 4: the characteristics of vehicle models under classification are extracted by suing the convolution neural network model, and then the vehicle categories to which vehicle images under classification belong are obtained by using the support vector machine classification model. All the connection layer output of the convolution neural network is used as the characteristic representation of the vehicle images and then the vehicle images are classified by using a SVM classifier so that the great accuracy of vehicle classification can be obtained.

Description

technical field [0001] The invention belongs to the technical field of machine learning and computer vision, and in particular relates to a vehicle classification method based on a convolutional neural network. Background technique [0002] With the continuous improvement of social living standards, the number of automobiles as a means of transportation shows a trend of rapid growth. But at the same time, it also brings a series of traffic problems such as traffic accidents and congestion, which brings huge challenges to traffic supervision. [0003] Vehicle recognition is an important part of intelligent transportation system. With the development of pattern recognition, image processing and computer vision technology, vehicle recognition technology based on image processing has received more and more attention. Especially in highway safety management, vehicle type identification plays an important role in traffic flow management, expressway toll collection and detection ...

AI Technical Summary

Problems solved by technology

[0005] The purpose of the present invention is to provide a vehicle classification method based on convolutional neural network, which solves the problem of low accuracy of existing vehicle type identification

Images