西南民族大学学报(自然科学版)2024,Vol.50Issue(5) :544-553.DOI:10.11920/xnmdzk.2024.05.009

基于深度哈希与VP-Tree的快速图像检索方法

Fast image retrieval method based on deep hashing and VP-Tree

吴宗胜 李红 薛茹
西南民族大学学报(自然科学版)2024,Vol.50Issue(5) :544-553.DOI:10.11920/xnmdzk.2024.05.009

基于深度哈希与VP-Tree的快速图像检索方法

Fast image retrieval method based on deep hashing and VP-Tree

吴宗胜 1李红 1薛茹2
扫码查看

作者信息

  • 1. 咸阳师范学院计算机学院,陕西咸阳 712000
  • 2. 西藏民族大学信息工程学院,陕西咸阳 712000
  • 折叠

摘要

针对高维特征图像检索中的精度和速度挑战,提出了一种结合深度哈希技术和VP-Tree索引的快速图像检索方法.该方法首先设计了一个轻量级的深度卷积编码网络,并在网络中引入了卷积块注意力模块和空间金字塔池化技术,以增强特征提取能力;然后通过该网络模型将图像数据集中每幅图像的高维特征转化为二进制哈希编码,并与其对应的图像编号组成一个哈希表;接着使用所有图像的哈希编码来构建一个VP-Tree,在执行图像检索时将使用待查询图像的哈希编码从VP-Tree中快速找到与其距离最近的节点;最后根据这些节点的哈希值从哈希表中取出对应的结果图像.实验结果表明,所提方法在保持高检索精度的同时显著提升了检索速度(在MNIST、FASHION-MNIST和CI-FAR-10上的检索速度分别提高了 24.17、8.61和4.01倍).

Abstract

This paper proposed a fast image retrieval method that combines deep hashing technology and VP-Tree indexing to address the precision and speed challenges in high-dimensional feature image retrieval.The method first designed a lightweight deep convolutional encoding network which introduced convolutional block attention modules and spatial pyramid pooling tech-niques to enhance feature extraction capabilities.Then,through this network model,the high-dimensional features of each image in the image dataset were transformed into binary hash codes,which were combined with their corresponding image IDs to form a hash table.Subsequently,a VP-Tree was constructed using the hash codes of all images.During image retrieval,the hash code of the query image was used to quickly find the nearest nodes in the VP-Tree.Finally,the corresponding result images were re-trieved from the hash table based on the hash values of these nodes.Experimental results showed that the proposed method sig-nificantly improved retrieval speed while maintaining high retrieval accuracy(the retrieval speed on MNIST,FASHION-MNIST,and CIFAR-10 was increased by 24.17,8.61,and 4.01 times,respectively).

关键词

图像检索/深度哈希/卷积神经网络/VP-tree

Key words

image retrieval/deep hashing/convolutional neural network/VP-tree

引用本文复制引用

基金项目

国家自然科学基金资助项目(62073218)

陕西省科技厅自然科学基础研究计划面上项目(2023-JC-YB-524)

出版年

2024
西南民族大学学报(自然科学版)
西南民族大学

西南民族大学学报(自然科学版)

CSTPCD
影响因子:0.441
ISSN:2095-4271
段落导航相关论文