ModaHub魔搭社区：AI原生云向量数据库Zilliz Cloud与 PyTorch 集成搭建图片搜索系统

目录

准备工作

准备数据

主要参数

设置 Zilliz Cloud

插入数据

执行搜索

---

在本文中，我们将演示如何使用 Zilliz Cloud 搭建一个简单的图片搜索系统。该系统使用的数据集是 Kaggle 上提供的印象派大师分类器数据集（ Impressionist-Classifier Dataset ）。在本示例中，我们将该数据库重新托管到了一个公共的 Google Drive 中。

本示例需要使用一个 1 CU 的 Zilliz Cloud Cluster，并使用 PyTorch 的预训练 ResNet50 模型获取指定图片的向量表示。

现在，让我们开始吧！

准备工作

本页中的脚本需要使用 pymilvus 连接 Zilliz Cloud，使用 torch 运行 Embedding 模型࿰