test_encodings_path = "/home/taylan/gitFolder/stream-service/model/rf_test.pkl"
train_encoding_dict = None
test_encoding_dict = None
if os.path.exists(train_encodings_path) and os.path.exists(
test_encodings_path):
train_encoding_dict = load_encodings(train_encodings_path)
test_encoding_dict = load_encodings(test_encodings_path)
else:
face_recog = FaceRecognizer()
print("Face recognizer initialized")
start = time.time()
train_full_paths, train_base_paths = face_recog.load_image_folder(
train_folder)
train_encoding_dict = face_recog.generate_encodings(
train_full_paths, train_base_paths)
save_encodings(train_encoding_dict, train_encodings_path)
end = time.time()
print("Generated encodings for train set. It took: ", end - start)
print("Size of training images: ", len(train_full_paths))
start = time.time()
test_full_paths, test_base_paths = face_recog.load_image_folder(
test_folder)
test_encoding_dict = face_recog.generate_encodings(
test_full_paths, test_base_paths)
save_encodings(test_encoding_dict, test_encodings_path)
end = time.time()
print("Generated encodings for test set. It took: ", end - start)
print("Size of test images: ", len(test_full_paths))
print("Saved generated encodings")
