Skip to content

S
strategy_embedding
Commit d730af25 authored by 赵威's avatar 赵威
Browse files
Options

exception handle

parent c720a4ae
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
src/diary_cover_similarity.py
View file @ d730af25
...@@ -58,6 +58,7 @@ def save_diary_image_info(save_file, face_to_vec_f): ...@@ -58,6 +58,7 @@ def save_diary_image_info(save_file, face_to_vec_f):
count = 0 count = 0
results = es_scan("diary", q) results = es_scan("diary", q)
with open(save_file, "w") as f: with open(save_file, "w") as f:
try:
for item in results: for item in results:
diary_id = item["_id"] diary_id = item["_id"]
# before_cover_url = item["_source"]["before_cover_url"] + "-w" # before_cover_url = item["_source"]["before_cover_url"] + "-w"
...@@ -72,6 +73,8 @@ def save_diary_image_info(save_file, face_to_vec_f): ...@@ -72,6 +73,8 @@ def save_diary_image_info(save_file, face_to_vec_f):
line = str(diary_id) + "\t" + face["feature"] + "\n" line = str(diary_id) + "\t" + face["feature"] + "\n"
# print(line) # print(line)
f.write(line) f.write(line)
except Exception as e:
print(e)
def save_faiss_index(load_file, save_path): def save_faiss_index(load_file, save_path):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment