本申請實施例公開了一種圖像搜索方法、裝置和存儲介質，通過獲取待搜索圖像；獲取待搜索圖像在多個圖像相似維度上的特征信息；可以基于多個圖像相似維度上的特征信息，于候選圖像中在多個圖像相似維度上搜索相似圖像，得到相似圖像集合，相似圖像集合中的相似圖像與待搜索圖像在至少一個圖像相似維度上相似；再根據相似圖像集合中相似圖像對應的圖像相似維度，對相似圖像集合中的相似圖像進行排序，得到排序后集合；從排序后集合中選擇預設數量的目標相似圖像；輸出目標相似圖像，通過上述方案，可以對待搜索圖像在多個圖像相似維度上進行相似圖像的搜索，提升搜索結果的多樣性，提升圖像搜索能力。

技術領域

本申請涉及計算機技術領域，具體涉及一種圖像搜索方法、裝置和存儲介質。

背景技術

目前，以圖搜圖作為一種圖像搜索功能，在一些即時通訊軟件和搜索引擎中被提供給用戶使用，對于以圖搜圖，需要能夠根據輸入的圖像搜索目標圖像。

發明內容

有鑒于此，本申請實施例提供了一種圖像搜索方法、裝置和存儲介質，可以提升圖像搜索的搜索能力以及提升搜索結果的多樣性。

本申請實施例提供了一種圖像搜索方法，該方法包括：

獲取待搜索圖像；

獲取所述待搜索圖像在多個圖像相似維度上的特征信息；

基于所述多個圖像相似維度上的特征信息，于候選圖像中在多個圖像相似維度上搜索相似圖像，得到相似圖像集合，所述相似圖像集合中的相似圖像與所述待搜索圖像在至少一個圖像相似維度上相似；

根據所述相似圖像集合中相似圖像對應的圖像相似維度，對所述相似圖像集合中的相似圖像進行排序，得到排序后集合；

從所述排序后集合中選擇預設數量的目標相似圖像；

輸出目標相似圖像。

在一個實施例中，所述圖像相似維度包括：圖像承載對象相似維度、圖像語義相似維度以及圖像操作行為相似維度；

所述基于所述多個圖像相似維度上的特征信息，于候選圖像中在多個圖像相似維度上搜索相似圖像包括：

基于圖像承載對象相似維度上的特征信息，于候選圖像中搜索與所述待搜索圖像在圖像承載對象相似維度上相似的相似圖像；

基于圖像語義相似維度上的特征信息，于候選圖像中搜索與所述待搜索圖像在圖像語義相似維度上相似的相似圖像；

基于圖像操作行為相似維度上的特征信息，于候選圖像中搜索與所述待搜索圖像在圖像操作行為相似維度上相似的相似圖像。

在一個實施例中，圖像承載對象相似維度包括：圖像特征相似子維度；

獲取待搜索圖像在圖像承載對象相似維度上的特征信息，包括：

通過特征提取模型的局部特征提取塊，提取所述待搜索圖像的局部特征；

通過對應于所述局部特征提取塊設置的注意力模塊，學習所述局部特征提取塊提取的局部特征的特征權重；

基于所述特征權重對所述局部特征提取塊提取的局部特征進行處理，得到注意力局部特征，基于所述注意力局部特征得到所述待搜索圖像的圖像特征；

所述基于圖像承載對象相似維度上的特征信息，于候選圖像中搜索與所述待搜索圖像在圖像承載對象相似維度上相似的相似圖像，包括：

獲取候選圖像基于所述特征提取模型提取的圖像特征；

從所述候選圖像中，搜索圖像特征與所述待搜索圖像的圖像特征相似的相似圖像。

在一個實施例中，所述圖像承載對象相似維度包括：人臉相似子維度；