| import os | |
| ## Pretrained models paths | |
| e4e = './pti/e4e_w+.pt' | |
| stylegan2_ada_shhq = './pretrained_models/stylegan_human_v2_1024.pkl' | |
| ir_se50 = '' #'./model_ir_se50.pth' | |
| ## Dirs for output files | |
| checkpoints_dir = './outputs/pti/checkpoints/' | |
| embedding_base_dir = './outputs/pti/embeddings' | |
| experiments_output_dir = './outputs/pti/' | |
| ## Input info | |
| ### Input dir, where the images reside | |
| input_data_path = 'aligned_image/' | |
| ### Inversion identifier, used to keeping track of the inversion results. Both the latent code and the generator | |
| input_data_id = 'test' | |
| ## 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' | |