Hugging Face Blog·· 2022-03-16AI 评分
用 🤗 datasets 实现图像搜索
Image search with 🤗 datasets
Hugging Face 的 🤗 datasets 现已支持图像特征类型,可结合 sentence_transformers 和 faiss 为图像数据集构建相似度搜索应用。文中以英国图书馆数字化书籍插图数据集为例,通过 ImageFolder 加载器直接载入含 image 和 label 特征的 10000 行数据,并利用 datasets 内置的 faiss 索引能力实现图像检索。
来源:Hugging Face Blog · huggingface.co