add printer

src/diary_cover_similarity.py

@@ -78,10 +78,13 @@ def save_faiss_index(load_file, save_path):
         features = []
         lines = f.readlines()
         print("lines: " + str(len(lines)))
+        count = 0
         for line in lines:
+            count += 1
             tmp = line.split("\t")
             ids.append(tmp[0])
             features.append(np.array(json.loads(tmp[1])))
+            print(count + " " + tmp[0])
 
         ids_np = np.array(ids).astype("int")
         features_np = np.array(features).astype("float32")
@@ -107,7 +110,7 @@ def main():
     shape_predictor = dlib.shape_predictor(shape_model_path)
     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)
 
     a = [