Commit 3bd1ac93 authored by 赵威's avatar 赵威

get result

parent f48e1b8f
import json
import os
import random
import time
import dlib
......@@ -192,6 +193,7 @@ def main():
# res = get_similar_diary_ids_by_face_features(a, faiss_index, face_to_vec_f)
# print(res)
res_dict = {}
with open(diary_after_cover_vec_file) as f:
lines = f.readlines()
print("lines: " + str(len(lines)))
......@@ -202,9 +204,12 @@ def main():
id = tmp[0]
feature = np.array(json.loads(tmp[1]))
print("{} {}".format(count, id))
res = get_similar_diary_ids_by_face_features(feature, faiss_index, face_to_vec_f)
print(res)
print("=========================")
tup_res = get_similar_diary_ids_by_face_features(feature, faiss_index, face_to_vec_f)
if tup_res:
res_dict[id] = tup_res
print("done: " + str(len(res_dict)))
key = random.choice(list(res_dict.keys()))
print(res_dict[key])
if __name__ == "__main__":
......
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