def test_disable_faces(self):
self.test_add_face()
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}
fm = FaceModels(params)
labels = fm.get_labels()
print('labels', labels)
rel_im_tuples = []
for label in labels:
images = fm.get_images_for_label(label)
print('images', images)
for image in images:
rel_im_tuple = (label, image)
rel_im_tuples.append(rel_im_tuple)
fm.disable_faces(rel_im_tuples)
# Assert that file with enabled models does not exist
db_file_name = os.path.join(fm._data_dir_path, c.ENABLED_FACE_MODELS_FILE)
self.assertFalse(os.path.exists(db_file_name))
def test_enable_faces(self):
self.test_disable_faces()
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}
fm = FaceModels(params)
label = 3812
rel_im_tuples = []
images = fm.get_images_for_label(label)
for image in images:
rel_im_tuple = (label, image)
rel_im_tuples.append(rel_im_tuple)
fm.enable_faces(rel_im_tuples)
# Get labels from all models
labels = fm._models.getMat("labels")
self.assertEqual(len(labels), 6)
# Get labels from enabled models
labels = fm._en_models.getMat("labels")
self.assertEqual(len(labels), 2)
def test_change_label_to_face(self):
self.test_add_face()
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_REC_DATA_DIR_PATH_KEY: face_rec_data}
fm = FaceModels(params)
old_label = 3812
images = fm.get_images_for_label(old_label)
im_name = images[0]
new_label = 3813
fm.change_label_to_face(im_name, old_label, new_label)
images_new_label = fm.get_images_for_label(new_label)
self.assertIn(im_name, images_new_label)
