| 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' | |
| } | |