Auto Image Classification Based on Convolution Neural Network

Aiming at the low accuracy of vehicle image retrieval algorithm based on deep learning, an improved vehicle image classification retrieval model based on convolutional neural network is proposed. According to the complexity of the car image, using convolution neural network to extract the image features from Stanford Cars Dataset database, and use a local feature aggregation descriptor (vector of locally aggregated descriptors, VLAD) to represent a picture. Finally, SVM is used to classify the image of the car. The experimental results show that compared with the traditional visual feature classification algorithm, the accuracy of the model is higher and the retrieval effect is better.

[1]  Trevor Darrell,et al.  DeCAF: A Deep Convolutional Activation Feature for Generic Visual Recognition , 2013, ICML.

[2]  Stefan Carlsson,et al.  CNN Features Off-the-Shelf: An Astounding Baseline for Recognition , 2014, 2014 IEEE Conference on Computer Vision and Pattern Recognition Workshops.

[3]  Martin Layton,et al.  Augmented statistical models: exploiting generative models in discriminative classifiers , 2005 .

[4]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[5]  Josef Sivic,et al.  NetVLAD: CNN Architecture for Weakly Supervised Place Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[6]  Cordelia Schmid,et al.  Aggregating local descriptors into a compact image representation , 2010, 2010 IEEE Computer Society Conference on Computer Vision and Pattern Recognition.

[7]  Cordelia Schmid,et al.  Beyond Bags of Features: Spatial Pyramid Matching for Recognizing Natural Scene Categories , 2006, 2006 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR'06).