get resurt

src/diary_cover_similarity.py

@@ -18,6 +18,14 @@ def save_diary_image_info(save_file, face_to_vec_f):
                     "term": {
                         "is_online": True
                     }
+                }, {
+                    "term": {
+                        "has_cover": True
+                    }
+                }, {
+                    "term": {
+                        "is_sink": False
+                    }
                 }, {
                     "term": {
                         "has_before_cover": True
@@ -26,6 +34,14 @@ def save_diary_image_info(save_file, face_to_vec_f):
                     "term": {
                         "has_after_cover": True
                     }
+                }, {
+                    "terms": {
+                        "content_level": [6, 5, 4, 3.5, 3]
+                    }
+                }, {
+                    "term": {
+                        "content_simi_bol_show": 0
+                    }
                 }, {
                     "exists": {
                         "field": "before_cover_url"
@@ -122,11 +138,12 @@ def main():
     ]
     feature = np.array(a).astype("float32")
     index = faiss.read_index(faiss_index_path)
-    D, I  = index.search(np.array[feature], 5)
-    print(D)
-    print(I)
-    ids = list(I.flat)
-    print(ids)
+    D, I = index.search(np.array([feature]), 5)
+    ids = I.flat
+    scores = D.flat
+    res = list(zip(ids, scores))
+    res.sort(key = lambda x: x[1], reverse=True)
+    print(res)
 
 
 if __name__ == "__main__":