跳到正文
原文
Hugging Face Blog·· 2022-03-16AI 评分42

Hugging Face datasets 库结合 sentence_transformers 与 faiss 实现图像搜索

Image search with 🤗 datasets

AI 导读

Hugging Face 官方展示如何利用 datasets 库、sentence_transformers 和 faiss 构建图像搜索应用。该方案使用英国图书馆的书籍插图数据集,通过 datasets 的 ImageFolder 加载器读取图像,并借助 faiss 索引实现基于向量相似度的图像检索,超越了传统标签搜索。

来源:Hugging Face Blog · huggingface.co