def test_recognize_model_internal_models(self):
base_path = os.path.join('..', 'test_files', 'face_models')
face_rec_data = os.path.abspath(
os.path.join(base_path, 'face_rec_data'))
params = {c.GLOBAL_FACE_MODELS_MIN_DIFF_KEY: -1,
c.GLOBAL_FACE_REC_DATA_DIR_PATH_KEY: face_rec_data}
self.test_add_face()
fm = FaceModels(params)
fm.load_enabled_models()
model_id = 0
im_list = [os.path.join(base_path, '0000000_aligned.png'),
os.path.join(base_path, '0000005_aligned.png')]
model_path_0 = fm.create_model_from_image_list(im_list, model_id)
query_model = cv2.createLBPHFaceRecognizer()
query_model.load(model_path_0)
rec_results = fm.recognize_model(query_model)
print('rec_results', rec_results)
rec_result = rec_results[0]
self.assertEqual(rec_result[c.ASSIGNED_TAG_KEY], 3812)
def test_recognize_model_external_models(self):
base_path = os.path.join('..', 'test_files', 'face_models')
face_rec_data = os.path.abspath(
os.path.join(base_path, 'face_rec_data'))
params = {c.GLOBAL_FACE_MODELS_MIN_DIFF_KEY: -1,
c.GLOBAL_FACE_REC_DATA_DIR_PATH_KEY: face_rec_data}
model_path_0 = os.path.join(face_rec_data, c.FACE_MODELS_DIR, '0')
model_path_1 = os.path.join(face_rec_data, c.FACE_MODELS_DIR, '1')
model_path_2 = os.path.join(face_rec_data, c.FACE_MODELS_DIR, '2')
model_path_3 = os.path.join(face_rec_data, c.FACE_MODELS_DIR, '3')
model_0 = {c.MODEL_ID_KEY: 0,
c.MODEL_FILE_KEY: model_path_0,
c.TAG_KEY: 'Mameli_Giacomo'}
model_1 = {c.MODEL_ID_KEY: 1,
c.MODEL_FILE_KEY: model_path_1,
c.TAG_KEY: 'Fadda_Paolo'}
model_2 = {c.MODEL_ID_KEY: 2,
c.MODEL_FILE_KEY: model_path_2,
c.TAG_KEY: 'Giannotta_Michele'}
model_3 = {c.MODEL_ID_KEY: 3,
c.MODEL_FILE_KEY: model_path_3,
c.TAG_KEY: 'Leoni_Mario'}
fm = FaceModels(params, [model_0, model_1, model_2, model_3])
fm.delete_models()
self.test_create_models_from_image_list()
fm.load_models()
model_id = 0
im_list = [os.path.join(base_path, '0000000_aligned.png'),
os.path.join(base_path, '0000005_aligned.png')]
model_path_0 = fm.create_model_from_image_list(im_list, model_id)
query_model = cv2.createLBPHFaceRecognizer()
query_model.load(model_path_0)
rec_results = fm.recognize_model(query_model)
rec_result = rec_results[0]
self.assertEqual(rec_result[c.ASSIGNED_TAG_KEY], 0)
