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