Spaces:
Runtime error
Runtime error
| dataset_paths = { | |
| 'cars_train': '', | |
| 'cars_test': '', | |
| 'celeba_train': '', | |
| 'celeba_test': '', | |
| 'celeba_test_w_inv': '', | |
| 'celeba_test_w_latents': '', | |
| 'ffhq': '', | |
| 'ffhq_w_inv': '', | |
| 'ffhq_w_latents': '', | |
| 'afhq_wild_train': '', | |
| 'afhq_wild_test': '', | |
| } | |
| model_paths = { | |
| # models for backbones and losses | |
| 'ir_se50': 'pretrained_models/model_ir_se50.pth', | |
| 'resnet34': 'pretrained_models/resnet34-333f7ec4.pth', | |
| 'moco': 'pretrained_models/moco_v2_800ep_pretrain.pt', | |
| # stylegan2 generators | |
| 'stylegan_ffhq': 'pretrained_models/stylegan2-ffhq-config-f.pt', | |
| 'stylegan_cars': 'pretrained_models/stylegan2-car-config-f.pt', | |
| 'stylegan_ada_wild': 'pretrained_models/afhqwild.pt', | |
| # model for face alignment | |
| 'shape_predictor': 'pretrained_models/shape_predictor_68_face_landmarks.dat', | |
| # models for ID similarity computation | |
| 'curricular_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', | |
| # WEncoders for training on various domains | |
| 'faces_w_encoder': 'pretrained_models/faces_w_encoder.pt', | |
| 'cars_w_encoder': 'pretrained_models/cars_w_encoder.pt', | |
| 'afhq_wild_w_encoder': 'pretrained_models/afhq_wild_w_encoder.pt', | |
| # models for domain adaptation | |
| 'restyle_e4e_ffhq': 'pretrained_models/restyle_e4e_ffhq_encode.pt', | |
| 'stylegan_pixar': 'pretrained_models/pixar.pt', | |
| 'stylegan_toonify': 'pretrained_models/ffhq_cartoon_blended.pt', | |
| 'stylegan_sketch': 'pretrained_models/sketch.pt', | |
| 'stylegan_disney': 'pretrained_models/disney_princess.pt' | |
| } | |
| edit_paths = { | |
| 'age': 'editing/interfacegan_directions/age.pt', | |
| 'smile': 'editing/interfacegan_directions/smile.pt', | |
| 'pose': 'editing/interfacegan_directions/pose.pt', | |
| 'cars': 'editing/ganspace_directions/cars_pca.pt', | |
| 'styleclip': { | |
| 'delta_i_c': 'editing/styleclip/global_directions/ffhq/fs3.npy', | |
| 's_statistics': 'editing/styleclip/global_directions/ffhq/S_mean_std', | |
| 'templates': 'editing/styleclip/global_directions/templates.txt' | |
| } | |
| } |