Image classification method based on deep learning feature and maximum confidence path

Abstract

An image classification method based on a deep learning feature and a maximum confidence path belongs to the field of mode identification. The method comprises the steps of: training a convolutional neural network on a large enough image library; extracting an image feature by means of the trained convolutional neural network; calculating a mean vector of each class; performing iteration clustering on the mean vector that represents each class by means of a spectrum clustering algorithm so as to construct a visual tree; training svm for each non-leaf node of the tree; and for a given test image, from top to bottom, judging the probability of the test image to the corresponding node, and finding a leaf node with the biggest path probability, namely a final target class. The image feature is extracted by means of CNN, thereby achieving very good discrimination and robustness; a distance calculation formula of two classes is given out, the complexity of calculation is greatly optimized by means of derivation and the similarity of the classes is obtained, so that the visual tree is constructed by iteratively using the spectrum clustering algorithm; and the use of a visual relationship between the classes can achieve very good effects for large image classification.

Description

technical field [0001] The invention belongs to the field of pattern recognition, and in particular relates to an image classification method based on deep learning features and maximum confidence paths that can be used for large-scale image classification. Background technique [0002] In the field of computer vision, image classification is a very important and a very classic research problem. However, as the number of images increases and the types of images increase, large-scale image classification is still a very challenging task. Due to the increase in the number of images, the amount of calculation will also increase, the time required will also increase, and the hardware requirements are also high. If you still use the traditional method to train a multi-class classifier as the final classification basis, there will be calculations. issues of complexity and accuracy. Therefore, it is necessary to design a new classification framework and classification method. ...

AI Technical Summary

Problems solved by technology

[0005] The purpose of the present invention is to provide an image classification method based on deep learning features and maximum confidence paths for large-scale image classification with a large amount of calculation and low classification accuracy.

Images