1.1.2 版本
https://github.com/shibing624/similarities
- 新增图像相似度计算/图文相似度计算 + 图搜图/文搜图
CLIP图文匹配模型,可用于图文特征(embeddings)、相似度计算、图文检索、零样本图片分类,本项目基于PyTorch实现了CLIP模型的向量表征、构建索引(基于AutoFaiss)、批量检索、后台服务(基于FastAPI)、前端展现(基于Gradio)功能
- 支持openai/clip-vit-base-patch32等CLIP系列模型
- 支持OFA-Sys/chinese-clip-vit-huge-patch14等Chinese-CLIP系列模型
- 支持前后端分离部署,FastAPI后端服务,Gradio前端展现
- 支持亿级数据高效检索,基于Faiss检索,支持GPU加速
- 支持图搜图、文搜图、向量搜图
- 支持图像embedding提取、文本embedding提取
- 支持图像相似度计算、图文相似度计算
What's Changed
- Code refactoring by @tosemml in #15
- Faiss by @shibing624 in #19
New Contributors
- @tosemml made their first contribution in #15
- @shibing624 made their first contribution in #19
Full Changelog: 1.0.5...1.1.2