Commit 796fe29d authored by 赵威's avatar 赵威

update url

parent fb62b9a0
......@@ -117,7 +117,7 @@ def save_faiss_index(load_file, save_path):
print("faiss index saved")
@bind("strategy_embedding/face_similarity/diary/url")
@bind("strategy_embedding/face_similarity/diary_url")
def get_similar_diary_ids_by_url(url, limit=0.1):
img = url_to_ndarray(url)
if img.any():
......@@ -138,7 +138,7 @@ def get_similar_diary_ids_by_url(url, limit=0.1):
return []
@bind("strategy_embedding/face_similarity/diary/feature")
@bind("strategy_embedding/face_similarity/diary_feature")
def get_similar_diary_ids_by_face_features(feature, limit=0.1):
feature = np.array(feature).astype("float32")
_sources, _ids = FAISS_DIARY_INDEX.search(np.array([feature]), 10)
......
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