Skip to content

1.1.2

Latest
Compare
Choose a tag to compare
@shibing624 shibing624 released this 22 Sep 02:49
· 56 commits to main since this release

1.1.2 版本

https://github.com/shibing624/similarities

  1. 新增图像相似度计算/图文相似度计算 + 图搜图/文搜图

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提取
  • 支持图像相似度计算、图文相似度计算
  1. 新增命令行工具(CLI),可以无需代码开发,用命令行图像转向量(多卡)、建索引、批量检索、启动服务

  2. 新增亿级图片gradio图像检索:
    dog-img

What's Changed

New Contributors

Full Changelog: 1.0.5...1.1.2