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)