Skip to content

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

add printer

parent cca04a81
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
src/diary_cover_similarity.py
View file @ 5545458c
...@@ -78,10 +78,13 @@ def save_faiss_index(load_file, save_path): ...@@ -78,10 +78,13 @@ def save_faiss_index(load_file, save_path):
features = [] features = []
lines = f.readlines() lines = f.readlines()
print("lines: " + str(len(lines))) print("lines: " + str(len(lines)))
count = 0
for line in lines: for line in lines:
count += 1
tmp = line.split("\t") tmp = line.split("\t")
ids.append(tmp[0]) ids.append(tmp[0])
features.append(np.array(json.loads(tmp[1]))) features.append(np.array(json.loads(tmp[1])))
print(count + " " + tmp[0])
ids_np = np.array(ids).astype("int") ids_np = np.array(ids).astype("int")
features_np = np.array(features).astype("float32") features_np = np.array(features).astype("float32")
...@@ -107,7 +110,7 @@ def main(): ...@@ -107,7 +110,7 @@ def main():
shape_predictor = dlib.shape_predictor(shape_model_path) shape_predictor = dlib.shape_predictor(shape_model_path)
face_to_vec_f = lambda img: face_to_vec(img, face_rec, face_detector, shape_predictor) face_to_vec_f = lambda img: face_to_vec(img, face_rec, face_detector, shape_predictor)
save_diary_image_info(diary_after_cover_vec_file, face_to_vec_f) # save_diary_image_info(diary_after_cover_vec_file, face_to_vec_f)
save_faiss_index(diary_after_cover_vec_file, faiss_index_path) save_faiss_index(diary_after_cover_vec_file, faiss_index_path)
a = [ a = [
......
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