MCPcopy Create free account

hub / github.com/OpenMeshLab/MeshXL / functions

Functions118 in github.com/OpenMeshLab/MeshXL

⨍ Functions 118◇ Types & classes 13
↓ 15 callersMethodlog_messages
(self, message: str)
utils/logger.py:35
↓ 12 callersFunctionprint
(*args, **kwargs)
utils/dist.py:43
↓ 10 callersMethodupdate
(self, value, n=1)
utils/misc.py:53
↓ 6 callersFunctionis_distributed
()
utils/dist.py:8
↓ 5 callersFunctioncompute_length_vec3
(vec3)
utils/ply_helper.py:57
↓ 5 callersMethodmax
(self)
utils/misc.py:86
↓ 5 callersFunctionwrite_ply
(verts, colors, indices, output_file)
utils/ply_helper.py:27
↓ 4 callersMethod_preprocess_data
(self, data_chunk)
datasets/sft/base_dataset.py:33
↓ 4 callersFunctionget_world_size
()
utils/dist.py:24
↓ 4 callersMethodtrain
(self, mode: bool = True)
models/mesh_xl/get_model.py:11
↓ 3 callersFunctionsave_checkpoint
( checkpoint_dir, model_no_ddp, optimizer, epoch, args, best_val_metrics, filename
utils/io.py:8
↓ 3 callersMethodtokenize
Turn 3D meshes into sequential tokens: <bos> [<x>, <y>, <z>], ... <eos>
models/mesh_xl/tokenizer.py:47
↓ 2 callersFunctionall_reduce_sum
(tensor)
utils/dist.py:67
↓ 2 callersFunctionbuild_dataloader_func
(args, dataset, split)
main.py:85
↓ 2 callersFunctionconvert_oriented_box_to_trimesh_fmt
(box)
utils/pc_util.py:197
↓ 2 callersFunctioncreate_cylinder_mesh
(radius, p0, p1, stacks=10, slices=10)
utils/ply_helper.py:53
↓ 2 callersMethoddetokenize
Turn sequential tokens: <bos> [<x>, <y>, <z>], ... <eos> into 3D meshes
models/mesh_xl/tokenizer.py:103
↓ 2 callersMethoddetokenize
Turn sequential tokens: <bos> [<x>, <y>, <z>], ... <eos> into 3D meshes
models/x_mesh_xl/tokenizer.py:103
↓ 2 callersFunctionheading2rotmat
(heading_angle)
utils/pc_util.py:188
↓ 2 callersFunctionis_primary
()
utils/dist.py:20
↓ 2 callersMethodlog_scalars
(self, scalar_dict, step, prefix=None)
utils/logger.py:23
↓ 2 callersFunctionperplexity
(neg_log_likelihood: list)
eval_utils/perplexity.py:13
↓ 2 callersMethodtrain
(self, mode: bool = True)
models/x_mesh_xl/get_model.py:63
↓ 1 callersMethod__init__
(self, args)
models/x_mesh_xl/get_model.py:70
↓ 1 callersMethod_fetch_data
(self, idx)
datasets/sft/base_dataset.py:44
↓ 1 callersFunctionadjust_learning_rate
(args, optimizer, curr_iter, max_iters)
engine.py:24
↓ 1 callersFunctionbarrier
()
utils/dist.py:30
↓ 1 callersFunctionbuild_dataset_func
(args)
main.py:104
↓ 1 callersFunctionbuild_model_func
(args)
main.py:135
↓ 1 callersFunctioncompute_learning_rate
(args, curr_iter, max_iters)
engine.py:10
↓ 1 callersFunctionconvert_box_to_trimesh_fmt
(box)
utils/pc_util.py:158
↓ 1 callersFunctiondiscretize
( t: Tensor, continuous_range: Tuple[float, float], num_discrete: int = 128 )
models/mesh_xl/tokenizer.py:8
↓ 1 callersFunctiondiscretize
( t: Tensor, continuous_range: Tuple[float, float], num_discrete: int = 128 )
models/x_mesh_xl/tokenizer.py:8
↓ 1 callersFunctiondo_train
( args, model, accelerator, optimizer, dataloaders, best_val_metrics, logger )
engine.py:32
↓ 1 callersMethodencode_text
(self, input_ids, attention_mask)
models/x_mesh_xl/get_model.py:33
↓ 1 callersMethodgenerate
(self, data_dict: dict=None, num_return_sequences: int=8, generation_config: dict=dict())
models/mesh_xl/get_model.py:170
↓ 1 callersMethodgenerate
(self, data_dict: dict=None, num_return_sequences: int=8, generation_config: dict=dict())
models/x_mesh_xl/get_model.py:141
↓ 1 callersFunctionget_bbox_edges
(bbox_min, bbox_max)
utils/ply_helper.py:122
↓ 1 callersFunctionget_bbox_verts
(bbox_min, bbox_max)
utils/ply_helper.py:123
↓ 1 callersFunctionget_model
(args)
models/mesh_xl/get_model.py:197
↓ 1 callersFunctionget_rank
()
utils/dist.py:14
↓ 1 callersMethodloss_wrapper
(self, loss: Tensor)
models/mesh_xl/get_model.py:79
↓ 1 callersFunctionmain
(args)
main.py:141
↓ 1 callersFunctionmake_args_parser
()
sample_t2m.py:26
↓ 1 callersFunctionmake_args_parser
()
main.py:13
↓ 1 callersFunctionnormalize_mesh
(vertices: np.ndarray, scale_range: tuple=(-1.0, 1.0))
datasets/sft/base_dataset.py:17
↓ 1 callersMethodperplexity
(self, data_dict: dict)
models/mesh_xl/get_model.py:128
↓ 1 callersFunctionpost_process_mesh
(mesh_coords: Tensor, filename: str)
sample_t2m.py:12
↓ 1 callersFunctionpost_process_mesh
(mesh_coords: Tensor, filename: str)
eval_utils/perplexity.py:20
↓ 1 callersFunctionprocess_mesh
(mesh_coords: Tensor, filename: str)
eval_utils/sample_generation.py:13
↓ 1 callersFunctionresume_if_possible
Resume if checkpoint is available. Return - epoch of loaded checkpoint.
utils/io.py:34
↓ 1 callersFunctionrotation
(axis, angle)
utils/ply_helper.py:60
↓ 1 callersFunctionscale_mesh
(vertices: np.ndarray, scale_factor: tuple=(0.75, 1.25))
datasets/sft/base_dataset.py:10
↓ 1 callersFunctionsetup_print_for_distributed
This function disables printing when not in primary process
utils/dist.py:36
↓ 1 callersMethodtokenize
Turn 3D meshes into sequential tokens: <bos> [<x>, <y>, <z>], ... <eos>
models/x_mesh_xl/tokenizer.py:47
↓ 1 callersMethodtrain_one_step
(self, data_dict: dict)
models/mesh_xl/get_model.py:86
↓ 1 callersFunctionundiscretize
( t: Tensor, continuous_range = Tuple[float, float], num_discrete: int = 128 )
models/mesh_xl/tokenizer.py:23
↓ 1 callersFunctionundiscretize
( t: Tensor, continuous_range = Tuple[float, float], num_discrete: int = 128 )
models/x_mesh_xl/tokenizer.py:23
Method__getitem__
(self, idx)
datasets/dummy_dataset.py:16
Method__getitem__
(self, idx)
datasets/sft/base_dataset.py:51
Method__init__
(self, window_size=20, fmt=None)
utils/misc.py:45
Method__init__
(self, log_dir=None, accelerator=None)
utils/logger.py:9
Method__init__
(self, *args, **kwargs)
datasets/dummy_dataset.py:9
Method__init__
(self, args, split_set="train", augment=False)
datasets/sft/shapenet_table.py:13
Method__init__
(self, args, split_set="train", augment=False)
datasets/sft/shapenet_bench.py:13
Method__init__
(self, *args, **kwargs)
datasets/sft/base_dataset.py:28
Method__init__
(self, args, split_set="train", augment=False)
datasets/sft/shapenet_chair.py:13
Method__init__
(self, args, split_set="train", augment=False)
datasets/sft/shapenet_lamp.py:13
Method__init__
(self, args)
models/mesh_xl/get_model.py:15
Method__init__
(self, args)
models/mesh_xl/tokenizer.py:39
Method__init__
(self, args, hidden_size)
models/x_mesh_xl/get_model.py:19
Method__init__
(self, args)
models/x_mesh_xl/tokenizer.py:39
Method__len__
(self)
datasets/dummy_dataset.py:13
Method__len__
(self)
datasets/sft/base_dataset.py:48
Method__str__
(self)
utils/misc.py:93
Functionall_gather_dict
Run all_gather on data which is a dictionary of Tensors
utils/dist.py:159
Functionall_gather_pickle
Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]
utils/dist.py:114
Functionall_reduce_average
(tensor)
utils/dist.py:80
Methodavg
(self)
utils/misc.py:77
Functionevaluate
( args, curr_epoch, model, accelerator, dataset_loader, logout=print, curr_train_i
eval_utils/sample_generation.py:28
Functionevaluate
( args, curr_epoch, model, accelerator, dataset_loader, logger, curr_train_iter=-1
eval_utils/perplexity.py:36
Methodforward
( self, data_dict: dict=None, is_eval: bool=False, is_gener
models/mesh_xl/get_model.py:48
Methodforward
(self, data_dict: dict)
models/mesh_xl/tokenizer.py:137
Methodforward
(self, input_ids, attention_mask)
models/x_mesh_xl/get_model.py:41
Methodforward
( self, data_dict: dict=None, is_eval: bool=False, is_gener
models/x_mesh_xl/get_model.py:105
Methodforward
(self, data_dict: dict)
models/x_mesh_xl/tokenizer.py:137
Functionget_model
(args)
models/x_mesh_xl/get_model.py:169
Methodglobal_avg
(self)
utils/misc.py:82
Functionhuber_loss
Ref: https://github.com/charlesq34/frustum-pointnets/blob/master/models/model_util.py x = error = pred - gt or dist(pred,gt) 0.5 * |x|^2
utils/misc.py:25
Functioninit_distributed
(gpu_id, global_rank, world_size, dist_url, dist_backend)
utils/dist.py:51
Methodmedian
(self)
utils/misc.py:72
Functionmy_worker_init_fn
(worker_id)
utils/misc.py:9
Functionnms_2d
(boxes, overlap_threshold)
utils/nms.py:11
Functionnms_2d_faster
(boxes, overlap_threshold, old_type=False)
utils/nms.py:43
Functionnms_3d_faster
(boxes, overlap_threshold, old_type=False)
utils/nms.py:79
Functionnms_3d_faster_samecls
(boxes, overlap_threshold, old_type=False)
utils/nms.py:120
Functionpoint_cloud_to_bbox
Extract the axis aligned box from a pcl or batch of pcls Args: points: Nx3 points or BxNx3 output is 6 dim: xyz pos of center and
utils/pc_util.py:132
Functionrandom_sampling
Input is NxC, output is num_samplexC
utils/pc_util.py:24
Functionread_mesh_vertices_rgb_normal
read XYZ RGB normals point cloud from filename PLY file
utils/ply_helper.py:7
Functionreduce_dict
Args: input_dict (dict): all the values will be reduced average (bool): whether to do average or sum Reduce the values in the
utils/dist.py:86
next →1–100 of 118, ranked by callers