save index

src/diary_cover_similarity.py

@@ -67,10 +67,12 @@ def save_faiss_index(load_file, save_path):
 
         ids_np = np.array(ids).astype("float32")
         features_np = np.array(features).astype("int")
+        print(ids_np)
+        print(features_np)
         index = faiss.IndexHNSWFlat(128, 32)
-        index = faiss.IndexIDMap(index)
-        index.add_with_ids(features_np, ids_np)
-        faiss.write_index(index, save_path)
+        index2 = faiss.IndexIDMap(index)
+        index2.add_with_ids(features_np, ids_np)
+        faiss.write_index(index2, save_path)
 
 
 def main():
@@ -88,7 +90,7 @@ def main():
     shape_predictor = dlib.shape_predictor(shape_model_path)
     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_diary_image_info(diary_after_cover_vec_file, face_to_vec_f)
     save_faiss_index(diary_after_cover_vec_file, faiss_index_path)
 
     # img_url = "https://pic.igengmei.com/2020/07/03/1437/1b9975bb0b81-w"
@@ -127,4 +129,4 @@ if __name__ == "__main__":
 
     main()
 
-    print("total cost: {:.2f}mins".format((time.time() - begin_time)) / 60)
+    print("total cost: {:.2f}mins".format((time.time() - begin_time) / 60))