本项目旨在利用Milvus实现以图搜图,其步骤为:
- 数据准备:我们首先需要将图片集载入至milvus,利用VGG16提取图片特征,将得到的128维向量利用milvus的insert_vectors接口插入milvus。insert_vectors随机生成的id会对应到每个图像的文件名。
- 图片查询:输入要查询图像的文件名,以.txt结尾,将图像载入。同样利用VGG16提取该图偏特征,再通过milvus的search_vectors接口搜索与该向量类似的向量。得到相似的向量id后,利用对应关系查询图片文件名并返回。