[发明专利]图像检索方法、装置、计算机设备及存储介质

权利要求书

1.一种图像检索方法，其特征在于，所述方法包括：

对目标图像进行特征提取，得到所述目标图像的第一特征图，所述第一特征图用于表征所述目标图像中第一注意力区域的图像特征；

基于所述第一特征图对所述目标图像进行注意力引导，得到所述目标图像对应的引导图像，所述引导图像的注意力区域为第二注意力区域，且所述第二注意力区域不同于所述第一注意力区域；

对所述引导图像进行特征提取，得到所述引导图像的第二特征图，所述第二特征图用于表征所述引导图像中所述第二注意力区域的图像特征；

基于所述第一特征图生成第一特征向量，并基于所述第二特征图生成第二特征向量；

基于所述第一特征向量和所述第二特征向量进行图像检索，得到图像检索结果，所述图像检索结果中包含所述目标图像的相似图像。

2.根据权利要求1所述的方法，其特征在于，所述基于所述第一特征图对所述目标图像进行注意力引导，得到所述目标图像对应的引导图像，包括：

基于所述第一特征图生成所述目标图像对应的第一掩膜，所述第一掩膜表征的注意力区域为所述第一注意力区域；

基于所述第一掩膜生成第二掩膜，所述第二掩膜表征的注意力区域为所述第二注意力区域；

基于所述第二掩膜和所述目标图像生成所述引导图像。

3.根据权利要求2所述的方法，其特征在于，所述基于所述第一特征图生成所述目标图像对应的第一掩膜，包括：

基于各个通道的所述第一特征图生成原始掩膜，所述原始掩膜中的数值为连续值；

基于所述原始掩膜的数值平均值，对所述原始掩膜中的各个数值进行二值化处理，得到所述第一掩膜；

所述基于所述第一掩膜生成第二掩膜，包括：

对所述第一掩膜中的数值进行反转，得到所述第二掩膜。

4.根据权利要求2所述的方法，其特征在于，所述基于所述第二掩膜和所述目标图像生成所述引导图像，包括：

将所述第二掩膜中的数值由特征空间反映射至图像空间，得到图像掩膜；

利用所述图像掩膜对所述目标图像进行掩膜处理，生成所述引导图像。

5.根据权利要求1至4任一所述的方法，其特征在于，所述基于所述第一特征向量和所述第二特征向量进行图像检索，得到图像检索结果，包括：

对所述第一特征向量和所述第二特征向量进行拼接，得到目标融合特征向量；

确定所述目标融合特征向量与图库中图像对应融合特征向量之间的向量距离；

基于所述向量距离确定所述图像检索结果。

6.根据权利要求5所述的方法，其特征在于，所述基于所述第一特征向量和所述第二特征向量进行图像检索，得到图像检索结果之后，所述方法还包括：

响应于所述图库中不存在与所述目标图像重复的图像，将所述目标图像和所述目标融合特征向量关联存储至所述图库。

7.根据权利要求1至4任一所述的方法，其特征在于，所述第一特征图和所述第二特征图由特征提取网络的卷积层提取得到，所述第一特征向量和所述第二特征向量由所述特征提取网络的全连接层生成。