mirror of https://github.com/XingangPan/DragGAN
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
726 B
Python
25 lines
726 B
Python
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'
|