Few-shot image recognition method and apparatus, device, and storage medium

Abstract

The present application discloses a few-shot image recognition method and apparatus, a device, and a storage medium. The method includes: obtaining to-be-recognized images, and constructing an image episode according to the to-be-recognized image, the image episode including a support set and a query set; inputting the image episode into a pre-trained image recognition model, the image recognition model being a few-shot image recognition model based on hard episode training; and calculating a similarity between an image in the query set and each class in the support set according to the image recognition model, and determining the class of to-be-recognized images in the query set according to the similarity. According to the image recognition method provided by the embodiments of the present application, model training and image recognition can be performed by using fewer image samples, and hard episodes are fused into a training process of a few-shot image recognition model, whereby the few-shot image recognition model can be trained more efficiently and quickly, and the trained model is higher in stability and higher in accuracy of image recognition.