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

用 Hugging Face Datasets 和 Transformers 构建图像相似度系统

Image Similarity with Hugging Face Datasets and Transformers

AI 导读

Hugging Face 发布教程,演示如何用 Transformers 和 Datasets 库构建图像相似度检索系统,通过 ViT 模型(nateraw/vit-base-beans)提取图像嵌入向量,再用余弦相似度找出与查询图像最接近的候选图。

来源:Hugging Face Blog · huggingface.co