MCPcopy Create free account

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

Functions329 in github.com/CGuangyan-BIT/PointGPT

⨍ Functions 329◇ Types & classes 69
↓ 1 callersFunctionround_to_int_32
Takes a Numpy array of float values between -1 and 1, and rounds them to significant 32-bit integer values, to be used in the morton
segmentation/models/z_order.py:4
↓ 1 callersFunctionsample_and_group
Input: npoint: radius: nsample: xyz: input points position data, [B, N, 3] points: input points data, [B,
segmentation/pointnet_util.py:99
↓ 1 callersFunctionsample_and_group
Input: npoint: radius: nsample: xyz: input points position data, [B, N, 3] points: input points data, [B,
segmentation/models/pointnet2_utils.py:107
↓ 1 callersFunctionsample_and_group_all
Input: xyz: input points position data, [B, N, 3] points: input points data, [B, N, D] Return: new_xyz: sampled point
segmentation/pointnet_util.py:139
↓ 1 callersFunctionsample_and_group_all
Input: xyz: input points position data, [B, N, 3] points: input points data, [B, N, D] Return: new_xyz: sampled point
segmentation/models/pointnet2_utils.py:138
↓ 1 callersFunctionsave_experiment_config
(args, config, logger=None)
utils/config.py:66
↓ 1 callersMethodsimplied_morton_sorting
Simplifying the Morton code sorting to iterate and set the nearest patch to the last patch as the next patch, we found this to be more effici
models/PointGPT.py:98
↓ 1 callersMethodsimplied_morton_sorting
(self, xyz, center)
segmentation/models/pt.py:36
↓ 1 callersMethodsplit_scope_key
Split scope and key. The first scope will be split from key. Examples: >>> Registry.split_scope_key('mmdet.ResNet')
utils/registry.py:89
↓ 1 callersFunctiontest
(base_model, test_dataloader, args, config, logger=None)
tools/runner_finetune.py:380
↓ 1 callersFunctiontest
(base_model, test_dataloader, args, config, logger=None)
tools/runner.py:34
↓ 1 callersFunctiontest_vote
(base_model, test_dataloader, epoch, val_writer, args, config, logger=None, times=10)
tools/runner_finetune.py:429
↓ 1 callersFunctionvalidate
(base_model, test_dataloader, epoch, val_writer, args, config, logger=None)
tools/runner_finetune.py:249
↓ 1 callersFunctionvalidate_vote
(base_model, test_dataloader, epoch, val_writer, args, config, logger=None, times=10)
tools/runner_finetune.py:293
FunctionPYBIND11_MODULE
extensions/emd/cuda/emd.cpp:23
FunctionPYBIND11_MODULE
extensions/chamfer_dist/chamfer_cuda.cpp:36
FunctionPYBIND11_MODULE
segmentation/extensions/emd/cuda/emd.cpp:23
FunctionPYBIND11_MODULE
segmentation/extensions/chamfer_dist/chamfer_cuda.cpp:36
Method__call__
(self, pc)
datasets/data_transforms.py:7
Method__call__
(self, pc)
datasets/data_transforms.py:26
Method__call__
(self, pc)
datasets/data_transforms.py:40
Method__call__
(self, pc)
datasets/data_transforms.py:55
Method__call__
(self, pc)
datasets/data_transforms.py:68
Method__call__
(self, pc)
datasets/data_transforms.py:83
Method__call__
(self, coords)
datasets/data_transforms.py:109
Method__contains__
(self, key)
utils/registry.py:59
Method__getitem__
(self, index)
datasets/ModelNetDatasetFewShot.py:59
Method__getitem__
(self, idx)
datasets/UnlabeledHybrid.py:66
Method__getitem__
(self, idx)
datasets/LabeledHybrid.py:55
Method__getitem__
(self, idx)
datasets/ScanObjectNNDataset.py:33
Method__getitem__
(self, idx)
datasets/ScanObjectNNDataset.py:73
Method__getitem__
(self, idx)
datasets/ShapeNet55Dataset.py:65
Method__getitem__
(self, index)
datasets/ModelNetDataset.py:142
Method__getitem__
(self, index)
segmentation/dataset.py:62
Method__getitem__
(self, index)
segmentation/dataset.py:138
Method__init__
(self, acc=0.)
tools/runner_pretrain.py:32
Method__init__
(self, acc=0.)
tools/runner_finetune.py:38
Method__init__
( self, model, bn_lambda, last_epoch=-1, setter=set_bn_momentum_default )
utils/misc.py:99
Method__init__
(self, items=None)
utils/AverageMeter.py:3
Method__init__
(self, name, build_func=None, parent=None, scope=None)
utils/registry.py:32
Method__init__
(self, config)
datasets/ModelNetDatasetFewShot.py:30
Method__init__
(self, scale_low=2. / 3., scale_high=3. / 2., translate_range=0.2)
datasets/data_transforms.py:21
Method__init__
(self, std=0.01, clip=0.05)
datasets/data_transforms.py:37
Method__init__
(self, scale_low=2. / 3., scale_high=3. / 2.)
datasets/data_transforms.py:51
Method__init__
(self, translate_range=0.2)
datasets/data_transforms.py:65
Method__init__
(self, max_dropout_ratio=0.5)
datasets/data_transforms.py:79
Method__init__
upright_axis: axis index among x,y,z, i.e. 2 for z
datasets/data_transforms.py:98
Method__init__
(self, config)
datasets/UnlabeledHybrid.py:12
Method__init__
(self, config)
datasets/LabeledHybrid.py:11
Method__init__
(self, config, **kwargs)
datasets/ScanObjectNNDataset.py:53
Method__init__
(self, config)
datasets/ShapeNet55Dataset.py:12
Method__init__
(self, config)
datasets/ModelNetDataset.py:54
Method__init__
(self)
extensions/emd/emd.py:29
Method__init__
(self, ignore_zeros=False)
extensions/chamfer_dist/__init__.py:49
Method__init__
(self, ignore_zeros=False)
extensions/chamfer_dist/__init__.py:67
Method__init__
( self, embed_dim, num_heads, num_layers, num_classes, trans_dim, group_size, pretrained=False )
models/GPT.py:32
Method__init__
( self, embed_dim, num_heads, num_layers, trans_dim, group_size )
models/GPT.py:105
Method__init__
(self, encoder_channel)
models/PointGPT.py:19
Method__init__
(self, encoder_channel)
models/PointGPT.py:56
Method__init__
(self, num_group, group_size)
models/PointGPT.py:91
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
models/PointGPT.py:208
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
models/PointGPT.py:236
Method__init__
(self, n_dim: int = 1, d_model: int = 256, temperature=10000, scale=None)
models/PointGPT.py:268
Method__init__
(self, config, **kwargs)
models/PointGPT.py:308
Method__init__
(self, config)
models/PointGPT.py:443
Method__init__
(self, config, **kwargs)
models/PointGPT.py:504
Method__init__
( self, model, bn_lambda, last_epoch=-1, setter=set_bn_momentum_default )
segmentation/misc.py:104
Method__init__
(self, root, npoint=1024, split='train', uniform=False, normal_channel=True, cache_size=15000)
segmentation/dataset.py:10
Method__init__
(self, root='/data/cgy/ShapenetPart/shapenetcore_partanno_segmentation_benchmark_v0_normal', npoints=2500, spl
segmentation/dataset.py:67
Method__init__
(self, npoint, radius_list, nsample_list, in_channel, mlp_list, knn=False)
segmentation/pointnet_util.py:200
Method__init__
(self, in_channel, mlp)
segmentation/pointnet_util.py:262
Method__init__
(self)
segmentation/extensions/emd/emd.py:29
Method__init__
(self, ignore_zeros=False)
segmentation/extensions/chamfer_dist/__init__.py:49
Method__init__
(self, ignore_zeros=False)
segmentation/extensions/chamfer_dist/__init__.py:67
Method__init__
( self, embed_dim, num_heads, num_layers, trans_dim, group_size, drop_path_rate )
segmentation/models/gpt2_seg.py:36
Method__init__
( self, embed_dim, num_heads, num_layers, trans_dim, group_size, drop_path_rate )
segmentation/models/gpt2_seg.py:96
Method__init__
(self, num_group, group_size)
segmentation/models/pt.py:29
Method__init__
(self, encoder_channel)
segmentation/models/pt.py:122
Method__init__
(self, encoder_channel)
segmentation/models/pt.py:155
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
segmentation/models/pt.py:211
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
segmentation/models/pt.py:241
Method__init__
(self, embed_dim=768, depth=4, num_heads=12, mlp_ratio=4., qkv_bias=False, qk_scale=None, dro
segmentation/models/pt.py:265
Method__init__
(self, n_dim: int = 1, d_model: int = 256, temperature=10000, scale=None)
segmentation/models/pt.py:299
Method__init__
(self, cls_dim, trans_dim=384, depth=12, drop_path_rate=0.1, num_heads=6, decoder_depth=4, group_size=32, num_
segmentation/models/pt.py:339
Method__init__
(self)
segmentation/models/pt.py:523
Method__init__
(self, npoint, radius_list, nsample_list, in_channel, mlp_list)
segmentation/models/pointnet2_utils.py:203
Method__init__
(self, in_channel, mlp)
segmentation/models/pointnet2_utils.py:263
Method__len__
(self)
utils/registry.py:56
Method__len__
(self)
datasets/ModelNetDatasetFewShot.py:56
Method__len__
(self)
datasets/UnlabeledHybrid.py:78
Method__len__
(self)
datasets/LabeledHybrid.py:66
Method__len__
(self)
datasets/ScanObjectNNDataset.py:46
Method__len__
(self)
datasets/ScanObjectNNDataset.py:86
Method__len__
(self)
datasets/ShapeNet55Dataset.py:76
Method__len__
(self)
segmentation/dataset.py:36
Method__len__
(self)
segmentation/dataset.py:163
Method__repr__
(self)
utils/registry.py:62
Method_init_weights
(self, m)
models/PointGPT.py:360
Method_init_weights
(self, m)
models/PointGPT.py:618
Function_named_modules_with_dup
The same as `model.named_modules()`, except that it includes duplicated modules that have more than one name.
utils/checkpoint.py:121
← previousnext →101–200 of 329, ranked by callers