Vehicle type classification based on convolutional neural network

Abstract

This paper aims to develop a framework for vehicle type classification using convolutional neural network based on vehicle rear view images. Compared with the extraction of the appearance features from vehicle side view and frontal view images, there has been relatively little research on vehicle type classification by using vehicle rear view images' information. The vehicle rear view images are detected from the traffic video frames. The image is normalized to the fixed size of 32∗32 pixels, and then through the removing-mean operation, the normalized image is sent into the trained network. As the last layer of the neural network, the softmax classifier can output the probability that the input image belongs to a certain type of vehicle. The experimental result shows that compared with traditional pattern recognition method, our algorithm performs better and eventually achieves the accuracy of 97.88% in vehicle type classification.