Spaces:
Paused
Paused
File size: 516 Bytes
3de0e37 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import sys
sys.path.append('.')
from external_code.landmarks_util import detect_landmarks, predictor_dict
from global_value_utils import GLOBAL_DATA_ROOT, DATASET_NAME
data_name = DATASET_NAME
root_dir = GLOBAL_DATA_ROOT
for landmark_num in [81, 68]:
print('detect %d landmarks' % landmark_num)
detect_landmarks(root_dir, data_name,
landmark_output_file_path=root_dir + '/landmark%d.pkl' % landmark_num,
output_dir=None, predictor=predictor_dict[landmark_num])
|