提交 20b35497 编写于 作者: D dongshuilong

update faiss

上级 687c1352
...@@ -109,7 +109,8 @@ class GalleryBuilder(object): ...@@ -109,7 +109,8 @@ class GalleryBuilder(object):
if operation_method != "remove": if operation_method != "remove":
start_id = max(ids.keys()) + 1 if ids else 0 start_id = max(ids.keys()) + 1 if ids else 0
ids_now = np.arange(0, len(gallery_images)) + start_id ids_now = (
np.arange(0, len(gallery_images)) + start_id).astype(np.int64)
if operation_method == "new": if operation_method == "new":
index.train(gallery_features) index.train(gallery_features)
index.add_with_ids(gallery_features, ids_now) index.add_with_ids(gallery_features, ids_now)
......
# 向量检索 # 向量检索
*****注意**:由于系统适配性问题,在新版本中,此检索算法将被废弃。新版本中将使用[faiss](https://github.com/facebookresearch/faiss),整体检索模型库的使用流程不变,但是简历索引及检索时的yaml文件有所修改。
## 1. 简介 ## 1. 简介
......
# Vector search # Vector search
*Due to the system adaptability problem, this retrieval algorithm will be abandoned in the new version. [faiss](https://github.com/facebookresearch/faiss) will be used in the new version. The use process of the overall retrieval system base will remain unchanged, but the yaml files for build indexes and retrieval will be modified.
## 1. Introduction ## 1. Introduction
Some vertical domain recognition tasks (e.g., vehicles, commodities, etc.) require a large number of recognized categories, and often use a retrieval-based approach to obtain matching predicted categories by performing a fast nearest neighbor search with query vectors and underlying library vectors. The vector search module provides the basic approximate nearest neighbor search algorithm based on Baidu's self-developed Möbius algorithm, a graph-based approximate nearest neighbor search algorithm for maximum inner product search (MIPS). This module provides python interface, supports numpy and tensor type vectors, and supports L2 and Inner Product distance calculation. Some vertical domain recognition tasks (e.g., vehicles, commodities, etc.) require a large number of recognized categories, and often use a retrieval-based approach to obtain matching predicted categories by performing a fast nearest neighbor search with query vectors and underlying library vectors. The vector search module provides the basic approximate nearest neighbor search algorithm based on Baidu's self-developed Möbius algorithm, a graph-based approximate nearest neighbor search algorithm for maximum inner product search (MIPS). This module provides python interface, supports numpy and tensor type vectors, and supports L2 and Inner Product distance calculation.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册