An image classification method based on bi-directional neural network structure

Abstract

The invention discloses an image classification method based on a bi-directional neural network structure, comprising the following steps: 1. The Directional layer replaces the full-connection layer in a traditional convolution network, the image classification model based on bi-directional neural network is built, 2. network forward propagation is performed, and by adding transformation matricesL and R, the rectangular structure of the last convolution layer is preserved, steps 2, 3 are repeated, the bi-directional neural network is fine-tuned until the classification network converges. 5, the class number of the image is obtained through forward propagation on the trained model. A bidirectional neural network structure is used on the premise of keeping the depth and width of the networkconstant, the dimension of the feature is transformed by designing the transformation matrix, which ensures the matrix form of convolution, effectively preserves the structure information of the original feature space of the image, and avoids the loss of spatial structure information in the process of matrix drawing into vectors in the whole connection layer.

Description

technical field [0001] The invention belongs to the field of deep learning and artificial intelligence, in particular to an image classification algorithm, which is used for large-scale image labeling and attribute classification. Background technique [0002] Image classification is one of the important directions of artificial intelligence research. It has many applications in real life. For example, in intelligent traffic monitoring, whether vehicles are covered or not, hanging license plates, agricultural satellites determine agricultural and non-agricultural areas, and Baidu launched Baidu image recognition, Google's flower recognition, etc., if the attribute prediction and classification of pictures rely entirely on manual work, a lot of manpower, material and financial resources will be invested. Using image classification technology based on deep learning can quickly and accurately classify images. [0003] At present, with the advent of the era of big data, how to i...

AI Technical Summary

Problems solved by technology

[0003] At present, with the advent of the big data era, how to improve the classification accuracy of the model through massive samples has become an important challenge and difficulty. Image classification algorithms have made great progress in recent years. Some algorithms believe that by increasing the width of the network It can improve the feature extraction ability and further improve the classification accuracy. The current algorithms rely on increasing the width or depth of the network to improve the performance of the model. If you want to further improve the performance of the algorithm, you need to repeatedly design and adjust the network and train the fine-tuning model, which will consume a lot of time. A lot of manpower and material resources, and there is no guarantee that the final classification effect will be improved

Images