File size: 399 Bytes
ed697ed |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
dataset_paths = {
'celeba_test': '',
'ffhq': '',
}
model_paths = {
'pretrained_psp_encoder': 'pretrained_models/psp_ffhq_encode.pt',
'ir_se50': 'pretrained_models/model_ir_se50.pth',
'stylegan_ffhq': 'pretrained_models/stylegan2-ffhq-config-f.pt',
'shape_predictor': 'shape_predictor_68_face_landmarks.dat',
'age_predictor': 'pretrained_models/dex_age_classifier.pth'
}
|