Update main.py

main.py

@@ -6,6 +6,7 @@ import numpy as np
 import torch
 import faiss
 import h5py
+import tqdm
 import os
 import random
 from PIL import Image
@@ -14,7 +15,7 @@ import matplotlib as mpl
 
 img_names = []
 with open('list_eval_partition.txt', 'r') as f:
-    for line in f:
+    for line in tqdm(f):
         img_name, dtype = line.rstrip().split(' ')
         img_names.append(img_name)
 
@@ -41,7 +42,7 @@ print('Make index .............')
 index = faiss.IndexFlatL2(512)
 
 hf = h5py.File('face_vecs_full.h5', 'r')
-for key in hf.keys():
+for key in tqdm(hf.keys()):
     vec = np.array(hf.get(key))
     index.add(vec)