Spaces:
Runtime error
Runtime error
File size: 1,174 Bytes
2e34814 |
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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
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"
|