Image classification method and device, readable storage medium and terminal equipment

Abstract

The invention relates to the technical field of image processing, in particular to an image classification method and device, a storage medium and terminal equipment. The image classification method provided by the invention comprises the following steps: obtaining a to-be-classified image; inputting the to-be-classified image into a trained image classification model to obtain an image label output by the image classification model, wherein the image classification model comprises a teacher model and a student model, and the teacher model is a convolutional neural network model obtained by carrying out weak supervision training by utilizing noise data and non-noise data, and the student model is a model obtained by carrying out joint training according to a knowledge migration mechanism and a multi-task learning mechanism based on the teacher model; obtaining a classification result corresponding to the to-be-classified image according to the image label, using the teacher model to perform knowledge migration, and using the noise data and the non-noise data to perform multi-task learning training, thereby improving the classification efficiency and the classification accuracy of the image classification model.

Description

technical field [0001] The present invention relates to the technical field of image processing, in particular to an image classification method, device, computer-readable storage medium and terminal equipment. Background technique [0002] With the development of the era of Internet big data, deep convolutional neural network technology has been successfully applied to the field of artificial intelligence, for example, the category of images can be identified through deep convolutional neural network technology for image classification, the existing deep convolutional neural network The key factor for network technology to achieve image classification is that when training deep convolutional neural network technology, the collected image sets for training must have very reliable label data. In the current situation of massive network data images and crawler images, it is very difficult and time-consuming to obtain reliable and high-quality label data, and once the label dat...

AI Technical Summary

Problems solved by technology

[0003] The embodiment of the present invention provides an image classification method, device, computer-readable storage medium and terminal equipment, which can solve the problem of classification efficiency and classification in existing image classification Problems with lower accuracy

Images