欢迎大家提pr,点点star✨
🌟 概述
本项目是基于DINOv2和CLIP模型的图像检索系统,支持以图搜图和以文搜图两种检索方式。用户可以上传图片或输入文本描述,从预定义的图像数据库中检索相似的图像。
🚀 特征
- 使用DINOv2模型进行图像特征提取和以图搜图
- 使用CLIP模型支持文本到图像的检索
- 支持不同大小的DINOv2模型(小型、基础型、大型、巨型)
- 基于余弦相似度的图像检索
- 使用FastAPI构建的Web界面
- 对大规模图像数据库进行特征缓存
界面
欢迎大家使用,提交pr和点点star🌟。