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

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


 

  • 查询向量数据库


 

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


 

 

02.

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


 

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


 

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

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
乘风的头像乘风管理团队
上一篇 2023年9月15日
下一篇 2023年9月15日

相关推荐