以下函数定义了如何将图像转换为向量并插入到 Milvus Cloud 向量数据库中。代码会循环遍历所有图像。(注意:如果需要开启 Milvus Cloud 全新特性动态 Schema,需要修改代码。)
- 查询向量数据库
以下代码演示了如何使用输入图像查询 Milvus Cloud 向量数据库,以检索和上传衣服图像最相似的的前三个结果。
02.
匹配更多风格:标示每张图像中的时尚单品
除了直接使用上述代码,查找与你着装风格最相似的 3 位明星以外,我们还可以稍微修改一下代码,拓展项目的应用场景。可以修改代码获取如下所示,不包含边界框的图像。
import torch
from torch import nn, tensor
from transformers import AutoFeatureExtractor, SegformerForSeman