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