MCPcopy Create free account

hub / github.com/Zhiyuan-R/Tiger-Diffusion / functions

Functions261 in github.com/Zhiyuan-R/Tiger-Diffusion

↓ 1 callersFunctionpairwise_dist
(x, y)
metrics/ChamferDistancePytorch/chamfer_python.py:4
↓ 1 callersFunctionparse_args
()
test_generation.py:533
↓ 1 callersFunctionparse_args
()
train_generation.py:781
↓ 1 callersFunctionpcwrite
Save a point cloud to a polygon .ply file.
utils/visualize.py:94
↓ 1 callersMethodq_mean_variance
(self, x_start, t)
train_generation.py:153
↓ 1 callersMethodq_posterior_mean_variance
Compute the mean and variance of the diffusion posterior q(x_{t-1} | x_t, x_0)
test_generation.py:126
↓ 1 callersMethodq_sample
Diffuse the data (t == 0 means diffused for 1 step)
test_generation.py:113
↓ 1 callersMethodrandom_rm_parts
(self, raw_pc, part_labels)
datasets/partnet.py:142
↓ 1 callersMethodread_point_cloud_part_label
(path)
datasets/partnet.py:136
↓ 1 callersFunctionrotate_point_cloud
(points, transformation_mat)
datasets/partnet.py:37
↓ 1 callersFunctiontest_chamfer
(distChamfer, dim)
metrics/ChamferDistancePytorch/unit_test.py:14
↓ 1 callersMethodtimestep_embedding
Create sinusoidal timestep embeddings. :param t: a 1-D Tensor of N indices, one per batch element. These ma
model/transformer_branch.py:33
↓ 1 callersFunctiontimings
(distChamfer, dim)
metrics/ChamferDistancePytorch/unit_test.py:38
↓ 1 callersFunctiontrain
(gpu, opt, output_dir, noises_init)
train_generation.py:546
↓ 1 callersMethodtrain
(self)
train_generation.py:459
↓ 1 callersFunctionunit_cube_grid_point_cloud
Returns the center coordinates of each cell of a 3D grid with resolution^3 cells, that is placed in the unit-cube. If clip_sphere it True it d
utils/metrics.py:13
↓ 1 callersFunctionunit_cube_grid_point_cloud
Returns the center coordinates of each cell of a 3D grid with resolution^3 cells, that is placed in the unit-cube. If clip_sphere it True it d
metrics/evaluation_metrics.py:184
↓ 1 callersFunctionupsample_point_cloud
upsample points by random choice :param points: (n, 3) :param n_pts: int, > n :return:
datasets/partnet.py:68
FunctionEMD_CD
(sample_pcs, ref_pcs, batch_size, reduced=True)
metrics/evaluation_metrics.py:29
FunctionNN_loss
(x, y, dim=0)
metrics/ChamferDistancePytorch/chamfer_python.py:12
FunctionPYBIND11_MODULE
modules/functional/src/bindings.cpp:10
FunctionPYBIND11_MODULE
metrics/PyTorchEMD/cuda/emd.cpp:23
FunctionPYBIND11_MODULE
metrics/ChamferDistancePytorch/chamfer2D/chamfer_cuda.cpp:30
FunctionPYBIND11_MODULE
metrics/ChamferDistancePytorch/chamfer5D/chamfer_cuda.cpp:30
FunctionPYBIND11_MODULE
metrics/ChamferDistancePytorch/chamfer3D/chamfer_cuda.cpp:30
Method__call__
(self, depth_pth, depth_minmax_pth)
datasets/shapenet_data_sv.py:235
Method__call__
(self, points)
datasets/shapenet_data_pc.py:248
Method__getitem__
(self, index)
datasets/shapenet_data_sv.py:160
Method__getitem__
(self, idx)
datasets/shapenet_data_pc.py:157
Method__getitem__
(self, index)
datasets/partnet.py:157
Method__init__
(self,betas, loss_type, model_mean_type, model_var_type)
test_generation.py:54
Method__init__
(self, num_classes, embed_dim, use_att,dropout, extra_feature_channels=3, width_multiplier=1,
test_generation.py:263
Method__init__
(self,betas, loss_type, model_mean_type, model_var_type)
train_generation.py:100
Method__init__
(self, num_classes, embed_dim, use_att,dropout, extra_feature_channels=3, width_multiplier=1,
train_generation.py:392
Method__init__
(self, in_channels, out_channels, dim=1)
modules/shared_mlp.py:12
Method__init__
(self, resolution, normalize=True, eps=0)
modules/voxelization.py:10
Method__init__
(self, channel, reduction=8, use_relu=False)
modules/se.py:9
Method__init__
(self, in_ch, num_groups, D=3)
modules/pvconv.py:17
Method__init__
(self, in_channels, out_channels, kernel_size, resolution, attention=False, leak=0.2, dropout
modules/pvconv.py:101
Method__init__
(self, radius, num_neighbors, include_coordinates=True)
modules/ball_query.py:10
Method__init__
(self, num_heading_angle_bins, num_size_templates, size_templates, box_loss_weight=1.0, corne
modules/frustum.py:12
Method__init__
(self, num_centers, radius, num_neighbors, in_channels, out_channels, include_coordinates=True)
modules/pointnet.py:50
Method__init__
(self, in_channels, out_channels)
modules/pointnet.py:97
Method__init__
(self)
metrics/ChamferDistancePytorch/chamfer2D/dist_chamfer_2D.py:67
Method__init__
(self)
metrics/ChamferDistancePytorch/chamfer5D/dist_chamfer_5D.py:69
Method__init__
(self)
metrics/ChamferDistancePytorch/chamfer3D/dist_chamfer_3D.py:70
Method__init__
(self, num_classes, embed_dim, use_att, dropout=0.1, extra_feature_channels=3, width_multipli
model/tiger.py:177
Method__init__
(self, hidden_size, frequency_embedding_size=256)
model/transformer_branch.py:23
Method__init__
(self, dim, num_heads=8, qkv_bias=False, attn_drop=0.2, proj_drop=0.2)
model/transformer_branch.py:75
Method__init__
(self, hidden_size, num_heads, mlp_ratio=4.0, **block_kwargs)
model/transformer_branch.py:126
Method__init__
(self, latent_size, hidden_size)
model/transformer_branch.py:151
Method__init__
(self, num_classes, embed_dim, use_att, dropout=0.1, extra_feature_channels=3, width_multipli
model/pvcnn_generation.py:174
Method__init__
(self, root_pc:str, root_views: str, cache: str, categories: list = ['chair'], split: str= 'val',
datasets/shapenet_data_sv.py:52
Method__init__
(self, cam_ext, cam_int)
datasets/shapenet_data_sv.py:229
Method__init__
(self, root_dir="data/ShapeNetCore.v2.PC15k", categories=['airplane'], tr_sample_size=10000,
datasets/shapenet_data_pc.py:202
Method__init__
(self, radius : float=10, elev: float =45, azim:float=315, )
datasets/shapenet_data_pc.py:241
Method__init__
(self, phase, data_root, category, n_pts)
datasets/partnet.py:107
Method__len__
Returns the length of the dataset.
datasets/shapenet_data_sv.py:156
Method__len__
(self)
datasets/shapenet_data_pc.py:154
Method__len__
(self)
datasets/partnet.py:177
Method__repr__
(self)
datasets/shapenet_data_sv.py:255
Method_basic_init
(module)
model/transformer_branch.py:209
Method_denoise
(self, data, t)
test_generation.py:295
Method_denoise
(self, data, t)
train_generation.py:424
Function_transform_
(m)
test_generation.py:506
Function_transform_
(m)
train_generation.py:587
Methodall_kl
(self, x0, clip_denoised=True)
test_generation.py:284
Functionavg_voxelize_backward
Function: average pool voxelization (backward) Args: grad_y : grad outputs, FloatTensor[b, c, s] indices: voxel index of each point, IntTens
modules/functional/src/voxelization/vox.cpp:54
Functionavg_voxelize_forward
Function: average pool voxelization (forward) Args: features: features, FloatTensor[b, c, n] coords : coords of each point, IntTensor[b, 3,
modules/functional/src/voxelization/vox.cpp:17
Methodbackward
(ctx, grad_output)
modules/functional/grouping.py:25
Methodbackward
(ctx, grad_output)
modules/functional/interpolatation.py:30
Methodbackward
:param ctx: :param grad_output: gradient of output, FloatTensor[B, C, R, R, R] :return: gradient of inputs, Float
modules/functional/voxelization.py:27
Methodbackward
:param ctx: :param grad_output: gradient of outputs, FloatTensor[B, C, N] :return: gradient of inputs, FloatTens
modules/functional/devoxelization.py:30
Methodbackward
(ctx, graddist1, graddist2, gradidx1, gradidx2)
metrics/ChamferDistancePytorch/chamfer5D/dist_chamfer_5D.py:51
Methodbackward
(ctx, graddist1, graddist2, gradidx1, gradidx2)
metrics/ChamferDistancePytorch/chamfer3D/dist_chamfer_3D.py:52
Functionball_query_forward
modules/functional/src/ball_query/ball_query.cpp:6
Functionchamfer_backward
metrics/ChamferDistancePytorch/chamfer2D/chamfer_cuda.cpp:22
Functionchamfer_backward
metrics/ChamferDistancePytorch/chamfer5D/chamfer_cuda.cpp:22
Functionchamfer_backward
metrics/ChamferDistancePytorch/chamfer3D/chamfer_cuda.cpp:22
Functionchamfer_forward
metrics/ChamferDistancePytorch/chamfer2D/chamfer_cuda.cpp:17
Functionchamfer_forward
metrics/ChamferDistancePytorch/chamfer5D/chamfer_cuda.cpp:17
Functionchamfer_forward
metrics/ChamferDistancePytorch/chamfer3D/chamfer_cuda.cpp:17
Functioncoverage
Computes the Coverage between two sets of point-clouds. Args: sample_pcs (numpy array SxKx3): the S point-clouds, each of K points that wi
utils/metrics.py:78
Functioncreate_pointnet_components
(blocks, in_channels, embed_dim, with_se=False, normalize=True, eps=0, width_mu
model/tiger.py:47
Functioncreate_pointnet_components
(blocks, in_channels, embed_dim, with_se=False, normalize=True, eps=0, width_mu
model/pvcnn_generation.py:46
Functiondiscretized_gaussian_log_likelihood
(x, *, means, log_scales)
test_generation.py:30
Functiondiscretized_gaussian_log_likelihood
(x, *, means, log_scales)
train_generation.py:77
Functionexport_to_obj
transform: f(vertices, faces) --> transformed (vertices, faces)
utils/visualize.py:26
Functionexport_to_obj_single
transform: f(vertices, faces) --> transformed (vertices, faces)
utils/visualize.py:43
Functionexport_to_pc_batch
(dir, pcs, colors=None)
utils/visualize.py:15
Methodextra_repr
(self)
modules/voxelization.py:27
Methodextra_repr
(self)
modules/ball_query.py:32
Methodextra_repr
(self)
modules/pointnet.py:45
Methodextra_repr
(self)
modules/pointnet.py:92
Methodforward
(self, x, y)
modules/loss.py:9
Methodforward
(self,x)
modules/shared_mlp.py:8
Methodforward
(self, inputs)
modules/shared_mlp.py:34
Methodforward
(self, features, coords)
modules/voxelization.py:16
Methodforward
(self,x)
modules/se.py:6
Methodforward
(self, inputs)
modules/se.py:18
← previousnext →101–200 of 261, ranked by callers