A method and device for establishing an image classification neural network model and image classification

Abstract

Embodiments of the present invention provide an image classification neural network model and an image classification method and device. In the method for establishing an image classification neural network model, a training set containing classified images is first obtained, and then a densely connected convolutional neural network The distance between each convolutional layer in the dense block of the model, determine the number of feature maps of each convolutional layer in the input dense block, form the second densely connected convolutional neural network model, and use the training set to the second densely connected The convolutional neural network model is trained to generate an image classification neural network model, and then the image classification neural network model is used to classify the image data to be classified. The embodiment of the present invention reduces the amount of network parameters in the image classification process.

Description

technical field [0001] The invention relates to the technical field of image processing, in particular to a method and device for establishing an image classification neural network model and image classification. Background technique [0002] Image classification technology has a wide range of applications in character recognition, face recognition, object recognition, pedestrian detection, image retrieval, etc. The main idea of ​​image classification is: first analyze the target or scene contained in the sample image, then convert the original pixels of the image into feature information of the image, use these feature information to represent the description information of each image, and finally according to the obtained These descriptive information are used for image classification. The traditional method of image classification has limitations. It needs to extract and classify image features separately, and the extraction of image features is the premise of accurate ...

AI Technical Summary

Problems solved by technology

[0004] In view of the above analysis, the embodiment of the present invention proposes an image classification neural network model and an image classification method and device to solve the existing densely connected convolutional neural network. With the widening of the network width and deepening of the depth, The problem of explosive growth in the number of network parameters

Images