MCPcopy Create free account

hub / github.com/CGuangyan-BIT/PointGPT / functions

Functions329 in github.com/CGuangyan-BIT/PointGPT

⨍ Functions 329◇ Types & classes 69
Functionrotate_point_cloud_by_angle
Rotate the point cloud along up direction with certain angle. Input: BxNx3 array, original batch of point clouds Return:
segmentation/provider.py:133
Functionrotate_point_cloud_by_angle_with_normal
Rotate the point cloud along up direction with certain angle. Input: BxNx6 array, original batch of point clouds with normal
segmentation/provider.py:152
Functionrotate_point_cloud_with_normal
Randomly rotate XYZ, normal point cloud. Input: batch_xyz_normal: B,N,6, first three channels are XYZ, last 3 all normal
segmentation/provider.py:86
Functionrotate_point_cloud_z
Randomly rotate the point clouds to augument the dataset rotation is per shape based along up direction Input: BxNx3 array,
segmentation/provider.py:66
Functionrun_net
(args, config, train_writer=None, val_writer=None)
tools/runner_pretrain.py:57
Functionrun_net
(args, config, train_writer=None, val_writer=None)
tools/runner_finetune.py:58
Functionsave_checkpoint
(base_model, optimizer, epoch, metrics, best_metrics, prefix, args, logger=None)
tools/builder.py:147
Methodscope
(self)
utils/registry.py:112
Functionseprate_point_cloud
seprate point cloud: usage : using to generate the incomplete point cloud with a setted number.
utils/misc.py:131
Functionseprate_point_cloud
seprate point cloud: usage : using to generate the incomplete point cloud with a setted number.
segmentation/misc.py:135
Functionset_bn_momentum_default
(bn_momentum)
utils/misc.py:91
Functionset_bn_momentum_default
(bn_momentum)
segmentation/misc.py:94
Functionset_random_seed
Set random seed. Args: seed (int): Seed to be used. deterministic (bool): Whether to set the deterministic option for
utils/misc.py:42
Functionset_random_seed
Set random seed. Args: seed (int): Seed to be used. deterministic (bool): Whether to set the deterministic option for
segmentation/misc.py:45
Functionshift_point_cloud
Randomly shift point cloud. Shift is per point cloud. Input: BxNx3 array, original batch of point clouds Return:
segmentation/provider.py:214
Functionshuffle_data
Shuffle data and labels. Input: data: B,N,... numpy array label: B,... numpy array Return: shuffled dat
segmentation/provider.py:22
Functionshuffle_points
Shuffle orders of points in each point cloud -- changes FPS behavior. Use the same shuffling idx for the entire batch. Input:
segmentation/provider.py:34
Methodstate_dict
(self)
tools/runner_finetune.py:52
Methodtest_chamfer_dist
(self)
extensions/chamfer_dist/test.py:24
Methodtest_chamfer_dist
(self)
segmentation/extensions/chamfer_dist/test.py:24
Functiontest_net
()
tools/runner_pretrain.py:272
Functiontest_net
(args, config)
tools/runner_finetune.py:361
Functiontimeit
(tag, t)
segmentation/pointnet_util.py:11
Functiontimeit
(tag, t)
segmentation/models/pointnet2_utils.py:7
Functionvalidate
(base_model, extra_train_dataloader, test_dataloader, epoch, val_writer, args, config, logger=None)
tools/runner_pretrain.py:209
Functionvisualize_KITTI
(path, data_list, titles = ['input','pred'], cmap=['bwr','autumn'], zdir='y', xlim=(
utils/misc.py:207
Functionvisualize_KITTI
(path, data_list, titles=['input', 'pred'], cmap=['bwr', 'autumn'], zdir='y', xlim=(-1, 1)
segmentation/misc.py:211
Functionworker_init_fn
(worker_id)
utils/misc.py:23
Functionworker_init_fn
(worker_id)
segmentation/misc.py:23
← previous301–329 of 329, ranked by callers