Commit f8c636df authored by 赵威's avatar 赵威

update for loop

parent 6f2dbf44
...@@ -55,24 +55,21 @@ def save_diary_image_info(save_file, face_to_vec_f): ...@@ -55,24 +55,21 @@ def save_diary_image_info(save_file, face_to_vec_f):
} }
count = 0 count = 0
with open(save_file, "w") as f: with open(save_file, "w") as f:
step = 100 res_dict = es_query("diary", q, 0, 500000)
for i in range(0, 500000, step): for item in res_dict["hits"]["hits"]:
res_dict = es_query("diary", q, 0, step) diary_id = item["_source"]["id"]
# before_cover_url = item["_source"]["before_cover_url"] + "-w"
for item in res_dict["hits"]["hits"]: after_cover_url = item["_source"]["after_cover_url"] + "-w"
diary_id = item["_source"]["id"] img = url_to_ndarray(after_cover_url)
# before_cover_url = item["_source"]["before_cover_url"] + "-w" if img.any():
after_cover_url = item["_source"]["after_cover_url"] + "-w" count += 1
img = url_to_ndarray(after_cover_url) print("count: " + str(count))
if img.any(): faces = face_to_vec_f(img)
count += 1 for face in faces:
print("count: " + str(count)) line = str(diary_id) + "\t" + face["feature"] + "\n"
faces = face_to_vec_f(img) # print(line)
for face in faces: f.write(line)
line = str(diary_id) + "\t" + face["feature"] + "\n"
# print(line)
f.write(line)
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