跳到正文
原文
Hugging Face Blog·· 2023-01-16AI 评分58

如何用 Hugging Face Datasets 和 Transformers 构建图像相似度检索系统

Image Similarity with Hugging Face Datasets and Transformers

AI 导读

Hugging Face 官方博客发布教程,讲解如何用 Transformers 和 Datasets 构建图像相似度系统:用 nateraw/vit-base-beans checkpoint 提取图像嵌入向量,通过余弦相似度在 beans 数据集 100 个候选样本中检索与查询图最相似的 Top 5。

来源:Hugging Face Blog · huggingface.co