基于Milvus Cloud向量数据库搭建寻找与你时尚风格最相似的明星系统

寻找与你时尚风格最相似的明星

上述步骤都完成后，就可以开始玩转这个系统了，它可以根据你上传的图片返回前 3 个与你穿搭风格最相似的明星。

 

• 将上传图像转化为向量

首先需要处理上传的图像。以下函数需要两个参数：数据和 （embedding）模型。我们使用模型将图像转化为向量、处理图像，图像转化为列表并返回图片列表。

def embed_search_images(data, model):
    with torch.no_grad():
        print(len(data[0]))
        print(data[0][0].size())
        output = model(torch.stack(data))
        print(type(output))
        print(len(output))
        print(len(output[0]))
        print(output[0])
        if len(output) > 1:
            return output.squeeze().tolist()
        Else:
     return torch.flatten(output, start_dim=1).tolist()

如下图所示，传入本函数的 data
实际上是 data[0]
对象。