[发明专利]相似图像确定方法及装置、电子设备和存储介质

权利要求书

1.一种相似图像确定方法，其特征在于，包括：

获取目标图像，并对所述目标图像进行初始特征提取处理以得到初始特征图；

对所述初始特征图进行全局平均池化处理和最大池化处理，分别得到第一平均池化初始特征和最大池化初始特征，并根据所述第一平均池化初始特征和所述最大池化初始特征确定对应的通道特征；

对所述初始特征图进行次大池化处理和平均池化处理，分别得到次大池化初始特征和第二平均池化初始特征，并根据将所述次大池化初始特征和所述第二平均池化初始特征确定对应的空间场景特征；

对所述通道特征和空间场景特征进行特征聚合处理，以得到对应的目标图像特征；

根据所述目标图像特征从多个候选图像中确定所述目标图像的相似图像。

2.根据权利要求1所述的方法，其特征在于，所述方法还包括：

获取预先构建的特征提取模型；所述特征提取模型包括通道注意力模块和空间注意力模块；

由所述通道注意力模块对所述初始特征图进行通道特征提取，得到所述通道特征；

由所述空间注意力模块对所述初始特征图进行空间场景特征提取，得到所述空间场景特征。

3.根据权利要求2所述的方法，其特征在于，所述特征提取模型通过下述步骤训练得到：

获取初始特征提取模型，所述初始特征提取模型包括通道注意力模块、初始空间注意力模块和重构空间注意力模块；

获取训练样本集，并根据所述初始特征提取模型确定模型损失函数；

基于所述训练样本集，并通过所述通道注意力模块与所述重构空间注意力模块进行第一训练次数的模型训练；

基于所述训练样本集，并通过所述通道注意力模块与所述初始空间注意力模块进行第二训练次数的模型训练，直至所述模型损失函数收敛，得到所述特征提取模型。

4.根据权利要求2所述的方法，其特征在于，所述由所述通道注意力模块对所述初始特征图进行通道特征提取，得到所述通道特征，包括：

对所述初始特征图进行全局平均池化处理和最大池化处理，以分别得到第一平均池化初始特征和最大池化初始特征；

分别对所述第一平均池化初始特征和最大池化初始特征进行全连接处理，以得到对应的第一平均池化特征和最大池化特征；

将所述第一平均池化特征和所述最大池化特征进行特征相加处理，得到第一中间特征；

基于所述第一中间特征并通过第一激活函数确定第一权重系数，以根据所述初始特征图与所述第一权重系数确定所述通道特征。

5.根据权利要求2所述的方法，其特征在于，所述由所述空间注意力模块对所述初始特征图进行空间场景特征提取，得到所述空间场景特征，包括：

对所述初始特征图进行次大池化处理和平均池化处理，以分别得到次大池化初始特征和第二平均池化初始特征；

将所述次大池化初始特征和所述第二平均池化初始特征进行通道拼接处理，得到初始中间特征；

对所述初始中间特征进行卷积特征提取，得到第二中间特征；

基于所述第二中间特征并通过第二激活函数确定第二权重系数，以根据所述初始特征图与所述第二权重系数确定所述空间场景特征。

6.根据权利要求1所述的方法，其特征在于，所述根据所述目标图像特征从多个候选图像中确定所述目标图像的相似图像，包括：

确定所述目标图像特征对应的目标特征向量；

根据所述目标特征向量进行向量搜索处理，得到多个初始相似向量；

确定各所述初始相似向量与所述目标特征向量之间的向量相似度，并确定多个所述向量相似度的相似度排序结果；

根据所述相似度排序结果从所述初始相似向量中确定相似图像向量，并将所述相似图像向量对应的图像作为所述相似图像。