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

说明书

本发明提供一种图像检索方法、装置、电子设备、计算机可读存储介质，其中，图像检索方法包括：利用注意力机制对待检索图像进行处理，进而得到所述待检索图像的待识别区域；对所述待识别区域进行特征提取，进而得到第一特征集合；基于所述第一特征集合与样本库中的样本图像的第二特征集合的相似度，得到所述待检索图像对应的匹配图像。以此能够减少特征提取数量，提高检索效率。

技术领域

本发明涉及图像处理技术领域，尤其是涉及一种图像检索方法、装置、电子设备、计算机可读存储介质。

背景技术

随着各种社交网络的兴起，网络中的图像、视频数据每天都以指数级的速度增长，进而形成了强大的图像数据库。如何快速有效地从巨大的图像数据库中检索出用户需要的图片，成为图像检索领域一个重要的发展方向。

图像检索大致可分为两种：(1)基于文本的图像检索技术，利用文本描述的方式表示图像特征。(2)基于内容的图像检索技术(Content-based Retrieval，简称CBR)，该技术对图像的颜色、纹理、布局等进行分析和检索。目前基于内容的图像检索技术占据主流地位，很多研究主要体现在特征提取方法上，忽略了输入图像上有效信息的占比大小，这会导致需要提取的特征较多，检索时间较长，降低检索效率。

发明内容

本发明提供一种图像检索方法、装置、电子设备、计算机可读存储介质。其能够提高检索效率。

为解决上述技术问题，本发明提供的第一个技术方案为：提供一种图像检索方法，包括：利用注意力机制对待检索图像进行处理，进而得到所述待检索图像的待识别区域；对所述待识别区域进行特征提取，进而得到第一特征集合；基于所述第一特征集合与样本库中的样本图像的第二特征集合的相似度，得到所述待检索图像对应的匹配图像。

其中，所述利用注意力机制对待检索图像进行处理，进而得到所述待检索图像的待识别区域的步骤，包括：利用空间变换网络对所述待检索图像进行处理，进而得到所述待识别区域。

其中，所述利用空间变换网络对所述待检索图像进行处理，进而得到所述待识别区域的步骤，包括：利用空间变换网络对所述待检索图像进行缩放、裁剪、旋转以及非刚性变形的空间变换中至少一种处理，进而得到所述待识别区域。

其中，所述利用空间变换网络对所述待检索图像进行处理，进而得到所述待识别区域的步骤，包括：对所述待检索图像进行处理，进而得到空间变换对应的参数；基于所述参数将所述待检索图像的源像素坐标系映射到目标坐标系上；计算所述目标坐标系中各个坐标的像素值；基于所述像素值得到所述待识别区域

其中，所述对所述待检索图像进行处理，进而得到空间变换对应的参数的步骤包括：利用定位网络对所述待检索图像进行处理，进而得到空间变换对应的参数；所述定位网络由全连接层以及回归层组成。

其中，所述基于所述参数将所述待检索图像的源像素坐标系映射到目标坐标系上的步骤，包括：利用网络生成器基于所述参数将所述待检索图像的源像素坐标系映射到目标坐标系上。

其中，所述计算所述目标坐标系中各个坐标的像素值的步骤，包括：利用采样器基于双线性插值的方式计算所述目标坐标系中各个坐标的像素值。

其中，所述对所述待识别区域进行特征提取，进而得到第一特征集合的步骤，包括：对所述待识别区域进行特征提取，进而得到所述待识别区域的颜色特征、几何特征、纹理特征、深度特征中至少一种，所述待识别区域的颜色特征、几何特征、纹理特征、深度特征中至少一种为所述第一特征集合。