Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/YvanYin/DrivingWorld
/ functions
Functions
280 in github.com/YvanYin/DrivingWorld
⨍
Functions
280
◇
Types & classes
35
Function
broadcast
(t: torch.Tensor, src_rank)
utils/dist.py:153
Method
build
Return imported object. Returns: Any: Imported object
utils/config_utils.py:495
Function
create_gif
(args, imgs, video_save_path, border_size = 10, fps=2)
utils/testing_utils.py:47
Function
create_mp4
(args, imgs, video_save_path, border_size = 10, fps=2)
utils/testing_utils.py:27
Function
create_video
(args, imgs, video_save_path, border_size = 10)
utils/testing_utils.py:6
Function
data_aug_for_seq
(imgs, img_h, img_w)
datasets/datasets_utils.py:14
Method
decode_code
(self, code_b, shape=None, channel_first=True)
modules/tokenizers/vq_model.py:88
Method
diff
(cfg1: Union[str, 'Config'], cfg2: Union[str, 'Config'])
utils/config_utils.py:2127
Function
dist_fmt_vals
(val: float, fmt: Union[str, None] = '%.2f')
utils/dist.py:163
Method
encode_hacked
(self, x)
modules/tokenizers/vq_model.py:73
Method
env_variables
get used environment variables.
utils/config_utils.py:1869
Method
filename
get file name of config.
utils/config_utils.py:1859
Function
finalize
()
utils/dist.py:213
Function
flow2rgb
(flow_map_np)
utils/utils.py:24
Function
for_visualize
(func)
utils/dist.py:201
Method
forward
(self, input)
modules/tokenizers/vq_model.py:107
Method
forward
(self, input)
modules/tokenizers/vq_model.py:146
Method
forward
(self, x)
modules/tokenizers/vq_model.py:191
Method
forward
(self, z)
modules/tokenizers/vq_model.py:242
Method
forward
(self, z)
modules/tokenizers/vq_model.py:275
Method
forward
(self, x)
modules/tokenizers/vq_model.py:377
Method
forward
(self, x)
modules/tokenizers/vq_model.py:405
Method
forward
( self, x: torch.Tensor, freqs_cis: torch.Tensor, mask: Optional[torch.Tensor]
modules/tokenizers/vq_model.py:448
Method
forward
( self, x: torch.Tensor, freqs_cis: torch.Tensor, mask: Optional[torch.Tensor]
modules/tokenizers/vq_model.py:488
Method
forward
(self, x)
modules/tokenizers/vq_model.py:553
Method
forward
(self, x)
modules/tokenizers/vq_model.py:565
Method
forward
(self, x)
modules/tokenizers/vq_model.py:623
Method
forward
(self, x)
modules/tokenizers/vq_model.py:638
Method
forward
(self, x, attn_mask)
models/ar.py:46
Method
forward
(self, x, attn_mask)
models/ar.py:80
Method
forward
(self, x,attn_mask)
models/ar.py:112
Method
forward
(self, x, attn_mask)
models/ar.py:146
Method
forward
(self, x, attn_mask)
models/ar.py:171
Method
forward
(self, x, attn_mask)
models/ar.py:197
Method
forward
(self, x, time_attn_mask,space_attn_mask)
models/ar.py:209
Method
forward
(self, feature_total, pose_indices_total, yaw_indices_total, drop_flag=False)
models/ar.py:381
Method
forward
(self, token, feature, pose, yaw, token_gt, feature_gt, pose_gt, yaw_gt, vqvae_codebook=None, eval=False)
models/model.py:164
Method
fromstring
Build a Config instance from config text. Args: cfg_str (str): Config text. file_format (str): Config file format cor
utils/config_utils.py:1007
Method
get_block_size
(self)
models/ar.py:331
Function
get_deepspeed_config
(args)
utils/deepspeed_utils.py:3
Function
get_device
()
utils/dist.py:64
Function
get_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: Create sinusoidal timestep embeddings. Args timesteps (
utils/embeddings.py:3
Function
get_undistort_map
Rectify camera distortions. Camera intrinsic parameters and distortions examples are as follows. camera_matrix = np.array([[fx, 0, cx
utils/undistort_image.py:5
Function
get_world_size
()
utils/dist.py:60
Function
indices_to_pose
(idx_x, idx_y, x_division=128, y_divisions=128)
modules/tokenizers/pose_tokenizer.py:21
Function
indices_to_yaws
(idx_yaw, division=512)
modules/tokenizers/pose_tokenizer.py:40
Function
init_lr_schedule
(optimizer, milstones=[1000000, 1500000, 2000000], gamma=0.5)
utils/running.py:15
Function
init_optimizer
(model, lr=1e-4, weight_decay=1e-3)
utils/running.py:10
Function
initialize
(fork=False, backend='nccl', gpu_id_if_not_distibuted=0, timeout=30)
utils/dist.py:20
Method
is_base_line
(c)
utils/config_utils.py:1719
Method
last_layer
(self)
modules/tokenizers/vq_model.py:239
Function
load_from_deepspeed_ckpt
(args, model)
utils/running.py:150
Function
local_master_only
(func)
utils/dist.py:188
Function
log_dict_args
Logs a dictionary of arguments. Args: dict_args (dict): The dictionary of arguments to be logged.
utils/running.py:176
Function
master_only
(func)
utils/dist.py:175
Method
merge
Merge another dictionary into current dictionary. Args: other (dict): Another dictionary.
utils/config_utils.py:776
Method
module
(self)
utils/config_utils.py:118
Method
module
(self)
utils/config_utils.py:532
Function
new_group
(ranks: List[int])
utils/dist.py:90
Method
pretty_text
get formatted python config text.
utils/config_utils.py:1874
Function
resume_ckpt
(local_rank, args, model, optimizer=None)
utils/running.py:71
Function
reverse_seq_data
(poses, seqs)
datasets/datasets_utils.py:6
Function
rgb2ycbcr
(img_np)
utils/utils.py:34
Function
save_ckpt
Save the model, optimizer, lr scheduler.
utils/running.py:54
Function
save_ckpt_deepspeed
Save the model, optimizer, lr scheduler.
utils/running.py:134
Function
save_model
(self, path, epoch, rank=0)
utils/running.py:20
Method
save_model
(self, path, epoch, rank=0)
models/model.py:212
Function
set_gpu_id
(gpu_id: int)
utils/dist.py:68
Function
setup_logger
(logger_name, save_dir, phase, level=logging.INFO, screen=False, to_file=False)
utils/utils.py:9
Method
text
get config text.
utils/config_utils.py:1864
Method
tokenize_pose
(self, poses)
modules/tokenizers/model_tokenizer.py:71
Method
top_k_logits
(self, logits, k)
models/model.py:172
Function
undistort_image
(img, map1, map2, roi)
utils/undistort_image.py:38
Method
values
Yield the values of the dictionary. If class attribute ``lazy`` is False, the value of ``LazyObject`` or ``LazyAttr`` will be built a
utils/config_utils.py:754
Method
visit_Assign
(self, node)
utils/config_utils.py:41
Method
visit_Import
Work with ``_gather_abs_import_lazyobj`` to hack the ``import ...`` syntax. Examples: >>> import mmcls.models
utils/config_utils.py:393
Method
visit_ImportFrom
Hack the ``from ... import ...`` syntax and update the global_dict. Examples: >>> from mmdet.models import RetinaNet Wil
utils/config_utils.py:299
Function
wrapper
(*args, **kwargs)
utils/dist.py:177
Method
z_to_clips_all
(self, indices)
modules/tokenizers/model_tokenizer.py:51
Method
z_to_image
(self, indices)
modules/tokenizers/model_tokenizer.py:35
← previous
201–280 of 280, ranked by callers