Similar image retrieval method and device, computer equipment and storage medium

Abstract

The invention is suitable for the technical field of computers, and provides a similar image retrieval method and device, computer equipment and a storage medium, and the method comprises the following steps: obtaining a to-be-retrieved image; processing the to-be-retrieved image according to a pre-trained unsupervised deep hash model, and determining a hash code of the to-be-retrieved image, wherein the unsupervised deep hash model is generated by iterative optimization training based on a clustering algorithm and a deep hash algorithm in advance, wherein a pseudo tag determined through a clustering algorithm is used as an optimization target in the deep hash algorithm; and according to the hash code of the to-be-retrieved image, determining a similar hash code meeting a preset similarity relationship with the hash code, and determining a corresponding image. According to the method provided by the invention, when the unsupervised deep hash model faces massive images in the training process, the images do not need to be labeled one by one in advance, and the clustering algorithm is directly utilized to carry out false label annotation on the images, so that the method has better applicability.

Description

technical field [0001] The invention belongs to the field of computer technology, and in particular relates to a similar image retrieval method, device, computer equipment and storage medium. Background technique [0002] The retrieval process of the hash image generally includes two steps. The first step is to extract the features of the training data set to obtain the high-dimensional feature vector of the image, and then learn these high-dimensional feature vectors through the designed algorithm, and then get the learned Good hash function. In the second step, the images in the database are extracted through feature extraction and hash function mapping to obtain hash codes and stored in the database. When using an image for retrieval, the corresponding feature extraction is also performed and then the hash function is performed to obtain the hash code of the corresponding query image, and the similarity calculation is performed between the hash code and the hash code in ...

AI Technical Summary

Problems solved by technology

[0005] The purpose of the embodiments of the present invention is to provide a similar image retrieval method, which aims to solve the problem that the hash function used in the existing similar image retrieval process needs to label the image in advance before learning. Technical issues with poor applicability in scenarios

Images