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

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

Image Similarity with Hugging Face Datasets and Transformers

AI 导读

Hugging Face 发布教程,演示如何用 🤗 Transformers 和 Datasets 库构建图像相似度检索系统。方案采用 ViT 模型(nateraw/vit-base-beans)提取图像嵌入向量,再用余弦相似度匹配查询图像与候选图像,示例使用 Beans 数据集,也可替换为 Swin Transformer、ConvNeXT、RegNet 等视觉模型。

来源:Hugging Face Blog · huggingface.co