Function set_model_info

pointersect/meta_script/submit_pbnr_v2.py:192–223

(
        config_dict: T.Dict[str, T.Any],
        dim: int = 512,
        dropout: float = 0.1,
        num_layers: int = 4,
        use_rgb_as_input: bool = True,
        use_dpsuv_as_input: bool = True,
        use_zdir_as_input: bool = True,
        num_heads: int = 4,
        dim_input_layers: T.List[int] = None,  # dimension of the linear layers (nLayer-1)
        use_vdir_as_input: bool = False,  # if true, use camera viewing direction (1 vector, 3 dim) as input
        use_rgb_indicator: bool = False,  # whether to add a binary indicator saying input has valid rgb
        use_feature_indicator: bool = False,  # whether to add a binary indicator saying input has valid feature
        estimate_surface_normal_weights: bool = False,
)

def set_model_info(
        config_dict: T.Dict[str, T.Any],
        dim: int = 512,
        dropout: float = 0.1,
        num_layers: int = 4,
        use_rgb_as_input: bool = True,
        use_dpsuv_as_input: bool = True,
        use_zdir_as_input: bool = True,
        num_heads: int = 4,
        dim_input_layers: T.List[int] = None,  # dimension of the linear layers (nLayer-1)
        use_vdir_as_input: bool = False,  # if true, use camera viewing direction (1 vector, 3 dim) as input
        use_rgb_indicator: bool = False,  # whether to add a binary indicator saying input has valid rgb
        use_feature_indicator: bool = False,  # whether to add a binary indicator saying input has valid feature
        estimate_surface_normal_weights: bool = False,
) -> T.Dict[str, T.Any]:
    config_dict['model_info']['dim_feature'] = dim
    config_dict['model_info']['dim_mlp'] = dim
    config_dict['model_info']['dropout'] = dropout
    config_dict['model_info']['num_layers'] = num_layers
    config_dict['model_info']['use_rgb_as_input'] = use_rgb_as_input
    config_dict['model_info']['use_dpsuv_as_input'] = use_dpsuv_as_input
    config_dict['model_info']['use_zdir_as_input'] = use_zdir_as_input
    config_dict['model_info']['num_heads'] = num_heads

    config_dict['model_info']['dim_input_layers'] = dim_input_layers
    config_dict['model_info']['use_vdir_as_input'] = use_vdir_as_input
    config_dict['model_info']['use_rgb_indicator'] = use_rgb_indicator
    config_dict['model_info']['use_feature_indicator'] = use_feature_indicator

    config_dict['model_info']['estimate_surface_normal_weights'] = estimate_surface_normal_weights

    return config_dict


def set_process_info(