MCPcopy Create free account

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

Functions339 in github.com/YuLiu-LY/ArtGS

⨍ Functions 339◇ Types & classes 54
↓ 1 callersFunction_sqrt_positive_part
Returns torch.sqrt(torch.max(0, x)) but with a zero subgradient where x is 0.
utils/dual_quaternion.py:29
↓ 1 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average=True)
utils/loss_utils.py:56
↓ 1 callersFunctionanimate_joint
(joint_info, obj_name, r_range=[-0.3, 0.3], p_range=[-0.05, 0.10], n_frames=100)
vis_utils/demo_gen_glb.py:103
↓ 1 callersFunctionbuild_scaling_rotation
(s, r)
utils/general_utils.py:165
↓ 1 callersFunctionbuild_scaling_rotation_inverse
(s, r)
utils/general_utils.py:177
↓ 1 callersMethodcal_art_type
(self)
scene/artgs.py:98
↓ 1 callersFunctioncal_cluster_centers
(cano_path, num_slots, visualize=False)
utils/other_utils.py:161
↓ 1 callersMethodcal_relative_pos
(self, x)
scene/artgs.py:57
↓ 1 callersFunctioncanonicalize_mesh
(meta, src_root, dst_root, obj_name)
vis_utils/canonicalize_mesh.py:56
↓ 1 callersMethodcat_tensors_to_optimizer
(self, tensors_dict)
scene/gaussian_model.py:370
↓ 1 callersFunctioncompute_chamfer
(recon_pts, gt_pts)
utils/metrics.py:46
↓ 1 callersFunctioncompute_scale_and_shift
(prediction, target, mask)
utils/depth_loss.py:7
↓ 1 callersFunctioncompute_translation_scales
(pts, max_dim=2, cluster=True, eps=0.005, min_samples=5)
utils/geo_utils.py:63
↓ 1 callersMethodconstruct_list_of_attributes
(self)
scene/gaussian_model.py:197
↓ 1 callersMethodcosine_anneal
(self, step, final_step, start_step=0, start_value=1.0, final_value=0.1)
scene/artgs.py:263
↓ 1 callersFunctioncreate_window
(window_size, channel)
utils/loss_utils.py:38
↓ 1 callersMethoddensify_and_clone
(self, grads=None, grad_threshold=None, scene_extent=None, selected_pts_mask=None)
scene/gaussian_model.py:454
↓ 1 callersMethoddensify_and_split
(self, grads=None, grad_threshold=None, scene_extent=None, N=2, selected_pts_mask=None, without_prune=False)
scene/gaussian_model.py:423
↓ 1 callersFunctiondepth2xyzmap
(depth, K)
utils/geo_utils.py:38
↓ 1 callersFunctiondual_quaternion_d_conjugate
(dq: DualQuaternions)
utils/dual_quaternion.py:332
↓ 1 callersFunctiondual_quaternion_mul
(dq1: DualQuaternions, dq2: DualQuaternions)
utils/dual_quaternion.py:315
↓ 1 callersFunctioneval_CD_2states
(gt_path, pred_path, id_d_list, n_trials=1)
utils/metrics.py:88
↓ 1 callersFunctionexport_axis_mesh
(arti, exp_dir, i)
vis_utils/canonicalize_mesh.py:49
↓ 1 callersFunctionexport_joint_info_json
(pred_joint_list, mesh_files, exp_dir)
render.py:149
↓ 1 callersFunctionexport_urdf
(input_dir, output_dir, scene, larger_motion_state)
vis_utils/json2urdf.py:5
↓ 1 callersMethodextract_mesh_bounded
Perform TSDF fusion given a fixed depth range, used in the paper. voxel_size: the voxel size of the volume sdf_trunc
utils/mesh_utils.py:180
↓ 1 callersFunctionfocus_point_fn
Calculate nearest point to all focal axes in poses.
utils/mesh_utils.py:74
↓ 1 callersFunctiongaussian
(window_size, sigma)
utils/loss_utils.py:33
↓ 1 callersFunctiongenerate
(views, args, N=30)
render_video.py:99
↓ 1 callersFunctiongenerate
(views)
render.py:88
↓ 1 callersFunctiongenerate_camera_poses
Generate camera poses around the scene center on a circle. Parameters: - r: Radius of the circle. - theta: Elevation angle in d
render_video.py:30
↓ 1 callersFunctiongenerate_camera_poses
Generate camera poses around the scene center on a circle. Parameters: - r: Radius of the circle. - theta: Elevation angle in d
render.py:32
↓ 1 callersFunctiongenerate_pcd
(scene, K, poses_start, poses_end, num_slots, reprocess=False, cluster=False, visualize=False)
data_tools/process_utils.py:34
↓ 1 callersFunctiongenerate_video
(imgs, video_name, fps=15, brighten=False)
render_video.py:115
↓ 1 callersFunctiongetNerfppNorm
(cam_info, apply=False)
scene/dataset_readers.py:51
↓ 1 callersFunctiongetProjectionMatrix
(znear, zfar, fovX, fovY)
utils/graphics_utils.py:56
↓ 1 callersMethodgetTestCameras
(self, scale=1.0)
scene/__init__.py:104
↓ 1 callersMethodget_attn
(self, slots, k, tau=1.0, gumbel=False, shift=None, is_training=False)
scene/module.py:160
↓ 1 callersFunctionget_center_and_diag
(cam_centers)
scene/dataset_readers.py:52
↓ 1 callersMethodget_covariance
(self, d_rot, scaling_modifier=1)
scene/gaussian_model.py:116
↓ 1 callersFunctionget_rotation_axis_angle
Rodrigues' rotation formula args: * k: direction unit vector of the axis to rotate about * theta: the (radian) angle to rotate wi
render.py:104
↓ 1 callersFunctionget_rotation_axis_angle
Rodrigues' rotation formula args: * k: direction vector of the axis to rotate about * theta: the (radian) angle to rotate with re
vis_utils/canonicalize_mesh.py:24
↓ 1 callersFunctiongradient_loss
(prediction, target, mask, reduction=reduction_batch_based)
utils/depth_loss.py:62
↓ 1 callersFunctiongru_cell
(input_size, hidden_size, bias=True)
scene/module.py:28
↓ 1 callersMethodinit_deform
(self,)
train.py:80
↓ 1 callersMethodinit_deform
(self,)
train_predict.py:61
↓ 1 callersMethodinterpolate
(self, gaussians: GaussianModel, time_list)
scene/artgs.py:242
↓ 1 callersMethoditer
(self, slots, k, v, tau=1.0, gumbel=False, shift=None, is_training=False)
scene/module.py:128
↓ 1 callersFunctionline_distance
(a_o, a_d, b_o, b_d)
utils/metrics.py:152
↓ 1 callersMethodload2device
(self, data_device='cuda')
scene/cameras.py:69
↓ 1 callersFunctionloadCam
(args, id, cam_info, resolution_scale, flow_dirs)
utils/camera_utils.py:23
↓ 1 callersFunctionload_joint_info
(meta)
vis_utils/demo_gen_glb.py:67
↓ 1 callersFunctionload_mesh_with_material
(src_root, meta, optimize=False)
vis_utils/demo_gen_glb.py:27
↓ 1 callersFunctionmatch_gaussians
(path, cano_gs, num_slots, visualize=False)
utils/other_utils.py:57
↓ 1 callersFunctionmatrix_to_quaternion
Convert rotations given as rotation matrices to quaternions. Args: matrix: Rotation matrices as tensor of shape (..., 3, 3). Re
utils/dual_quaternion.py:41
↓ 1 callersFunctionmkdir_p
(folder_path)
utils/system_utils.py:17
↓ 1 callersFunctionmse
(img1, img2)
utils/metrics.py:21
↓ 1 callersFunctionmse_loss
(prediction, target, mask, reduction=reduction_batch_based)
utils/depth_loss.py:53
↓ 1 callersFunctionnormalize
(v)
vis_utils/canonicalize_mesh.py:20
↓ 1 callersFunctionnormalize_dualquaternion
(qr: torch.Tensor, qd: torch.Tensor)
utils/dual_quaternion.py:134
↓ 1 callersFunctionoptimize_mesh
(obj, decimate_ratio=0.1)
vis_utils/demo_gen_glb.py:18
↓ 1 callersFunctionpad_poses
Pad [..., 3, 4] pose matrices with a homogeneous bottom row [0,0,0,1].
utils/mesh_utils.py:83
↓ 1 callersFunctionposes_avg
(poses)
utils/pose_utils.py:78
↓ 1 callersFunctionpost_process_mesh
Post-process a mesh to filter out floaters and disconnected parts
utils/mesh_utils.py:20
↓ 1 callersFunctionprocess
(src_root, dst_root, obj_name)
vis_utils/canonicalize_mesh.py:100
↓ 1 callersFunctionpsnr
(img1, img2)
utils/metrics.py:24
↓ 1 callersFunctionquaternion_translation_inverse
( q: torch.Tensor, t: torch.Tensor )
utils/dual_quaternion.py:241
↓ 1 callersFunctionquaternion_translation_to_se3
(q: torch.Tensor, t: torch.Tensor)
utils/dual_quaternion.py:264
↓ 1 callersFunctionrender_set
(args, name, iteration, views, gaussians, pipeline, background, deform, N_frames=60, inverse=False, recenter=F
render_video.py:141
↓ 1 callersFunctionrender_sets
(args, dataset: ModelParams, iteration, pipeline: PipelineParams, N_frames=30)
render_video.py:191
↓ 1 callersFunctionrender_sets
(args, dataset: ModelParams, iteration, pipeline: PipelineParams, skip_train: bool, skip_test: bool, mode: str
render.py:351
↓ 1 callersMethodreplace_tensor_to_optimizer
(self, tensor, name)
scene/gaussian_model.py:316
↓ 1 callersFunctionreset
(obj)
vis_utils/demo_gen_glb.py:12
↓ 1 callersFunctionsample_and_group
Input: npoint: radius: nsample: xyz: input points position data, [B, N, 3] points: input points data, [B,
utils/pointnet2_utils.py:143
↓ 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
utils/pointnet2_utils.py:174
↓ 1 callersMethodsave_2gs
(self, iteration, num_slots, vis_cano=False, vis_center=False)
scene/__init__.py:80
↓ 1 callersFunctionsave_axis_mesh
support rotate only for now
render.py:131
↓ 1 callersFunctionstorePly
(path, xyz, rgb)
scene/dataset_readers.py:98
↓ 1 callersFunctionstrip_lowerdiag
(L)
utils/general_utils.py:125
↓ 1 callersFunctionstrip_symmetric
(sym)
utils/general_utils.py:137
↓ 1 callersFunctionto_cam_open3d
(viewpoint_stack)
utils/mesh_utils.py:46
↓ 1 callersMethodtrain
(self, iters=5000)
train.py:101
↓ 1 callersMethodtrain
(self, iters=5000)
train_predict.py:74
↓ 1 callersMethodtrain
(self, iters=5000)
train_coarse.py:65
↓ 1 callersMethodtrain_step
(self)
train.py:105
↓ 1 callersMethodtrain_step
(self)
train_predict.py:78
↓ 1 callersMethodtrain_step
(self)
train_coarse.py:69
↓ 1 callersMethodtrainable_parameters
(self)
scene/artgs.py:148
↓ 1 callersFunctiontraining_report
(tb_writer, iteration, Ll1, depth_loss, mono_depth_loss, loss, elapsed, scene: Scene, joint_metrics)
utils/log_utils.py:43
↓ 1 callersFunctionunpad_poses
Remove the homogeneous bottom row from [..., 4, 4] pose matrices.
utils/mesh_utils.py:89
↓ 1 callersMethodupdate_step
(self, global_step)
scene/module.py:228
↓ 1 callersFunctionvideo2gif
(video_file)
render_video.py:134
FunctionArrayToTorch
(array, resolution)
utils/general_utils.py:43
FunctionPYBIND11_MODULE
utils/pointnet_lib/src/pointnet2_api.cpp:11
Method__get_prediction_ssi
(self)
utils/depth_loss.py:139
Method__init__
(self, args, dataset, opt, pipe, saving_iterations)
train.py:32
Method__init__
(self, args, dataset, opt, pipe)
train_predict.py:29
Method__init__
(self, args, dataset, opt, pipe)
train_coarse.py:29
Method__init__
(self, sh_degree: int, fea_dim=0, with_motion_mask=True, all_the_same=False)
scene/gaussian_model.py:499
Method__init__
b :param path: Path to colmap scene main folder.
scene/__init__.py:24
← previousnext →101–200 of 339, ranked by callers