File size: 1,179 Bytes
87c126b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
model_paths = {
'ir_se50': 'pretrained_models/model_ir_se50.pth',
'resnet34': 'pretrained_models/resnet34-333f7ec4.pth',
'stylegan_ffhq': 'pretrained_models/stylegan2-ffhq-config-f.pt',
'stylegan_cars': 'pretrained_models/stylegan2-car-config-f.pt',
'stylegan_church': 'pretrained_models/stylegan2-church-config-f.pt',
'stylegan_horse': 'pretrained_models/stylegan2-horse-config-f.pt',
'stylegan_ada_wild': 'pretrained_models/afhqwild.pt',
'stylegan_toonify': 'pretrained_models/ffhq_cartoon_blended.pt',
'shape_predictor':
'pretrained_models/shape_predictor_68_face_landmarks.dat',
'circular_face': 'pretrained_models/CurricularFace_Backbone.pth',
'mtcnn_pnet': 'pretrained_models/mtcnn/pnet.npy',
'mtcnn_rnet': 'pretrained_models/mtcnn/rnet.npy',
'mtcnn_onet': 'pretrained_models/mtcnn/onet.npy',
'moco': 'pretrained_models/moco_v2_800ep_pretrain.pt'
}
project_basedir = '/mnt/lustre/yslan/Repo/Research/SIGA22/BaseModels/StyleSDF'
for k, v in model_paths.items():
model_paths[k] = f'{project_basedir}/project/utils/misc/' + model_paths[k]
model_paths['ir_se50_hwc'] = '/home/yslan/datasets/model_ir_se50.pth'
|