get result

src/diary_cover_similarity.py

@@ -95,6 +95,7 @@ def save_faiss_index(load_file, save_path):
         index2 = faiss.IndexIDMap(index)
         index2.add_with_ids(features_np, ids_np)
         faiss.write_index(index2, save_path)
+        print("faiss index saved")
 
 
 def get_similar_diary_ids_by_url(url, index, face_to_vec_f, limit=0.1):
@@ -147,7 +148,7 @@ def main():
     face_to_vec_f = lambda img: face_to_vec(img, face_rec, face_detector, shape_predictor)
 
     # save_diary_image_info(diary_after_cover_vec_file, face_to_vec_f)
-    save_faiss_index(diary_after_cover_vec_file, faiss_index_path)
+    # save_faiss_index(diary_after_cover_vec_file, faiss_index_path)
 
     faiss_index = faiss.read_index(faiss_index_path)
 
@@ -188,7 +189,7 @@ def main():
         -0.015351934358477592, -0.04335442930459976, -0.26258283853530884, -0.021509556099772453, 0.12185295671224594,
         -0.011788002215325832, 0.01337978895753622, -0.008025042712688446
     ]
-    res = get_similar_diary_ids_by_face_features(a)
+    res = get_similar_diary_ids_by_face_features(a, faiss_index, face_to_vec_f)
     print(res)