MCPcopy Create free account

hub / github.com/baaivision/Uni3D / functions

Functions189 in github.com/baaivision/Uni3D

↓ 14 callersMethodupdate
(self, val, n=1)
main.py:649
↓ 12 callersFunctionprint_log
Print a log message. Args: msg (str): The message to be logged. logger (logging.Logger | str | None): The logger to be used.
data/utils/logger.py:103
↓ 11 callersFunctionis_master
(args, local=False)
utils/distributed.py:33
↓ 7 callersFunctionprint
(*args, **kwargs)
data/utils/utils.py:57
↓ 6 callersFunctionprint
(*args, **kwargs)
utils/utils.py:57
↓ 6 callersFunctiontest_zeroshot_3d_core
(test_loader, validate_dataset_name, model, clip_model, tokenizer, args=None, test_data=None)
main.py:502
↓ 5 callersMethodget
(cls, file_path)
data/utils/io.py:8
↓ 4 callersFunctionget_dataset
(train_transform, tokenizer, args, dataset_name=None)
utils/utils.py:247
↓ 4 callersFunctionget_parameters
(args, model, assigner, tower)
utils/optim.py:55
↓ 4 callersFunctionworld_info_from_env
()
utils/distributed.py:45
↓ 3 callersMethod__init__
(self, num_group, group_size)
models/point_encoder.py:94
↓ 3 callersMethod_register_module
(self, module_class, module_name=None, force=False)
data/utils/registry.py:168
↓ 3 callersFunction_warmup_lr
(base_lr, warmup_length, step)
utils/scheduler.py:9
↓ 3 callersMethodbackward
(ctx, *grads)
utils/utils.py:194
↓ 3 callersMethodget
Get the registry record. Args: key (str): The class name in string format. Returns: class: The corresponding c
data/utils/registry.py:123
↓ 3 callersFunctionget_world_size
()
utils/utils.py:73
↓ 3 callersFunctionget_world_size
()
data/utils/utils.py:73
↓ 3 callersFunctionnormalize_pc
(pc)
data/utils/data.py:11
↓ 2 callersFunctionbytes_to_unicode
Returns list of utf-8 byte and a corresponding list of unicode strings. The reversible bpe codes work on unicode strings. This means you
utils/tokenizer.py:18
↓ 2 callersFunctioncreate_optimizer
(args, model, return_params=False)
utils/optim.py:196
↓ 2 callersMethoddisplay
(self, batch)
main.py:677
↓ 2 callersMethodencode_pc
(self, pc)
models/uni3d.py:15
↓ 2 callersFunctionget_all_parameters
(args, model)
utils/optim.py:178
↓ 2 callersFunctionget_grad_norm_
(parameters, norm_type: float = 2.0)
utils/optim.py:229
↓ 2 callersFunctionget_logger
Initialize and get a logger by name. If the logger has not been initialized, this method will initialize the logger by adding one or two handl
data/utils/logger.py:29
↓ 2 callersFunctionget_pairs
Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).
utils/tokenizer.py:40
↓ 2 callersFunctionis_dist_avail_and_initialized
()
utils/utils.py:65
↓ 2 callersFunctionis_dist_avail_and_initialized
()
data/utils/utils.py:65
↓ 2 callersFunctionis_main_process
()
utils/utils.py:85
↓ 2 callersFunctionis_main_process
()
data/utils/utils.py:85
↓ 2 callersFunctionrandom_scale_point_cloud
Randomly scale the point cloud. Scale is per point cloud. Input: BxNx3 array, original batch of point clouds Return:
data/datasets.py:88
↓ 2 callersFunctionrandom_seed
(seed=42, rank=0)
main.py:33
↓ 2 callersFunctionrotate_perturbation_point_cloud
Randomly perturb the point clouds by small rotations Input: BxNx3 array, original batch of point clouds Return: B
data/datasets.py:127
↓ 2 callersFunctionrotate_point_cloud
Randomly rotate the point clouds to augument the dataset rotation is per shape based along up direction Input: BxNx3 array,
data/datasets.py:59
↓ 2 callersFunctionshift_point_cloud
Randomly shift point cloud. Shift is per point cloud. Input: BxNx3 array, original batch of point clouds Return:
data/datasets.py:101
↓ 2 callersMethodsynchronize
(self)
main.py:683
↓ 1 callersMethod_add_children
Add children for a registry. The ``registry`` will be added as children based on its scope. The parent registry could build objects fr
data/utils/registry.py:149
↓ 1 callersMethod_get_batch_fmtstr
(self, num_batches)
main.py:687
↓ 1 callersMethod_read_bin
(cls, file_path)
data/utils/io.py:47
↓ 1 callersMethod_read_h5
(cls, file_path)
data/utils/io.py:42
↓ 1 callersMethod_read_npy
(cls, file_path)
data/utils/io.py:26
↓ 1 callersMethod_read_pcd
(cls, file_path)
data/utils/io.py:32
↓ 1 callersMethod_read_txt
(cls, file_path)
data/utils/io.py:38
↓ 1 callersFunctionaccuracy
Computes the accuracy over the k top predictions for the specified values of k
main.py:693
↓ 1 callersFunctionassign_learning_rate
(optimizer, new_lr)
utils/scheduler.py:4
↓ 1 callersFunctionbasic_clean
(text)
utils/tokenizer.py:52
↓ 1 callersMethodbpe
(self, token)
utils/tokenizer.py:82
↓ 1 callersMethodbuild
(self, *args, **kwargs)
data/utils/registry.py:146
↓ 1 callersMethodbuild_3d_dataset
(self, args, config)
data/datasets.py:698
↓ 1 callersFunctionbuild_dataset_from_cfg
Build a dataset, defined by `dataset_name`. Args: cfg (eDICT): Returns: Dataset: a constructed dataset specified by data
data/utils/build.py:6
↓ 1 callersFunctioncfg_from_yaml_file
(cfg_file)
data/datasets.py:668
↓ 1 callersFunctioncfg_from_yaml_file
(cfg_file)
data/utils/config.py:37
↓ 1 callersFunctioncompute_embedding
(clip_model, texts, image)
main.py:38
↓ 1 callersFunctioncreate_deepspeed_config
(args)
utils/distributed.py:109
↓ 1 callersMethoddecode
(self, tokens)
utils/tokenizer.py:131
↓ 1 callersFunctiondefault_bpe
()
utils/tokenizer.py:13
↓ 1 callersMethoddeprecated_register_module
(self, cls=None, force=False)
data/utils/registry.py:183
↓ 1 callersMethodencode
(self, text)
utils/tokenizer.py:123
↓ 1 callersFunctionfps
data B N 3 number int
models/point_encoder.py:7
↓ 1 callersMethodgenerate_random_text
(self, embedding_dim=1024)
data/datasets.py:329
↓ 1 callersFunctionget_assigner
(args, model)
utils/optim.py:145
↓ 1 callersFunctionget_default_params
(model_name)
utils/params.py:4
↓ 1 callersFunctionget_loss_scale_for_deepspeed
(model)
utils/optim.py:220
↓ 1 callersFunctionget_num_layer_for_transformer
(param_name, num_max_layer)
utils/optim.py:14
↓ 1 callersFunctionget_rank
()
utils/utils.py:79
↓ 1 callersFunctionget_rank
()
data/utils/utils.py:79
↓ 1 callersMethodget_scale
(self, layer_id)
utils/optim.py:49
↓ 1 callersMethodinfer_scope
Infer the scope of registry. The name of the package where registry is defined will be returned. Example: # in mmdet/model
data/utils/registry.py:69
↓ 1 callersFunctioninit_distributed_device
(args)
utils/distributed.py:65
↓ 1 callersFunctionis_dist_avail_and_initialized
()
utils/distributed.py:13
↓ 1 callersFunctionis_global_master
(args)
utils/distributed.py:25
↓ 1 callersFunctionis_local_master
(args)
utils/distributed.py:29
↓ 1 callersFunctionis_using_distributed
()
utils/distributed.py:37
↓ 1 callersFunctionknn_point
Input: nsample: max sample number in local region xyz: all points, [B, N, C] new_xyz: query points, [B, S, C] Return:
models/point_encoder.py:17
↓ 1 callersFunctionmain
(args)
main.py:54
↓ 1 callersFunctionmerge_new_config
(config, new_config)
utils/utils.py:14
↓ 1 callersFunctionmerge_new_config
(config, new_config)
data/datasets.py:649
↓ 1 callersFunctionmerge_new_config
(config, new_config)
data/utils/config.py:18
↓ 1 callersFunctionmerge_new_config
(config, new_config)
data/utils/utils.py:14
↓ 1 callersFunctionparse_args
(args)
utils/params.py:13
↓ 1 callersMethodpc_norm
pc: NxC, return NxC
data/datasets.py:231
↓ 1 callersMethodpc_norm
pc: NxC, return NxC
data/datasets.py:335
↓ 1 callersMethodpc_norm
pc: NxC, return NxC
data/datasets.py:521
↓ 1 callersFunctionpc_normalize
(pc)
data/datasets.py:29
↓ 1 callersFunctionrandom_point_dropout
batch_pc: BxNx3
data/datasets.py:79
↓ 1 callersFunctionrandom_point_dropout
batch_pc: BxNx3
data/utils/data.py:20
↓ 1 callersFunctionrandom_scale_point_cloud
Randomly scale the point cloud. Scale is per point cloud. Input: BxNx3 array, original batch of point clouds Return:
data/utils/data.py:29
↓ 1 callersMethodreset
(self)
main.py:643
↓ 1 callersFunctionrotate_perturbation_point_cloud
Randomly perturb the point clouds by small rotations Input: BxNx3 array, original batch of point clouds Return: B
data/utils/data.py:55
↓ 1 callersFunctionsave_experiment_config
(args, config, logger = None)
data/utils/config.py:60
↓ 1 callersFunctionsetup_for_distributed
This function disables printing when not in master process
utils/utils.py:50
↓ 1 callersFunctionsetup_for_distributed
This function disables printing when not in master process
data/utils/utils.py:50
↓ 1 callersFunctionsetup_logging
(log_file, level, include_host=False)
utils/logger.py:4
↓ 1 callersFunctionshift_point_cloud
Randomly shift point cloud. Shift is per point cloud. Input: BxNx3 array, original batch of point clouds Return:
data/utils/data.py:42
↓ 1 callersMethodsplit_scope_key
Split scope and key. The first scope will be split from key. Examples: >>> Registry.split_scope_key('mmdet.ResNet')
data/utils/registry.py:89
↓ 1 callersFunctionsquare_distance
Calculate Euclid distance between each two points. src^T * dst = xn * xm + yn * ym + zn * zm; sum(src^2, dim=-1) = xn*xn + yn*yn + zn*zn;
models/point_encoder.py:30
↓ 1 callersMethodsynchronize
(self)
main.py:655
↓ 1 callersFunctiontest_zeroshot_3d
(args, model, clip_model)
main.py:603
↓ 1 callersFunctiontrain
(train_loader, clip_model, model, criterion, optimizer, scaler, scheduler, epoch, args)
main.py:377
↓ 1 callersFunctionwarmup_cosine_lr
(optimizer, args, steps)
utils/scheduler.py:25
next →1–100 of 189, ranked by callers