📸 嵌入照片 🖼️
欢迎来到嵌入照片,这是一个由@harperreed构建的强大的照片相似度搜索引擎!🎉 该项目利用CLIP(对比语言-图像预训练)模型,基于文本描述查找视觉上相似的图像。🔍🖼️
🌟 特性
- 🚀 使用CLIP模型进行快速高效的图像搜索
- 💻 仅适用于Apple Silicon (MLX)
- 💾 使用SQLite和Chroma持久存储图像嵌入
- 🌐 易于交互和探索的Web界面
- 🔒 安全的图像服务和处理
- 📊 用于性能分析的日志记录和监控
- 🔧 使用环境变量进行可配置设置
截图
📂 仓库结构
embed-photos/
├── README.md
├── generate_embeddings.py
├── requirements.txt
├── start_web.py
└── templates
├── README.md
├── base.html
├── display_image.html
├── index.html
├── output.txt
└── query_results.html
generate_embeddings.py
:使用CLIP模型生成图像嵌入的脚本requirements.txt
:列出所需的Python依赖项start_web.py
:用于照片相似度搜索的Flask Web应用程序templates/
:包含Web界面的HTML模板
🚀 开始使用
-
克隆仓库:
git clone https://github.com/harperreed/photo-similarity-search.git
-
安装所需依赖:
pip install -r requirements.txt
-
通过在
.env
文件中设置必要的环境变量来配置应用程序。 -
生成图像嵌入:
python generate_embeddings.py
-
启动Web应用程序:
python start_web.py
-
打开Web浏览器并导航至
http://localhost:5000
以探索照片相似度搜索!
待办事项
- 使用siglip代替clip
- 添加更强大的配置
- 使mlx可选
🙏 致谢
嵌入照片项目建立在Apple(mlx!)、CLIP模型的工作基础之上,并利用了各种开源库。我们向这些项目的作者和贡献者表示感谢。
祝搜索愉快!🔍✨