Image classification method and device, computer equipment and medium

Abstract

The embodiment of the invention discloses an image classification method and device, computer equipment and a medium, and belongs to the field of image processing. The image classification method comprises the steps of calling a first classification sub-model in a classification model, and performing classification processing on a target image to obtain a first prediction result of the target image; calling a second classification sub-model in the classification model, and performing classification processing on the target image to obtain a second prediction result of the target image; callinga third classification sub-model in the classification model, performing classification processing on the target image to obtain a third prediction result of the target image, training sample data ofthe third classification sub-model comprising a sample image and a result obtained by statistics according to a plurality of annotation results; and outputting the third prediction result and the classification difficulty of the target image, determining the classification difficulty according to the first prediction result and the second prediction result, indicating the accuracy of the model output result through the classification difficulty, and avoiding misleading of the model to the user.

Description

technical field [0001] The present application relates to the field of image processing, in particular to an image classification method, device, computer equipment and media. Background technique [0002] With the development of artificial intelligence technology and image processing technology, current image classification processing is increasingly based on classification models. After the classification model is trained, the computer device can call the classification model to classify the image, and obtain the classification result of the image 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, it may cause misleading to users. Contents of the invention [0003] Embodiments of the present application provide an image classification method, device, computer equipment and media, which can reduce mis...

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