本文整理匯總了Python中faiss.IndexIVFPQ方法的典型用法代碼示例。如果您正苦於以下問題：Python faiss.IndexIVFPQ方法的具體用法？Python faiss.IndexIVFPQ怎麽用？Python faiss.IndexIVFPQ使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類faiss的用法示例。
在下文中一共展示了faiss.IndexIVFPQ方法的3個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: train_index
# 需要導入模塊: import faiss [as 別名]
# 或者: from faiss import IndexIVFPQ [as 別名]
def train_index(data, quantizer_path, trained_index_path, fine_quant='SQ8', cuda=False):
quantizer = faiss.read_index(quantizer_path)
if fine_quant == 'SQ8':
trained_index = faiss.IndexIVFScalarQuantizer(quantizer, quantizer.d, quantizer.ntotal, faiss.METRIC_L2)
elif fine_quant.startswith('PQ'):
m = int(fine_quant[2:])
trained_index = faiss.IndexIVFPQ(quantizer, quantizer.d, quantizer.ntotal, m, 8)
else:
raise ValueError(fine_quant)
if cuda:
if fine_quant.startswith('PQ'):
print('PQ not supported on GPU; keeping CPU.')
else:
res = faiss.StandardGpuResources()
gpu_index = faiss.index_cpu_to_gpu(res, 0, trained_index)
gpu_index.train(data)
trained_index = faiss.index_gpu_to_cpu(gpu_index)
else:
trained_index.train(data)
faiss.write_index(trained_index, trained_index_path) 示例2: __init__
# 需要導入模塊: import faiss [as 別名]
# 或者: from faiss import IndexIVFPQ [as 別名]
def __init__(self, database, method, M=128, nbits=8, nlist=316, nprobe=32):
super().__init__(database, method)
self.quantizer = {'cosine': faiss.IndexFlatIP,
'euclidean': faiss.IndexFlatL2}[method](self.D)
self.index = faiss.IndexIVFPQ(self.quantizer, self.D, nlist, M, nbits)
samples = database[np.random.permutation(np.arange(self.N))[:self.N]]
print("[ANN] train")
self.index.train(samples)
self.add()
self.index.nprobe = nprobe 示例3: __init__
# 需要導入模塊: import faiss [as 別名]
# 或者: from faiss import IndexIVFPQ [as 別名]
def __init__(self, database, method, M=128, nbits=8, nlist=316, nprobe=64):
super().__init__(database, method)
self.quantizer = {'cosine': faiss.IndexFlatIP,
'euclidean': faiss.IndexFlatL2}[method](self.D)
self.index = faiss.IndexIVFPQ(self.quantizer, self.D, nlist, M, nbits)
samples = database[np.random.permutation(np.arange(self.N))[:self.N // 5]]
print("[ANN] train")
self.index.train(samples)
self.add()
self.index.nprobe = nprobe