Skip to content

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

get result

parent f48e1b8f
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 3 deletions
src/diary_cover_similarity.py
View file @ 3bd1ac93
import json import json
import os import os
import random
import time import time
import dlib import dlib
...@@ -192,6 +193,7 @@ def main(): ...@@ -192,6 +193,7 @@ def main():
# res = get_similar_diary_ids_by_face_features(a, faiss_index, face_to_vec_f) # res = get_similar_diary_ids_by_face_features(a, faiss_index, face_to_vec_f)
# print(res) # print(res)
res_dict = {}
with open(diary_after_cover_vec_file) as f: with open(diary_after_cover_vec_file) as f:
lines = f.readlines() lines = f.readlines()
print("lines: " + str(len(lines))) print("lines: " + str(len(lines)))
...@@ -202,9 +204,12 @@ def main(): ...@@ -202,9 +204,12 @@ def main():
id = tmp[0] id = tmp[0]
feature = np.array(json.loads(tmp[1])) feature = np.array(json.loads(tmp[1]))
print("{} {}".format(count, id)) print("{} {}".format(count, id))
res = get_similar_diary_ids_by_face_features(feature, faiss_index, face_to_vec_f) tup_res = get_similar_diary_ids_by_face_features(feature, faiss_index, face_to_vec_f)
print(res) if tup_res:
print("=========================") 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__": 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