Spaces:
Runtime error
Runtime error
File size: 718 Bytes
a0bcaae bb0f5a9 a0bcaae bb0f5a9 a0bcaae bb0f5a9 a0bcaae bb0f5a9 a0bcaae bb0f5a9 a0bcaae bb0f5a9 a0bcaae |
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 |
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'
|