Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Zhiyuan-R/Tiger-Diffusion
/ functions
Functions
261 in github.com/Zhiyuan-R/Tiger-Diffusion
⨍
Functions
261
◇
Types & classes
46
Method
forward
(self,x)
modules/pvconv.py:12
Method
forward
(self, x)
modules/pvconv.py:39
Method
forward
(self, inputs)
modules/pvconv.py:90
Method
forward
(self, inputs)
modules/pvconv.py:126
Method
forward
(self, points_coords, centers_coords, temb, points_features=None)
modules/ball_query.py:16
Method
forward
(self, inputs, targets)
modules/frustum.py:27
Method
forward
(self, inputs)
modules/pointnet.py:32
Method
forward
(self, inputs)
modules/pointnet.py:80
Method
forward
(self, inputs)
modules/pointnet.py:101
Method
forward
:param ctx: :param features: features of points, FloatTensor[B, C, N] :param indices: neighbor indices of centers, IntTensor[
modules/functional/grouping.py:10
Method
forward
:param ctx: :param points_coords: coordinates of points, FloatTensor[B, 3, N] :param centers_coords: coordinates of centers,
modules/functional/interpolatation.py:10
Method
forward
: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
Method
forward
: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
Method
forward
Gather :param ctx: :param features: features of points, FloatTensor[B, C, N] :param indices: centers' indices in poin
modules/functional/sampling.py:12
Method
forward
(ctx, xyz1, xyz2)
metrics/PyTorchEMD/emd.py:7
Method
forward
(ctx, xyz1, xyz2)
metrics/ChamferDistancePytorch/chamfer2D/dist_chamfer_2D.py:27
Method
forward
(ctx, xyz1, xyz2)
metrics/ChamferDistancePytorch/chamfer5D/dist_chamfer_5D.py:29
Method
forward
(ctx, xyz1, xyz2)
metrics/ChamferDistancePytorch/chamfer3D/dist_chamfer_3D.py:30
Method
forward
(self, inputs, t)
model/tiger.py:259
Method
forward
(self, t)
model/transformer_branch.py:53
Method
forward
(self, x, pe=None, time=None, coor=None)
model/transformer_branch.py:99
Method
forward
(self, x, c, pe=None, time=None, coord = None)
model/transformer_branch.py:139
Method
forward
(self, x, c)
model/transformer_branch.py:160
Method
forward
(self, inputs, t)
model/pvcnn_generation.py:225
Method
forward_with_cfg
Forward pass of DiT, but also batches the unconditional forward pass for classifier-free guidance.
model/transformer_branch.py:247
Function
furthest_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
Function
furthest_point_sampling_forward
modules/functional/src/sampling/sampling.cpp:43
Function
gather_features_backward
modules/functional/src/sampling/sampling.cpp:25
Function
gather_features_forward
modules/functional/src/sampling/sampling.cpp:6
Function
get_constrain_function
:param target_shape_constraint: target voxels :return: constrained x
test_generation.py:361
Method
get_loss_iter
(self, data, noises=None)
test_generation.py:305
Function
grouping_backward
modules/functional/src/grouping/grouping.cpp:26
Function
grouping_forward
modules/functional/src/grouping/grouping.cpp:6
Function
huber_loss
(error, delta)
modules/functional/loss.py:13
Function
jsd_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
Function
kl_loss
(x, y)
modules/functional/loss.py:7
Function
logits_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
Function
meshwrite
Save a 3D mesh to a polygon .ply file.
utils/visualize.py:58
Function
minimum_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
Function
normal_kl
KL divergence between normal distributions parameterized by mean and log-variance.
test_generation.py:23
Function
plot_att
(att_tensor, T)
model/transformer_branch.py:62
Function
plot_stats
(output_dir, stats, interval)
utils/visualize.py:213
Method
prior_kl
(self, x0)
test_generation.py:281
Method
prior_kl
(self, x0)
train_generation.py:410
Function
project_pc_to_image
project point clouds into 2D image :param points: (n, 3) range(-1, 1) :return: binary image
datasets/partnet.py:10
Method
q_mean_variance
(self, x_start, t)
test_generation.py:107
Method
reconstruct
(self, x0, t, denoise_fn, noise_fn=torch.randn, constrain_fn=lambda x, t:x)
test_generation.py:230
Method
reconstruct
(self, x0, t, constrain_fn=lambda x, t:x)
test_generation.py:325
Method
renormalize
(self, mean, std)
datasets/shapenet_data_pc.py:146
Function
rotate
vertices: [numpoints, 3]
train_generation.py:34
Function
rotate_point_cloud_by_axis_angle
align 3depn shapes to shapenet coordinates
datasets/partnet.py:44
Function
set_global_gpu_env
(opt)
utils/file_utils.py:15
Function
three_nearest_neighbors_interpolate_backward
modules/functional/src/interpolate/neighbor_interpolate.cpp:42
Function
three_nearest_neighbors_interpolate_forward
modules/functional/src/interpolate/neighbor_interpolate.cpp:6
Method
train
(self)
test_generation.py:329
Function
trilinear_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
Function
trilinear_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
Function
visualize_pointcloud
r''' Visualizes point cloud data. Args: points (tensor): point data normals (tensor): normal data (if existing) out_file (
utils/visualize.py:148
Function
visualize_voxels
r''' Visualizes voxel data. show only first num_shown
utils/visualize.py:125
Function
weights_init
xavier initialization
train_generation.py:56
Function
write_ply
input: Nx3, write points to filename as PLY format.
datasets/partnet.py:28
← previous
201–261 of 261, ranked by callers