save url to file

src/diary_cover_similarity.py

@@ -67,7 +67,7 @@ def save_diary_image_info(save_file, face_to_vec_f):
                 if img.any():
                     faces = face_to_vec_f(img)
                     for face in faces:
-                        line = str(diary_id) + "\t" + face["feature"] + "\n"
+                        line = str(diary_id) + "\t" + face["feature"] + "\t" + after_cover_url + "\n"
                         # print(line)
                         f.write(line)
 
@@ -130,15 +130,17 @@ 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_faiss_index(diary_after_cover_vec_file, faiss_index_path)
+    save_diary_image_info(diary_after_cover_vec_file, face_to_vec_f)
+    save_faiss_index(diary_after_cover_vec_file, faiss_index_path)
 
     faiss_index = faiss.read_index(faiss_index_path)
-
-    img_url = "https://pic.igengmei.com/2020/07/03/1437/1b9975bb0b81-w"
-
-    res = get_similar_diary_ids_by_url(img_url, faiss_index, face_to_vec_f, limit=0.18232107)
-    print(res)
+    imgs = [
+        "https://pic.igengmei.com/2020/07/03/1437/1b9975bb0b81-w", "https://pic.igengmei.com/2020/07/01/1812/ca64827a83da-w",
+        "https://pic.igengmei.com/2020/07/04/1711/24f4131a9b1e-w", "https://pic.igengmei.com/2020/07/04/1507/e17a995be219-w"
+    ]
+    for img_url in imgs:
+        res = get_similar_diary_ids_by_url(img_url, faiss_index, face_to_vec_f, limit=0.18232107)
+        print(res)
 
     # a = [
     #     -0.08361373096704483, 0.06760436296463013, 0.10752949863672256, -0.020746365189552307, -0.07035162299871445,