Hugging Face Blog·· 2023-01-16AI 评分35
Hugging Face 基于 Transformers 构建图像相似度系统教程
Image Similarity with Hugging Face Datasets and Transformers
AI 导读
Hugging Face 演示如何利用 🤗 Transformers 和 Datasets 库构建图像相似度检索系统。该方案使用 `nateraw/vit-base-beans` ViT 模型提取图像嵌入向量,并通过余弦相似度计算匹配度。此方法支持并行处理,可扩展至 Swin Transformer、ConvNeXT 等其他视觉模型及多模态场景。
来源:Hugging Face Blog · huggingface.co