MCPcopy Create free account

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

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

Methodforward
(self,x)
modules/pvconv.py:12
Methodforward
(self, x)
modules/pvconv.py:39
Methodforward
(self, inputs)
modules/pvconv.py:90
Methodforward
(self, inputs)
modules/pvconv.py:126
Methodforward
(self, points_coords, centers_coords, temb, points_features=None)
modules/ball_query.py:16
Methodforward
(self, inputs, targets)
modules/frustum.py:27
Methodforward
(self, inputs)
modules/pointnet.py:32
Methodforward
(self, inputs)
modules/pointnet.py:80
Methodforward
(self, inputs)
modules/pointnet.py:101
Methodforward
:param ctx: :param features: features of points, FloatTensor[B, C, N] :param indices: neighbor indices of centers, IntTensor[
modules/functional/grouping.py:10
Methodforward
:param ctx: :param points_coords: coordinates of points, FloatTensor[B, 3, N] :param centers_coords: coordinates of centers,
modules/functional/interpolatation.py:10
Methodforward
:param ctx: :param features: Features of the point cloud, FloatTensor[B, C, N] :param coords: Voxelized Coordinates of each p
modules/functional/voxelization.py:10
Methodforward
:param ctx: :param coords: the coordinates of points, FloatTensor[B, 3, N] :param features: FloatTensor[B, C, R, R, R]
modules/functional/devoxelization.py:10
Methodforward
Gather :param ctx: :param features: features of points, FloatTensor[B, C, N] :param indices: centers' indices in poin
modules/functional/sampling.py:12
Methodforward
(ctx, xyz1, xyz2)
metrics/PyTorchEMD/emd.py:7
Methodforward
(ctx, xyz1, xyz2)
metrics/ChamferDistancePytorch/chamfer2D/dist_chamfer_2D.py:27
Methodforward
(ctx, xyz1, xyz2)
metrics/ChamferDistancePytorch/chamfer5D/dist_chamfer_5D.py:29
Methodforward
(ctx, xyz1, xyz2)
metrics/ChamferDistancePytorch/chamfer3D/dist_chamfer_3D.py:30
Methodforward
(self, inputs, t)
model/tiger.py:259
Methodforward
(self, t)
model/transformer_branch.py:53
Methodforward
(self, x, pe=None, time=None, coor=None)
model/transformer_branch.py:99
Methodforward
(self, x, c, pe=None, time=None, coord = None)
model/transformer_branch.py:139
Methodforward
(self, x, c)
model/transformer_branch.py:160
Methodforward
(self, inputs, t)
model/pvcnn_generation.py:225
Methodforward_with_cfg
Forward pass of DiT, but also batches the unconditional forward pass for classifier-free guidance.
model/transformer_branch.py:247
Functionfurthest_point_sample
Uses iterative furthest point sampling to select a set of npoint features that have the largest minimum distance to the sampled point set
modules/functional/sampling.py:37
Functionfurthest_point_sampling_forward
modules/functional/src/sampling/sampling.cpp:43
Functiongather_features_backward
modules/functional/src/sampling/sampling.cpp:25
Functiongather_features_forward
modules/functional/src/sampling/sampling.cpp:6
Functionget_constrain_function
:param target_shape_constraint: target voxels :return: constrained x
test_generation.py:361
Methodget_loss_iter
(self, data, noises=None)
test_generation.py:305
Functiongrouping_backward
modules/functional/src/grouping/grouping.cpp:26
Functiongrouping_forward
modules/functional/src/grouping/grouping.cpp:6
Functionhuber_loss
(error, delta)
modules/functional/loss.py:13
Functionjsd_between_point_cloud_sets
Computes the JSD between two sets of point-clouds, as introduced in the paper ```Learning Representations And Generative Models For 3D Point Clouds```
utils/metrics.py:142
Functionkl_loss
(x, y)
modules/functional/loss.py:7
Functionlogits_mask
Use logits to sample points :param coords: coords of points, FloatTensor[B, 3, N] :param logits: binary classification logits, FloatTenso
modules/functional/sampling.py:51
Functionmeshwrite
Save a 3D mesh to a polygon .ply file.
utils/visualize.py:58
Functionminimum_mathing_distance
Computes the MMD between two sets of point-clouds. Args: sample_pcs (numpy array SxKx3): the S point-clouds, each of K points that will be
utils/metrics.py:33
Functionnormal_kl
KL divergence between normal distributions parameterized by mean and log-variance.
test_generation.py:23
Functionplot_att
(att_tensor, T)
model/transformer_branch.py:62
Functionplot_stats
(output_dir, stats, interval)
utils/visualize.py:213
Methodprior_kl
(self, x0)
test_generation.py:281
Methodprior_kl
(self, x0)
train_generation.py:410
Functionproject_pc_to_image
project point clouds into 2D image :param points: (n, 3) range(-1, 1) :return: binary image
datasets/partnet.py:10
Methodq_mean_variance
(self, x_start, t)
test_generation.py:107
Methodreconstruct
(self, x0, t, denoise_fn, noise_fn=torch.randn, constrain_fn=lambda x, t:x)
test_generation.py:230
Methodreconstruct
(self, x0, t, constrain_fn=lambda x, t:x)
test_generation.py:325
Methodrenormalize
(self, mean, std)
datasets/shapenet_data_pc.py:146
Functionrotate
vertices: [numpoints, 3]
train_generation.py:34
Functionrotate_point_cloud_by_axis_angle
align 3depn shapes to shapenet coordinates
datasets/partnet.py:44
Functionset_global_gpu_env
(opt)
utils/file_utils.py:15
Functionthree_nearest_neighbors_interpolate_backward
modules/functional/src/interpolate/neighbor_interpolate.cpp:42
Functionthree_nearest_neighbors_interpolate_forward
modules/functional/src/interpolate/neighbor_interpolate.cpp:6
Methodtrain
(self)
test_generation.py:329
Functiontrilinear_devoxelize_backward
Function: trilinear devoxelization (backward) Args: grad_y : grad outputs, FloatTensor[b, c, n] indices : the voxel coordinates of point cu
modules/functional/src/interpolate/trilinear_devox.cpp:67
Functiontrilinear_devoxelize_forward
Function: trilinear devoxelization (forward) Args: r : voxel resolution trainig : whether is training mode coords : the coordi
modules/functional/src/interpolate/trilinear_devox.cpp:18
Functionvisualize_pointcloud
r''' Visualizes point cloud data. Args: points (tensor): point data normals (tensor): normal data (if existing) out_file (
utils/visualize.py:148
Functionvisualize_voxels
r''' Visualizes voxel data. show only first num_shown
utils/visualize.py:125
Functionweights_init
xavier initialization
train_generation.py:56
Functionwrite_ply
input: Nx3, write points to filename as PLY format.
datasets/partnet.py:28
← previous201–261 of 261, ranked by callers