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

用 🤗 datasets 实现图像搜索

Image search with 🤗 datasets

AI 导读

Hugging Face 的 🤗 datasets 现已支持图像特征类型,可结合 sentence_transformers 和 faiss 为图像数据集构建相似度搜索应用。文中以英国图书馆数字化书籍插图数据集为例,通过 ImageFolder 加载器直接载入含 image 和 label 特征的 10000 行数据,并利用 datasets 内置的 faiss 索引能力实现图像检索。

来源:Hugging Face Blog · huggingface.co