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)