import time
DB = Database()
requester = AddFace()
images_folder = Config.dataset_folder
images_subfolders = glob.glob(os.path.join(images_folder, "*"))
skip = True
for index, subfolder in enumerate(images_subfolders[0:800]):
print(subfolder)
if subfolder.split("/")[-1] == "855":
skip = False
if not skip:
images = glob.glob(os.path.join(subfolder, "*"))
image = images[0]
try:
request = requester.add_face(image)
DB.add_FaceId(image, request["persistedFaceId"])
print("{}./Finish adding {} with id {}".format(
index, image, request["persistedFaceId"]))
except Exception as e:
print(e)
time.sleep(60)
try:
request = requester.add_face(image)
DB.add_FaceId(image, request["persistedFaceId"])
print("{}./ Finish adding {} with id {}".format(
index, image, request["persistedFaceId"]))
except:
continue
# image_path = "/mnt/sda2/face_dataset/VN-celeb/1017/0.png"
# response = requester.add_face(image_path)
