首页|融合注意力机制的深度哈希图像检索方法

融合注意力机制的深度哈希图像检索方法

扫码查看
传统的基于深度哈希图像检索方法在获取图像的特征信息时,会关注到部分冗余信息,影响最终的图像检索精度.针对上述问题,提出一种应用于卷积神经网络中的融合跨维度交互注意力机制模块,该模块可以提高网络的性能,学习到更多有利于图像检索的特征信息.在深度哈希图像检索任务中,选用VGG16 与ResNet18 两种经典模型作为图像检索的基础模型,加入注意力模块并且重新设计哈希码目标损失函数后,在CIFAR-10 和NUS-WIDE数据集上进行了对比实验,实验结果表明添加了注意力机制后的图像检索精度有较大提高,验证了所提出方法的有效性.
Deep Hash Image Retrieval Method Integrating Attention Mechanism
Traditional deep hash-based image retrieval methods focus on some redundant information when obtaining feature information of images,which affects the final image retrieval accuracy.In response to the above issues,this article proposes a fusion cross dimensional interactive attention mechanism module,which can be applied to convolutional neural networks to improve net-work performance and learn more feature information that is conducive to image retrieval.In the deep hash image retrieval task,two classic models,VGG16 and ResNet18,were selected as the basic models for image retrieval.After adding an attention module and redesigning the hash code target loss function,comparative experiments were conducted on the CIFAR-10 and NUS-WIDE datasets.The experimental results showed that the addition of attention mechanism significantly improved the accuracy of image re-trieval,verifying the effectiveness of the proposed method.

image retrievalattention moduleconvolutional neural networkdeep hash

金川、付小思

展开 >

安徽广播影视职业技术学院艺术设计学院,安徽合肥 230001

华中师范大学计算机学院,湖北武汉 430001

图像检索 注意力模块 卷积神经网络 深度哈希

2024

荆楚理工学院学报
荆楚理工学院

荆楚理工学院学报

影响因子:0.168
ISSN:1008-4657
年,卷(期):2024.39(4)