Spaces:
Runtime error
Runtime error
year = "2010" | |
## Pretrained models paths | |
e4e = "./pretrained_models/e4e_ffhq_encode.pt" | |
stylegan2_ada_ffhq = f"../pretrained_models/{year}.pkl" | |
style_clip_pretrained_mappers = "" | |
ir_se50 = "/share/phoenix/nfs04/S7/wikitime_models/model_ir_se50.pth" | |
dlib = "./pretrained_models/align.dat" | |
deeplab = "/share/phoenix/nfs04/S7/wikitime_models/deeplab_model/deeplab_model.pth" | |
## Dirs for output files | |
checkpoints_dir = "./checkpoints" | |
embedding_base_dir = "./embeddings" | |
styleclip_output_dir = "./StyleCLIP_results" | |
experiments_output_dir = "./output" | |
## Input info | |
### Input dir, where the images reside | |
input_data_path = ( | |
f"/share/phoenix/nfs04/S7/emc348/WikiFaces/datasets/new_crops/test/{year}" | |
) | |
input_data_id = f"{year}" | |
## Keywords | |
pti_results_keyword = "PTI" | |
e4e_results_keyword = "e4e" | |
sg2_results_keyword = "SG2" | |
sg2_plus_results_keyword = "SG2_plus" | |
multi_id_model_type = "multi_id_fine" | |
## Edit directions | |
interfacegan_age = "editings/interfacegan_directions/age.pt" | |
interfacegan_smile = "editings/interfacegan_directions/smile.pt" | |
interfacegan_rotation = "editings/interfacegan_directions/rotation.pt" | |
ffhq_pca = "editings/ganspace_pca/ffhq_pca.pt" | |