MCPcopy Create free account

hub / github.com/YuLiu-LY/ArtGS / functions

Functions339 in github.com/YuLiu-LY/ArtGS

⨍ Functions 339◇ Types & classes 54
↓ 15 callersFunctionquaternion_mul
(a: torch.Tensor, b: torch.Tensor)
utils/dual_quaternion.py:162
↓ 14 callersMethodextract
(self, args)
arguments/__init__.py:42
↓ 12 callersMethodsave
(self, iteration, is_best=False)
scene/__init__.py:70
↓ 9 callersFunctionindex_points
Input: points: input points data, [B, N, C] idx: sample index data, [B, S] Return: new_points:, indexed points data,
utils/pointnet2_utils.py:54
↓ 9 callersMethodwrite
(self, x)
utils/general_utils.py:196
↓ 7 callersMethod__init__
( self, in_size: int, out_size: int, *, activation
utils/pointnet_lib/pytorch_utils.py:202
↓ 6 callersFunctioncompute_recon_error
(recon_path, gt_path, n_samples=10000, vis=False)
utils/metrics.py:55
↓ 6 callersFunctionlinear
(in_features, out_features, bias=True, weight_init='xavier', gain=1., nonlinearity='relu')
scene/module.py:13
↓ 6 callersFunctionnormalize
(x)
utils/pose_utils.py:59
↓ 6 callersFunctionrender
Render the scene. Background tensor (bg_color) must be on GPU!
gaussian_renderer/__init__.py:22
↓ 6 callersMethodstep
(self, gaussians, is_training=True)
scene/deform_model.py:18
↓ 5 callersMethodget_slot_deform
(self)
scene/artgs.py:114
↓ 5 callersFunctionquaternion_conjugate
(q: torch.Tensor)
utils/dual_quaternion.py:105
↓ 5 callersFunctionsafe_state
(silent)
utils/general_utils.py:189
↓ 5 callersMethodsave_ply
(self, path)
scene/gaussian_model.py:213
↓ 5 callersMethodupdate
(self, iteration, *args, **kwargs)
scene/artgs.py:259
↓ 5 callersMethodupdate_learning_rate
(self, iteration)
scene/deform_model.py:55
↓ 4 callersMethod__init__
(self, reduction='batch-based')
utils/depth_loss.py:83
↓ 4 callersFunctiondual_quaternion_to_quaternion_translation
( dq: DualQuaternions, )
utils/dual_quaternion.py:281
↓ 4 callersFunctionfarthest_point_sample
Input: xyz: pointcloud data, [B, N, 3] npoint: number of samples Return: centroids: sampled pointcloud index, [B, npo
utils/pointnet2_utils.py:74
↓ 4 callersFunctiongetWorld2View2
(R, t, translate=np.array([.0, .0, .0]), scale=1.0)
utils/graphics_utils.py:42
↓ 4 callersMethodinit_from_file
(self, path)
scene/artgs.py:51
↓ 4 callersMethodload_ply
(self, path, og_number_points=-1)
scene/gaussian_model.py:240
↓ 4 callersFunctionquaternion_apply
Apply the rotation given by a quaternion to a 3D point. Usual torch rules for broadcasting apply. Args: quaternion: Tensor of qu
utils/dual_quaternion.py:213
↓ 4 callersFunctionvis_depth
(depth, save_path=None)
utils/general_utils.py:209
↓ 3 callersFunctionSH2RGB
(sh)
utils/sh_utils.py:119
↓ 3 callersMethod__init__
:param mlp: list of int, spec of the pointnet before the global max_pool :param npoint: int, number of features :param radius
utils/pointnet_lib/pointnet2_modules.py:98
↓ 3 callersMethod__init__
(self, parser: ArgumentParser, name: str, fill_none=False)
arguments/__init__.py:22
↓ 3 callersMethodadd_densification_stats
(self, viewspace_point_tensor, update_filter)
scene/gaussian_model.py:493
↓ 3 callersMethodbackward
(ctx, a=None, b=None)
utils/pointnet_lib/pointnet2_utils.py:105
↓ 3 callersFunctionbuild_rotation
(r)
utils/general_utils.py:141
↓ 3 callersMethodcreate_from_pcd
(self, pcd: BasicPointCloud, spatial_lr_scale: float=5., print_info=True, max_point_num=150_000)
scene/gaussian_model.py:129
↓ 3 callersMethoddensify_and_prune
(self, max_grad, min_opacity, extent, max_screen_size)
scene/gaussian_model.py:473
↓ 3 callersFunctiondual_quaternion_q_conjugate
(dq: DualQuaternions)
utils/dual_quaternion.py:325
↓ 3 callersFunctionfov2focal
(fov, pixels)
utils/graphics_utils.py:79
↓ 3 callersMethodgetTrainCameras_end
(self, scale=1.0)
scene/__init__.py:92
↓ 3 callersMethodgetTrainCameras_start
(self, scale=1.0)
scene/__init__.py:89
↓ 3 callersMethodget_mask
(self, x, is_training=False)
scene/artgs.py:155
↓ 3 callersFunctiongumbel_softmax
(logits, tau=1., hard=True, dim=-1, is_training=False)
scene/module.py:55
↓ 3 callersFunctionl1_loss
(network_output, gt)
utils/loss_utils.py:18
↓ 3 callersMethodload_weights
(self, model_path, iteration=-1)
scene/deform_model.py:43
↓ 3 callersMethodoneupSHdegree
(self)
scene/gaussian_model.py:125
↓ 3 callersMethodreset_opacity
(self)
scene/gaussian_model.py:235
↓ 3 callersFunctionssim
(img1, img2, window_size=11, size_average=True)
utils/loss_utils.py:45
↓ 3 callersFunctionstorePly
(path, xyz, rgb)
data_tools/process_utils.py:16
↓ 3 callersFunctiontoOpen3dCloud
(points, colors=None, normals=None)
utils/geo_utils.py:26
↓ 3 callersMethodtraining_setup
(self, training_args)
scene/gaussian_model.py:165
↓ 2 callersMethod__init__
(self, d_in, d_out, d_hid, num_layers)
scene/module.py:43
↓ 2 callersMethod__init__
(self, npoint, radius, nsample, in_channel, mlp, group_all)
utils/pointnet2_utils.py:195
↓ 2 callersMethod__init__
(self, use_xyz: bool = True)
utils/pointnet_lib/pointnet2_utils.py:311
↓ 2 callersFunctioncameraList_from_camInfos
(cam_infos, resolution_scale, args, flow_dirs_list=None)
utils/camera_utils.py:87
↓ 2 callersFunctioncompute_pcd
(glcam_in_worlds, K, rgbs=None, depths=None, masks=None, cluster=False, eps=0.075, mi
utils/geo_utils.py:96
↓ 2 callersMethoddeform_pts
(self, xc, mask, slot_qr, slot_qd, state)
scene/artgs.py:141
↓ 2 callersMethoddensification_postfix
(self, new_xyz, new_features_dc, new_features_rest, new_opacities, new_scaling,
scene/gaussian_model.py:396
↓ 2 callersFunctiondual_quaternion_apply
(dq: DualQuaternions, point: torch.Tensor)
utils/dual_quaternion.py:291
↓ 2 callersFunctiondual_quaternion_inverse
(dq: DualQuaternions)
utils/dual_quaternion.py:345
↓ 2 callersFunctioneval_axis_and_state
(axis_a, axis_b, joint_type='r', reverse=False)
utils/metrics.py:161
↓ 2 callersFunctioneval_axis_and_state_all
(pred_joint_list, pred_joint_types, gt_info_list, print_perm=False)
utils/metrics.py:189
↓ 2 callersMethodextract_mesh
(self)
utils/mesh_utils.py:162
↓ 2 callersFunctionfetchPly
(path)
scene/dataset_readers.py:88
↓ 2 callersFunctionfind_biggest_cluster
(pts, eps=0.005, min_samples=5)
utils/geo_utils.py:53
↓ 2 callersFunctionfov2focal
(fov, pixels)
data_tools/process_utils.py:140
↓ 2 callersMethodgetTrainCameras
(self, scale=1.0)
scene/__init__.py:101
↓ 2 callersFunctionget_combined_args
(parser: ArgumentParser)
arguments/__init__.py:128
↓ 2 callersFunctionget_expon_lr_func
Copied from Plenoxels Continuous learning rate decay function. Adapted from JaxNeRF The returned rate is lr_init when step=0 and lr_
utils/general_utils.py:53
↓ 2 callersMethodget_joint_param
(self, joint_type_list)
scene/artgs.py:162
↓ 2 callersFunctionget_linear_noise_func
Copied from Plenoxels Continuous learning rate decay function. Adapted from JaxNeRF The returned rate is lr_init when step=0 and lr_
utils/general_utils.py:89
↓ 2 callersFunctionget_rank
()
scene/module.py:175
↓ 2 callersFunctioninverse_sigmoid
(x)
utils/general_utils.py:23
↓ 2 callersFunctionload_mesh
(path)
utils/metrics.py:28
↓ 2 callersMethodload_ply_cano
(self, path, state=0.5)
scene/gaussian_model.py:294
↓ 2 callersFunctionmatch_pcd
Input: pc0, pc1: tensor [1, N0, 3], [1, N1, 3] N: downsample number Return: idx_s, idx_e: [N], [N]
utils/other_utils.py:34
↓ 2 callersMethodone_transform
(self, gaussians:GaussianModel, state, is_training)
scene/artgs.py:197
↓ 2 callersFunctionprepare_output_and_logger
(args, use_tensorboard=True)
utils/log_utils.py:22
↓ 2 callersMethodprune_points
(self, mask)
scene/gaussian_model.py:351
↓ 2 callersFunctionquaternion_to_axis_angle
Convert rotations given as quaternions to axis/angle. Args: quaternions: quaternions with real part first, as tensor of
utils/dual_quaternion.py:350
↓ 2 callersFunctionquaternion_to_matrix
Convert rotations given as quaternions to rotation matrices. Args: quaternions: quaternions with real part first, as ten
utils/dual_quaternion.py:175
↓ 2 callersFunctionquaternion_translation_apply
( q: torch.Tensor, t: torch.Tensor, point: torch.Tensor )
utils/dual_quaternion.py:234
↓ 2 callersFunctionquery_ball_point
Input: radius: local region radius nsample: max sample number in local region xyz: all points, [B, N, 3] new_xyz:
utils/pointnet2_utils.py:102
↓ 2 callersFunctionreadCamerasFromTransforms
(path, transformsfile, white_background, extension=".png", no_bg=False, load_depth=True, load_mono_depth=True)
scene/dataset_readers.py:116
↓ 2 callersFunctionread_gt
(gt_path)
utils/metrics.py:268
↓ 2 callersFunctionrender_set
(args, load2gpt_on_the_fly, name, iteration, views, gaussians, pipeline, background, deform, eval_app=False, v
render.py:186
↓ 2 callersMethodreset_extrinsic
(self, R, T)
scene/cameras.py:63
↓ 2 callersFunctionrev_pixel
(pixel, H)
utils/camera_utils.py:74
↓ 2 callersFunctionsave_axis_mesh
support rotate only for now
vis_utils/canonicalize_mesh.py:8
↓ 2 callersMethodsave_weights
(self, model_path, iteration, is_best=False)
scene/deform_model.py:31
↓ 2 callersFunctionsearchForMaxIteration
(folder)
utils/system_utils.py:28
↓ 2 callersMethodslotdq_to_gsdq
(self, slot_qr, slot_qd, mask)
scene/artgs.py:108
↓ 2 callersFunctionsquare_distance
Calculate Euclid distance between each two points. src^T * dst = xn * xm + yn * ym + zn * zm； sum(src^2, dim=-1) = xn*xn + yn*yn + zn*zn
utils/pointnet2_utils.py:30
↓ 2 callersMethodtrain_setting
(self, training_args)
scene/deform_model.py:21
↓ 2 callersMethodupdate
(self, iteration)
scene/deform_model.py:64
↓ 2 callersFunctionviewmatrix
(z, up, pos)
utils/pose_utils.py:70
↓ 1 callersFunctionPILtoTorch
(pil_image, resolution)
utils/general_utils.py:27
↓ 1 callersFunctionRGB2SH
(rgb)
utils/sh_utils.py:115
↓ 1 callersFunctionR_from_direction_angle
(direction, theta)
vis_utils/demo_gen_glb.py:90
↓ 1 callersMethod__init__
(self, sh_degree: int, fea_dim=0, with_motion_mask=False, dynamic_threshold_ratio=0.02)
scene/gaussian_model.py:27
↓ 1 callersMethod__init__
(self, args)
scene/artgs.py:73
↓ 1 callersMethod_prune_optimizer
(self, mask)
scene/gaussian_model.py:334
↓ 1 callersFunction_quaternion_conjugate
https://mathworld.wolfram.com/QuaternionConjugate.html when q is unit quaternion, inv(q) = conjugate(q)
utils/dual_quaternion.py:97
↓ 1 callersFunction_quaternion_mul
Multiply two quaternions. Usual torch rules for broadcasting apply. Args: a: Quaternions as tensor of shape (..., 4), real part
utils/dual_quaternion.py:141
next →1–100 of 339, ranked by callers