File size: 1,203 Bytes
2cd560a |
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 |
import os
import os.path as osp
import sys
class Config:
## model setting
resnet_type = 50
hand_resnet_type = 50
face_resnet_type = 18
model_type = 'OSX'
# osx_model_settiing
upscale = 4
with_wrist = False
hand_pos_joint_num = 20
num_task_token = 24
feat_dim = 1024
encoder_config_file = 'grounded-sam-osx/transformer_utils/configs/osx/encoder/body_encoder_large.py'
## input, output
input_img_shape = (512, 384)
input_body_shape = (256, 192)
output_hm_shape = (16, 16, 12)
# output_hm_shape = (8, 8, 6)
input_hand_shape = (256, 256)
# output_hand_hm_shape = (8, 8, 8)
output_hand_hm_shape = (16, 16, 16)
input_face_shape = (192, 192)
focal = (5000, 5000) # virtual focal lengths
princpt = (input_body_shape[1]/2, input_body_shape[0]/2) # virtual principal point position
body_3d_size = 2
hand_3d_size = 0.3
face_3d_size = 0.3
camera_3d_size = 2.5
## human models
flame_shape_params = 100
flame_expression_params = 50
## directory
cur_dir = osp.dirname(os.path.abspath(__file__))
human_model_path = osp.join(cur_dir, 'utils', 'human_model_files')
cfg = Config()
|