add limit

src/diary_cover_similarity.py

@@ -95,7 +95,7 @@ def save_faiss_index(load_file, save_path):
         faiss.write_index(index2, save_path)
 
 
-def get_similar_diary_ids_by_url(url, index, face_to_vec_f, limit=0):
+def get_similar_diary_ids_by_url(url, index, face_to_vec_f, limit=0.1):
     img = url_to_ndarray(url)
     if img.any():
         faces = face_to_vec_f(img)
@@ -104,7 +104,11 @@ def get_similar_diary_ids_by_url(url, index, face_to_vec_f, limit=0):
             _scores, _ids = index.search(np.array([face_feature]), 10)
             ids = _ids.flat
             scores = _scores.flat
-            res = list(zip(ids, scores))
+            tmp = list(zip(ids, scores))
+            res = []
+            for (id, score) in tmp:
+                if score >= limit:
+                    res.append((id, score))
             res.sort(key=lambda x: x[1], reverse=True)
             return res
     else:
@@ -133,7 +137,7 @@ def main():
 
     img_url = "https://pic.igengmei.com/2020/07/03/1437/1b9975bb0b81-w"
 
-    res = get_similar_diary_ids_by_url(img_url, faiss_index, face_to_vec_f)
+    res = get_similar_diary_ids_by_url(img_url, faiss_index, face_to_vec_f, limit=0.18232107)
     print(res)
 
     # a = [