MCPcopy Create free account

hub / github.com/YvanYin/DrivingWorld / functions

Functions280 in github.com/YvanYin/DrivingWorld

↓ 32 callersMethoditems
Yield the keys and values of the dictionary. If class attribute ``lazy`` is False, the value of ``LazyObject`` or ``LazyAttr`` will b
utils/config_utils.py:765
↓ 21 callersMethod__setattr__
(self, name, value)
utils/config_utils.py:2003
↓ 14 callersMethodpop
Pop the value of the key. If class attribute ``lazy`` is True, the LazyObject will be built and returned. Args: key (str)
utils/config_utils.py:707
↓ 12 callersMethod__init__
(self, config: ModelArgs)
modules/tokenizers/vq_model.py:113
↓ 8 callersMethod_hook
(cls, item)
utils/config_utils.py:655
↓ 8 callersFunctionadd_border
(img, border_size = 10, value=[255, 255, 255])
utils/testing_utils.py:69
↓ 7 callersMethod__init__
(self, vocab_size, block_size, n_layer=[12, 6], n_head=8, n_embd=1024, embd_pdrop=0., resid_p
models/ar.py:223
↓ 6 callersMethodfromfile
Build a Config instance from config file. Args: filename (str or Path): Name of config file. use_predefined_variables
utils/config_utils.py:932
↓ 6 callersMethodget
Get the value of the key. If class attribute ``lazy`` is True, the LazyObject will be built and returned. Args: key (str)
utils/config_utils.py:694
↓ 5 callersFunctionNormalize
(in_channels, norm_type='group')
modules/tokenizers/vq_model.py:608
↓ 5 callersFunction_is_builtin_module
Check if a module is a built-in module. Arg: module_name: name of module.
utils/config_utils.py:569
↓ 5 callersMethodbuild_lazy
If class attribute ``lazy`` is False, the LazyObject will be built and returned. Args: value (Any): The value to be built
utils/config_utils.py:740
↓ 5 callersMethodget_yaw_pose_emb
(self, pose_indices, yaw_indices)
models/ar.py:343
↓ 5 callersMethodspatial_temporal_blocks
(self, x)
modules/tokenizers/vq_model.py:49
↓ 4 callersMethoddecode
(self, quant)
modules/tokenizers/vq_model.py:136
↓ 4 callersMethodeval_third
(self, auto_regressive_token_embeddings, yaws=None, poses=None, new_tokens=None, drop_flag=False)
models/ar.py:446
↓ 4 callersMethodget_codebook_entry
(self, indices, shape=None, channel_first=True)
modules/tokenizers/vq_model.py:342
↓ 4 callersFunctionload_parameters
(model, load_ckpt_file)
utils/running.py:94
↓ 4 callersFunctionnonlinearity
(x)
modules/tokenizers/vq_model.py:603
↓ 4 callersMethodupdate
Override this method to make sure the LazyObject will not be built during updating.
utils/config_utils.py:720
↓ 3 callersFunctionapply_rotary_emb
( xq: torch.Tensor, xk: torch.Tensor, freqs_cis: torch.Tensor, )
modules/tokenizers/vq_model.py:578
↓ 3 callersMethodencode
(self, x)
modules/tokenizers/vq_model.py:124
↓ 3 callersFunctionget_fourier_embeds_from_coordinates
Args: embed_dim: int coordinates: a tensor [B x N x C] representing the coordinates of N points in C dimensions Returns:
utils/embeddings.py:56
↓ 3 callersMethodorganize_attn_mask
(self, drop_flg)
models/ar.py:369
↓ 3 callersFunctionposes_to_indices
Tokenize pose (x, y) to indices. Args: poses: (x, y), [b, f, 2] Return: pose dicces: [b, f, 2]
modules/tokenizers/pose_tokenizer.py:3
↓ 3 callersMethodto_dict
Convert all data in the config to a builtin ``dict``. Args: keep_imported (bool): Whether to keep the imported field.
utils/config_utils.py:2208
↓ 3 callersFunctionyaws_to_indices
(yaws, division=512)
modules/tokenizers/pose_tokenizer.py:33
↓ 2 callersMethod_is_lazy_import
(filename: str)
utils/config_utils.py:2157
↓ 2 callersMethod_merge_a_into_b
merge dict ``a`` into dict ``b`` (non-inplace). Values in ``a`` will overwrite ``b``. ``b`` is copied first to avoid in-place modific
utils/config_utils.py:1787
↓ 2 callersFunctionbarrier
()
utils/dist.py:96
↓ 2 callersMethodbuild
Return the attribute of the imported object. Returns: Any: attribute of the imported object.
utils/config_utils.py:130
↓ 2 callersFunctioncompute_axial_cis
(dim, end_x, end_y, theta = 100.0)
utils/rope_2d.py:9
↓ 2 callersFunctioncompute_entropy_loss
(affinity, loss_type="softmax", temperature=0.01)
modules/tokenizers/vq_model.py:648
↓ 2 callersFunctioncreate_mp4_imgs
(args, imgs, video_save_path, border_size = 10, fps=2, name="")
utils/testing_utils.py:79
↓ 2 callersFunctioncreate_test_datasets
(args, split='test')
datasets/create_dataset.py:5
↓ 2 callersMethoddecode
(self, quant)
modules/tokenizers/vq_model.py:81
↓ 2 callersMethoddump
Dump config to file or return config text. Args: file (str or Path, optional): If not specified, then the object is d
utils/config_utils.py:2053
↓ 2 callersMethodencode_to_z
(self, x)
modules/tokenizers/model_tokenizer.py:21
↓ 2 callersMethodeval_first_second
(self, feature, pose_indices, yaw_indices, drop_flag)
models/ar.py:427
↓ 2 callersMethodforward_hacked
(self, z)
modules/tokenizers/vq_model.py:308
↓ 2 callersMethodgenerate_gt_pose_gt_yaw
(self, token, feature, pose, yaw, pose_gt, yaw_gt, vqvae_codebook, sampling_mtd = None, temperature_k = 1.0, t
models/model.py:179
↓ 2 callersMethodmerge_from_dict
Merge list into cfg_dict. Merge the dict parsed by MultipleKVAction into this cfg. Args: options (dict): dict of configs
utils/config_utils.py:2079
↓ 2 callersMethodposes_to_indices
(self, poses)
modules/tokenizers/model_tokenizer.py:78
↓ 2 callersFunctionprecompute_freqs_cis
(dim: int, end: int, theta: float = 10000.0)
modules/tokenizers/vq_model.py:151
↓ 2 callersFunctionrepeat_kv
torch.repeat_interleave(x, dim=2, repeats=n_rep)
modules/tokenizers/vq_model.py:591
↓ 2 callersFunctionset_text
(image, pose)
utils/testing_utils.py:95
↓ 2 callersMethodz_to_clips
(self, indices)
modules/tokenizers/model_tokenizer.py:43
↓ 1 callersMethod__getattr__
(self, name: str)
utils/config_utils.py:1997
↓ 1 callersMethod__getitem__
(self, name)
utils/config_utils.py:2000
↓ 1 callersMethod__init__
(self, args, local_rank=-1)
modules/tokenizers/model_tokenizer.py:9
↓ 1 callersMethod__init__
( self, cfg_dict: dict = None, cfg_text: Optional[str] = None, filename: Optio
utils/config_utils.py:893
↓ 1 callersMethod__loadarray_tum_single
(self, array)
datasets/dataset_nuplan.py:49
↓ 1 callersMethod__setitem__
(self, name, value)
utils/config_utils.py:2008
↓ 1 callersMethod_dict_to_config_dict
Recursively converts ``dict`` to :obj:`ConfigDict`. Args: cfg (dict): Config dict. scope (str, optional): Scope of in
utils/config_utils.py:1647
↓ 1 callersMethod_dict_to_config_dict_lazy
Recursively converts ``dict`` to :obj:`ConfigDict`. The only difference between ``_dict_to_config_dict_lazy`` and ``_dict_to_config_di
utils/config_utils.py:1623
↓ 1 callersMethod_file2dict
Transform file to variables dictionary. Args: filename (str): Name of config file. use_predefined_variables (bool, op
utils/config_utils.py:1324
↓ 1 callersFunction_find_free_port
()
utils/comm.py:5
↓ 1 callersFunction_gather_abs_import_lazyobj
Experimental implementation of gathering absolute import information.
utils/config_utils.py:160
↓ 1 callersMethod_get_base_files
Get the base config file. Args: filename (str): The config file. Raises: TypeError: Name of config file.
utils/config_utils.py:1701
↓ 1 callersMethod_get_base_modules
Get base module name from parsed code. Args: nodes (list): Parsed code of the config file. Returns: list: Na
utils/config_utils.py:1043
↓ 1 callersMethod_get_cfg_path
Get the config path from the current or external package. Args: cfg_path (str): Relative path of config. filename (st
utils/config_utils.py:1745
↓ 1 callersFunction_is_free_port
(port)
utils/comm.py:12
↓ 1 callersMethod_norm
(self, x)
modules/tokenizers/vq_model.py:562
↓ 1 callersMethod_parse_lazy_import
Transform file to variables dictionary. Args: filename (str): Name of config file. Returns: Tuple[dict, dict
utils/config_utils.py:1484
↓ 1 callersMethod_parse_scope
Adds ``_scope_`` to :obj:`ConfigDict` instance, which means a base variable. If the config dict already has the scope, scope will not
utils/config_utils.py:1683
↓ 1 callersMethod_pre_substitute_base_vars
Preceding step for substituting variables in base config with actual value. Args: filename (str): Filename of config.
utils/config_utils.py:1252
↓ 1 callersMethod_substitute_base_vars
Substitute base variables from strings to their actual values. Args: Any : Config dictionary. base_var_dict (dict): A
utils/config_utils.py:1280
↓ 1 callersMethod_substitute_env_variables
Substitute environment variables in config with actual values. Sometimes, we want to change some items in the config with environment
utils/config_utils.py:1191
↓ 1 callersMethod_substitute_predefined_vars
Substitute predefined variables in config with actual values. Sometimes we want some variables in the config to be related to the cur
utils/config_utils.py:1132
↓ 1 callersMethod_validate_py_syntax
Validate syntax of python config. Args: filename (str): Filename of python config file.
utils/config_utils.py:1117
↓ 1 callersFunctionadd_arguments
()
tools/test_long_term_demo.py:23
↓ 1 callersFunctionadd_arguments
()
tools/test_change_road_demo.py:23
↓ 1 callersFunctionadd_arguments
()
video_data_preprocess/create_nuplan_json.py:17
↓ 1 callersFunctionallreduce
(t: torch.Tensor, async_op=False)
utils/dist.py:101
↓ 1 callersMethodaug_seq
(self, imgs)
datasets/dataset_demo.py:27
↓ 1 callersMethodaug_seq
(self, imgs)
datasets/dataset_nuplan.py:56
↓ 1 callersFunctioncheck_file_exist
(filename, msg_tmpl='file "{}" does not exist')
utils/config_utils.py:48
↓ 1 callersMethoddecode_code
(self, code_b, shape=None, channel_first=True)
modules/tokenizers/vq_model.py:141
↓ 1 callersMethoddecode_code_temporal
(self, code_b, shape=None, channel_first=True)
modules/tokenizers/vq_model.py:93
↓ 1 callersMethoddecode_code_temporal_all
(self, code_b, shape=None, channel_first=True)
modules/tokenizers/vq_model.py:100
↓ 1 callersMethoddownsample_sequences
(self, img_ts, poses)
datasets/dataset_nuplan.py:69
↓ 1 callersMethodencode
(self, x)
modules/tokenizers/vq_model.py:65
↓ 1 callersMethodencode_hacked
(self, x)
modules/tokenizers/vq_model.py:130
↓ 1 callersFunctionget_local_rank
()
utils/dist.py:56
↓ 1 callersFunctionget_meta_data
(poses)
datasets/datasets_utils.py:35
↓ 1 callersFunctionget_rank
()
utils/dist.py:52
↓ 1 callersFunctionget_timestamp
()
utils/utils.py:6
↓ 1 callersMethodgetimg
(self, index)
datasets/dataset_demo.py:40
↓ 1 callersMethodgetimg
(self, index)
datasets/dataset_nuplan.py:77
↓ 1 callersFunctioninit_environment
(args)
tools/test_long_term_demo.py:37
↓ 1 callersFunctioninit_environment
(args)
tools/test_change_road_demo.py:37
↓ 1 callersFunctioninit_t_xy
(end_x, end_y)
utils/rope_2d.py:3
↓ 1 callersFunctioninitialized
()
utils/dist.py:16
↓ 1 callersFunctionis_local_master
()
utils/dist.py:82
↓ 1 callersFunctionis_master
()
utils/dist.py:78
↓ 1 callersFunctionis_visualizer
()
utils/dist.py:86
↓ 1 callersMethodload_pose
(self, pose_path, front_cam_list)
datasets/dataset_nuplan.py:34
↓ 1 callersFunctionload_single_log_db_data
(log_db_name="2021.05.12.22.28.35_veh-35_00620_01164", log_db=None)
video_data_preprocess/create_nuplan_json.py:25
↓ 1 callersFunctionloop_over_db_files
( rank=0, workers=1, split_db_list=None, ego_save_dir='', seq_save_dir='', nuplandb_wrappe
video_data_preprocess/create_nuplan_json.py:120
↓ 1 callersFunctionmain
(args)
tools/test_long_term_demo.py:112
next →1–100 of 280, ranked by callers