Skip to content

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

save result

parent 0c3c5865
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
src/diary_cover_similarity.py
View file @ cf4db7d7
...@@ -148,13 +148,14 @@ def save_diary_similarity(load_file, index_path, face_to_vec_f): ...@@ -148,13 +148,14 @@ def save_diary_similarity(load_file, index_path, face_to_vec_f):
print("{} {}".format(count, id)) print("{} {}".format(count, id))
tup_res = get_similar_diary_ids_by_face_features(feature, index_path, face_to_vec_f) tup_res = get_similar_diary_ids_by_face_features(feature, index_path, face_to_vec_f)
if tup_res: if tup_res:
res_dict[id] = tup_res res_dict[id] = json.dumps(tup_res)
print("done: " + str(len(res_dict))) print("done: " + str(len(res_dict)))
key = random.choice(list(res_dict.keys())) key = random.choice(list(res_dict.keys()))
print(res_dict[key]) print(res_dict[key])
redis_key = "doris:diary:face_similary" redis_key = "doris:diary:face_similary"
redis_client3.hmset(redis_key, json.dumps(res_dict)) redis_client3.hmset(redis_key, res_dict)
def main(): def 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