Elasticsearch向量搜索：从语义搜索到图搜图只有一步之遥

最新推荐文章于 2025-01-07 21:28:05 发布

jwolf2

最新推荐文章于 2025-01-07 21:28:05 发布

阅读量723

点赞数 2

CC 4.0 BY-SA版权

文章标签： elasticsearch 搜索引擎 AI

本文链接：https://blog.csdn.net/qq_39506978/article/details/143981433

续

上集说到语义搜索，这集接着玩一下图搜图，这种场景在电商中很常见——拍照搜商品。图搜图实现非常类似语义搜索，代码逻辑结构都很类似…

开搞

还是老地方modelscope找个Vision Transformer模型，这里选用vit-base-patch16-224,如果还想玩玩文搜图，可以选用支持多模态的multi-modal_clip-vit-base-patch16_zh

D:\python2023>modelscope download --model AI-ModelScope/vit-base-patch16-224

准备测试数据

在这里插入图片描述

运行代码

from PIL import Image
from transformers import ViTFeatureExtractor, ViTModel
import torch
import time
from elasticsearch import Elasticsearch

# 初始化模型和图片特征提取器
MODEL_PATH = 'C:\\Users\\Administrator\\.cache\\modelscope\\hub\\AI-ModelScope\\vit-base-patch16-224'
feature_extractor = ViTFeatureExtractor.from_pretrained(MODEL_PATH)
model = ViTModel.from_pretrained(MODEL_PATH)


def extract_features(image_path):
    image = Image.open(image_path)
    inputs = feature_extractor(images=image, return_tensors="pt")
    with torch.no_grad(