用AI + Milvus Cloud搭建着装搭配推荐系统教程

以下函数定义了如何将图像转换为向量并插入到 Milvus Cloud 向量数据库中。代码会循环遍历所有图像。（注意：如果需要开启 Milvus Cloud 全新特性动态 Schema，需要修改代码。）

 

• 查询向量数据库

 

以下代码演示了如何使用输入图像查询 Milvus Cloud 向量数据库，以检索和上传衣服图像最相似的的前三个结果。

 

 

02.

匹配更多风格：标示每张图像中的时尚单品

 

除了直接使用上述代码，查找与你着装风格最相似的 3 位明星以外，我们还可以稍微修改一下代码，拓展项目的应用场景。可以修改代码获取如下所示，不包含边界框的图像。

 

import torch
from torch import nn, tensor
from transformers import AutoFeatureExtractor, SegformerForSeman