Updata export_index
This commit is contained in:
parent
b306a0ecfa
commit
121205c090
|
@ -0,0 +1,20 @@
|
|||
import os
|
||||
import pickle
|
||||
|
||||
import faiss
|
||||
|
||||
path = "crs"
|
||||
indexs_file_path = f"checkpoints/{path}/feature_and_index.pkl"
|
||||
indexs_out_dir = f"checkpoints/{path}/"
|
||||
|
||||
with open("feature_and_index.pkl",mode="rb") as f:
|
||||
indexs = pickle.load(f)
|
||||
|
||||
for k in indexs:
|
||||
print(f"Save {k} index")
|
||||
faiss.write_index(
|
||||
indexs[k],
|
||||
os.path.join(indexs_out_dir,f"Index-{k}.index")
|
||||
)
|
||||
|
||||
print("Saved all index")
|
Loading…
Reference in New Issue