Image classification method, device, computer equipment and medium

Abstract

The embodiment of the present application discloses an image classification method, device, computer equipment and medium, belonging to the field of image processing. The method includes: calling the first classification sub-model in the classification model, classifying the target image, and obtaining the first prediction result of the target image; calling the second classification sub-model in the classification model, classifying the target image, and obtaining the target image The second prediction result of the image; call the third classification sub-model in the classification model, classify the target image, and obtain the third prediction result of the target image, the training sample data of the third classification sub-model includes sample images and according to multiple The result obtained by labeling the result statistics; output the third prediction result and the classification difficulty of the target image. The classification difficulty is determined according to the first prediction result and the second prediction result. The classification difficulty can indicate the accuracy of the model output result, and avoid the model's confusion on the user. misleading.

Description

technical field [0001] The present application relates to the field of image processing, and in particular, to an image classification method, apparatus, computer equipment and medium. Background technique [0002] With the development of artificial intelligence technology and image processing technology, the current classification of images is more and more based on classification models. After training the classification model, the computer device can call the classification model to classify the image, and obtain the classification result of the image, so as to determine whether the image belongs to the target category of the classification model. However, the classification model can only output classification results, and if the classification results are not accurate enough, users may be misled. SUMMARY OF THE INVENTION [0003] The embodiments of the present application provide an image classification method, apparatus, computer equipment and medium, which can redu...

AI Technical Summary

Problems solved by technology

However, the classification model can only output classification results. If the classification results are not accurate enough, it may mislead users.

Images