-
Notifications
You must be signed in to change notification settings - Fork 63
/
constants.py
79 lines (66 loc) · 3.32 KB
/
constants.py
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
INDICES = {'SHAPE': (0, 100),
'EXP': (100, 150),
'POSE': (150, 156),
'TRANS': (156, 159),
'JAW_ROT': (153, 156),
'GLOBAL_ROT': (150, 153),
'ROT_JAW_TRANS': (150, 159),
'ALL': (0, 159)}
DECA_IDX = {'cam': (156, 159),
'tex': (159, 209),
'lit': (209, 236)}
def get_idx_list(atrb_name):
'''atrb_name: tuple of name string or single name string'''
if atrb_name in INDICES:
return list(range(*INDICES[atrb_name]))
else:
indx_list = []
for cmp_name in atrb_name:
indx_list += list(range(*INDICES[cmp_name]))
return indx_list
############################################# path constants ###########################################################
############################################# Modify according to your needs ###########################################
resources_root = '/is/cluster/scratch/pghosh/GIF_resources'
input_root_dir = f'{resources_root}/input_files'
# DECA and FLAME Resource files
deca_inferred_root = f'{input_root_dir}/DECA_inferred'
deca_data_path = f'{deca_inferred_root}/data/'
flame_resource_path = f'{input_root_dir}/flame_resource/'
flame_texture_space_dat_file = f'{flame_resource_path}texture_data_256.npy'
# Geenral parameter and image and artifact generation paths
output_root = f'{resources_root}/output_files/'
save_dir_voca_vid = f'{output_root}voca_video'
voca_flame_seq_file = f'{input_root_dir}sentence36.npz'
ffhq_images_root_dir = '/raid/data/pghosh/face_gan_data/FFHQ/images1024x1024/'
# all_flame_params_file = f'{deca_inferred_root}/deca_flame_params_camera_corrected.npy'
all_flame_params_file = f'{deca_inferred_root}/flame_params_public_texture_model.npy'
true_iamge_lmdb_path = f'{input_root_dir}/FFHQ/multiscale.lmdb'
rendered_flame_root = f'{deca_inferred_root}/deca_rendered_with_public_texture.lmdb'
true_img_stats_dir = f'{input_root_dir}/FFHQ/ffhq_fid_stats/'
face_region_mask_file = f'{flame_resource_path}texture_map_256X256_face_only_mask.png'
flm_3_sigmaparams_dir = f'{input_root_dir}/GIF_teaser/data/'
list_deca_failed_iamges = f'{deca_inferred_root}/b_box_stats.npz'
# Placing random images and their rendering side by side
generated_random_image_root_dir = output_root
random_imagesdestination_dir = output_root
# reinference MSE error for evaluation paths
# Amazon Mechanical turk related constants
amt_bucket_base_url = 'https://flameparameterassociation.s3-eu-west-1.amazonaws.com'
five_pt_likert_scale_result_csv_path = f'{input_root_dir}/mturk_results/textured_rend_flm_asso_right_likert_scale.csv'
flame_style_vec_association_result_csv_path = f'{input_root_dir}/mturk_results/Result_flm_asso_10k.csv'
flame_config = {
# FLAME
'flame_model_path': f'{flame_resource_path}generic_model.pkl', # acquire it from FLAME project page
'flame_lmk_embedding_path': f'{flame_resource_path}landmark_embedding.npy',
'mesh_file': f'{flame_resource_path}head_template_mesh.obj',
'tex_space_path': f'{flame_resource_path}FLAME_texture.npz', # acquire it from FLAME project page
'camera_params': 3,
'shape_params': 100,
'expression_params': 50,
'pose_params': 6,
'tex_params': 50,
'use_face_contour': True,
'cropped_size': 256,
'batch_size': 1,
'image_size': 256,
}