Attention mechanism and hash-based image retrieval method and device and storage medium
An image retrieval and attention technology, applied in still image data retrieval, metadata still image retrieval, still image data indexing, etc., can solve problems such as limiting accuracy, interfering with image embedding learning, etc., to achieve rapid retrieval and improve learning ability , improve the effect of expression
- Summary
- Abstract
- Description
- Claims
- Application Information
AI Technical Summary
Problems solved by technology
Method used
Image
Examples
Embodiment 1
[0057] like figure 1 As shown, the present embodiment provides an image retrieval method based on attention mechanism and hash, including:
[0058] S01: For each image in the original data set, the discrete hash code of each image is obtained based on the pre-trained hash feature extraction model based on the attention mechanism, and then the image feature library corresponding to the original data set is established;
[0059] S02: Obtain the discrete hash code of the image to be detected based on the pre-trained hash feature extraction model based on the attention mechanism;
[0060] S03: Query the data in the image feature library that has the closest discrete hash coding Hamming distance to the image to be detected, and the image corresponding to the data in the original data set is the retrieval result;
[0061] Wherein, the hash feature extraction model based on the attention mechanism is obtained by training the attention hash network model based on a labeled data set, ...
Embodiment 2
[0101] like figure 2 As shown, the present embodiment provides an image retrieval device based on attention mechanism, including:
[0102] The image feature library building module 1 is used to obtain the discrete hash code of each image based on the pre-trained hash feature extraction model based on the attention mechanism for each image in the original data set, and then establish a The image feature library corresponding to the set;
[0103] The discrete hash code generation module 2 of the image to be detected is used to obtain the discrete hash code of the image to be detected based on the pre-trained hash feature extraction model based on the attention mechanism;
[0104] Retrieval module 3, querying the data with the nearest discrete hash coding Hamming distance of the image to be detected in the image feature database, and the image corresponding to the data in the original data set is the retrieval result;
[0105] Wherein, the hash feature extraction model based o...
Embodiment 3
[0108]This embodiment provides a computer-readable storage medium, which stores a computer program, and when the computer program is loaded by a processor, it executes an image retrieval method based on attention mechanism and hash provided in Embodiment 1.
[0109] Those skilled in the art should understand that the embodiments of the present application may be provided as methods, systems, or computer program products. Accordingly, the present application may take the form of an entirely hardware embodiment, an entirely software embodiment, or an embodiment combining software and hardware aspects. Furthermore, the present application may take the form of a computer program product embodied on one or more computer-usable storage media (including but not limited to disk storage, CD-ROM, optical storage, etc.) having computer-usable program code embodied therein.
[0110] The present application is described with reference to flowcharts and / or block diagrams of methods, apparat...
PUM
Abstract
Description
Claims
Application Information
- R&D Engineer
- R&D Manager
- IP Professional
- Industry Leading Data Capabilities
- Powerful AI technology
- Patent DNA Extraction
Browse by: Latest US Patents, China's latest patents, Technical Efficacy Thesaurus, Application Domain, Technology Topic, Popular Technical Reports.
© 2024 PatSnap. All rights reserved.Legal|Privacy policy|Modern Slavery Act Transparency Statement|Sitemap|About US| Contact US: help@patsnap.com