def test_remove_face(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} fm = FaceModels(params) fm.delete_models() self.test_add_face() training_set_path = os.path.join( fm._data_dir_path, c.TRAINING_SET_DIR) aligned_faces_path = os.path.join( training_set_path, c.ALIGNED_FACES_DIR) label = 3812 subject_path = os.path.join(aligned_faces_path, str(label)) im_name = '' for im in os.listdir(subject_path): im_name = im ok = fm.remove_face(label, im_name) self.assertTrue(ok)
def test_remove_face_not_removed(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} fm = FaceModels(params) fm.delete_models() self.test_add_face() label = 1812 im_name = 'Dummy image' ok = fm.remove_face(label, im_name) self.assertFalse(ok)