相似度查询工具
1.背景
想要实现一个图片特征向量相似度搜索的功能,项目使用的是Java开发,数据库是PostgreSQL,可选择的方案有:
- Vector database - Milvus部署方便,有可视化界面Attu,有JavaSDK(但是需要专门部署)。
- PostgreSQL插件(Cube 支持100维,Pase 支持512维,Vector 支持16000维)。
由于提取的图片的特征向量有1024维,所以只能使用Milvus和PostgreSQL插件Vector了。
2.应用
2.1 Milvus
Milvus官网有详细的安装流程和代码这里不再赘述,使用Docker安装,版本为2.2.9
,这里为大家提供一个简单的工具类,数据库连接参数没有参数化,小伙伴们可以优化,对结果数据进行了简单的格式化:
结果封装:
@Data
@Builder
public class MilvusRes {
public float score;