get data

5c3a43c8 · 赵威 · 08f2d679 · 5c3a43c8
Commit 5c3a43c8 authored Sep 14, 2020 by 赵威
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

diary_cover_similarity.py src/diary_cover_similarity.py +4 -1

No files found.
--- a/src/diary_cover_similarity.py
+++ b/src/diary_cover_similarity.py
@@ -201,7 +201,10 @@ def main():
            tmp = line.split("\t")
            id = tmp[0]
            feature = np.array(json.loads(tmp[1]))
-            print("{} {} {}".format(count, id, feature))
+            print("{} {}".format(count, id))
+            res = get_similar_diary_ids_by_face_features(a, faiss_index, face_to_vec_f)
+            print(res)
+            print("=========================")
 if __name__ == "__main__":