grounded-sam-osx/config.py

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()