MCPcopy Create free account

hub / github.com/OKIC-CA/3DGS / functions

Functions207 in github.com/OKIC-CA/3DGS

⨍ Functions 207◇ Types & classes 41
↓ 16 callersMethodwrite
(self, x)
utils/general_utils.py:113
↓ 13 callersFunctionread_next_bytes
Read and unpack the next bytes from a binary file. :param fid: :param num_bytes: Sum of combination of {2, 4, 8}, e.g. 2, 6, 16, 30, etc.
scene/colmap_loader.py:61
↓ 6 callersMethodextract
(self, args)
arguments/__init__.py:30
↓ 5 callersFunctionfocal2fov
(focal, pixels)
utils/graphics_utils.py:86
↓ 5 callersFunctionrender
(viewpoint_camera, pc : GaussianModel, pipe, bg_color : torch.Tensor, scaling_modifier = 1.0, override_color =
gaussian_renderer/__init__.py:9
↓ 5 callersMethodsave
(self, iteration)
scene/__init__.py:76
↓ 4 callersMethod__init__
(self, reduction='image-based')
utils/loss_utils.py:87
↓ 4 callersMethod__init__
(self)
lpipsPyTorch/modules/networks.py:89
↓ 4 callersMethodgetTrainCameras
(self, scale=1.0)
scene/__init__.py:80
↓ 3 callersFunctionStringPrintf
submodules/Propagation_SSIM/PatchMatch.cpp:57
↓ 3 callersMethod__init__
(self, parser: ArgumentParser, name : str, fill_none = False)
arguments/__init__.py:10
↓ 3 callersFunctionbuild_rotation
(r)
utils/general_utils.py:73
↓ 3 callersMethoddensification_postfix
(self, new_xyz, new_features_dc, new_features_rest, new_opacities, new_scaling, new_rotation)
scene/gaussian_model.py:318
↓ 3 callersFunctionfov2focal
(fov, pixels)
utils/graphics_utils.py:83
↓ 3 callersFunctioninverse_sigmoid
(x)
utils/general_utils.py:13
↓ 3 callersMethodset_requires_grad
(self, state: bool)
lpipsPyTorch/modules/networks.py:46
↓ 2 callersFunctionRGB2SH
(rgb)
utils/sh_utils.py:114
↓ 2 callersFunctionReadCamera
submodules/Propagation_SSIM/PatchMatch.cpp:129
↓ 2 callersFunctionSH2RGB
(sh)
utils/sh_utils.py:117
↓ 2 callersFunctioncameraList_from_camInfos
(cam_infos, resolution_scale, args)
utils/camera_utils.py:62
↓ 2 callersMethodconv_gauss
(self, img)
utils/loss_utils.py:219
↓ 2 callersFunctioncpu_deep_copy_tuple
(input_tuple)
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:17
↓ 2 callersFunctiondepth_propagation
(viewpoint_cam, rendered_depth, viewpoint_stack, src_idxs, dataset, patch_size)
utils/graphics_utils.py:387
↓ 2 callersFunctionfetchPly
(path)
scene/dataset_readers.py:125
↓ 2 callersFunctiongetNerfppNorm
(cam_info)
scene/dataset_readers.py:39
↓ 2 callersMethodgetTestCameras
(self, scale=1.0)
scene/__init__.py:83
↓ 2 callersFunctiongetWorld2View2
(R, t, translate=np.array([.0, .0, .0]), scale=1.0)
utils/graphics_utils.py:47
↓ 2 callersFunctionl1_loss
(network_output, gt)
utils/loss_utils.py:6
↓ 2 callersMethodlaplacian_kernel
(self, current)
utils/loss_utils.py:227
↓ 2 callersMethodprune_points
(self, mask)
scene/gaussian_model.py:280
↓ 2 callersFunctionpsnr
(img1, img2)
utils/image_utils.py:6
↓ 2 callersFunctionqvec2rotmat
(qvec)
scene/colmap_loader.py:32
↓ 2 callersFunctionreadCamerasFromTransforms
(path, transformsfile, white_background, extension=".png", is_train=True)
scene/dataset_readers.py:206
↓ 2 callersFunctionreadDepthDmb
(file_path)
utils/general_utils.py:180
↓ 2 callersFunctionrender_set
(model_path, name, iteration, views, gaussians, pipeline, background)
render.py:15
↓ 2 callersFunctionsafe_state
(silent)
utils/general_utils.py:107
↓ 2 callersFunctionsave_image
(image, iteration, output_path)
train.py:29
↓ 2 callersFunctionssim
(img1, img2, window_size=11, size_average=True, mask=None)
utils/loss_utils.py:22
↓ 2 callersFunctionstorePly
(path, xyz, rgb)
scene/dataset_readers.py:134
↓ 2 callersMethodtraining_setup
(self, training_args)
scene/gaussian_model.py:138
↓ 1 callersFunctionPILtoTorch
(pil_image, resolution)
utils/general_utils.py:16
↓ 1 callersFunctionStringAppendV
submodules/Propagation_SSIM/PatchMatch.cpp:7
↓ 1 callersMethod_prune_optimizer
(self, mask)
scene/gaussian_model.py:262
↓ 1 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average=True, mask=None)
utils/loss_utils.py:32
↓ 1 callersMethodadd_densification_stats
(self, viewspace_point_tensor, update_filter)
scene/gaussian_model.py:394
↓ 1 callersMethodbackward
(ctx, grad_out_color, _)
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:101
↓ 1 callersFunctionbilinear_sampler
Wrapper for grid_sample, uses pixel coordinates
utils/graphics_utils.py:262
↓ 1 callersFunctionbuild_scaling_rotation
(s, r)
utils/general_utils.py:96
↓ 1 callersFunctioncamera_to_JSON
(id, camera : Camera)
utils/camera_utils.py:70
↓ 1 callersMethodcapture
(self)
scene/gaussian_model.py:50
↓ 1 callersMethodcat_tensors_to_optimizer
(self, tensors_dict)
scene/gaussian_model.py:296
↓ 1 callersFunctioncheck_geometric_consistency
(depth_ref, intrinsics_ref, extrinsics_ref, depth_src, intrinsics_src, extrinsics_src, thre1=1, thre2=0.01)
utils/graphics_utils.py:367
↓ 1 callersFunctioncompute_scale_and_shift
(prediction, target, mask)
utils/loss_utils.py:140
↓ 1 callersMethodconstruct_list_of_attributes
(self)
scene/gaussian_model.py:166
↓ 1 callersMethodcreate_from_pcd
(self, pcd : BasicPointCloud, spatial_lr_scale : float)
scene/gaussian_model.py:113
↓ 1 callersFunctioncreate_window
(window_size, channel)
utils/loss_utils.py:16
↓ 1 callersMethoddensify_and_clone
(self, grads, grad_threshold, scene_extent)
scene/gaussian_model.py:363
↓ 1 callersMethoddensify_and_prune
(self, max_grad, min_opacity, extent, max_screen_size)
scene/gaussian_model.py:378
↓ 1 callersMethoddensify_and_split
(self, grads, grad_threshold, scene_extent, N=2)
scene/gaussian_model.py:338
↓ 1 callersMethoddensify_from_depth_propagation
(self, viewpoint_cam, propagated_depth, filter_mask, gt_image)
scene/gaussian_model.py:398
↓ 1 callersFunctiondepth_to_xyz
(depth, intrinsic)
utils/graphics_utils.py:105
↓ 1 callersFunctioneval_sh
Evaluate spherical harmonics at unit directions using hardcoded SH polynomials. Works with torch/np/jnp. ... Can be 0 or more batch d
utils/sh_utils.py:57
↓ 1 callersFunctionevaluate
(model_paths)
metrics.py:25
↓ 1 callersMethodflush
(self)
utils/general_utils.py:120
↓ 1 callersFunctiongaussian
(window_size, sigma)
utils/loss_utils.py:12
↓ 1 callersFunctiongetProjectionMatrix
(znear, zfar, fovX, fovY)
utils/graphics_utils.py:61
↓ 1 callersFunctionget_center_and_diag
(cam_centers)
scene/dataset_readers.py:40
↓ 1 callersFunctionget_combined_args
(parser : ArgumentParser)
arguments/__init__.py:108
↓ 1 callersMethodget_covariance
(self, scaling_modifier = 1)
scene/gaussian_model.py:106
↓ 1 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_fina
utils/general_utils.py:24
↓ 1 callersFunctionget_network
(net_type: str)
lpipsPyTorch/modules/networks.py:12
↓ 1 callersFunctionget_state_dict
(net_type: str = 'alex', version: str = '0.1')
lpipsPyTorch/modules/utils.py:11
↓ 1 callersFunctionget_surface_normalv2
xyz: xyz coordinates patch: [p1, p2, p3, p4, p5, p6, p7, p8, p9] surface_normal = [(p9-p1) x (p3-p7)] + [(p6-p4)
utils/graphics_utils.py:114
↓ 1 callersFunctiongradient_loss
(prediction, target, mask, reduction=reduction_batch_based)
utils/loss_utils.py:120
↓ 1 callersFunctioninit_image_coord
(height, width)
utils/graphics_utils.py:89
↓ 1 callersFunctionloadCam
(args, id, cam_info, resolution_scale)
utils/camera_utils.py:10
↓ 1 callersFunctionload_pairs_relation
(path)
utils/general_utils.py:248
↓ 1 callersMethodload_ply
(self, path)
scene/gaussian_model.py:204
↓ 1 callersFunctionlpips
r"""Function that measures Learned Perceptual Image Patch Similarity (LPIPS). Arguments: x, y (torch.Tensor): the input tensors to co
lpipsPyTorch/__init__.py:6
↓ 1 callersFunctionmkdir_p
(folder_path)
utils/system_utils.py:5
↓ 1 callersFunctionmse_loss
(prediction, target, mask, reduction=reduction_batch_based)
utils/loss_utils.py:78
↓ 1 callersFunctionnormalize_activation
(x, eps=1e-10)
lpipsPyTorch/modules/utils.py:6
↓ 1 callersMethodoneupSHdegree
(self)
scene/gaussian_model.py:109
↓ 1 callersFunctionprepare_output_and_logger
(args)
train.py:363
↓ 1 callersFunctionpropagate
submodules/Propagation_SSIM/pro.cpp:13
↓ 1 callersFunctionrasterize_gaussians
( means3D, means2D, sh, colors_precomp, opacities, scales, rotations, cov3Ds_p
submodules/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:21
↓ 1 callersFunctionread
()
gaussian_renderer/network_gui.py:32
↓ 1 callersFunctionreadColmapCameras
(cam_extrinsics, cam_intrinsics, images_folder, sky_seg=False, load_normal=False, load_depth=False)
scene/dataset_readers.py:62
↓ 1 callersFunctionreadImages
(renders_dir, gt_dir)
metrics.py:13
↓ 1 callersFunctionreadNormalDmb
(file_path)
utils/general_utils.py:207
↓ 1 callersFunctionread_extrinsics_binary
see: src/base/reconstruction.cc void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBi
scene/colmap_loader.py:169
↓ 1 callersFunctionread_extrinsics_text
Taken from https://github.com/colmap/colmap/blob/dev/scripts/python/read_write_model.py
scene/colmap_loader.py:233
↓ 1 callersFunctionread_intrinsics_binary
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasBinary(const std::string& path) void Reconstruction::ReadCameras
scene/colmap_loader.py:204
↓ 1 callersFunctionread_intrinsics_text
Taken from https://github.com/colmap/colmap/blob/dev/scripts/python/read_write_model.py
scene/colmap_loader.py:145
↓ 1 callersFunctionread_points3D_binary
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints
scene/colmap_loader.py:114
↓ 1 callersFunctionread_points3D_text
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3D
scene/colmap_loader.py:72
↓ 1 callersFunctionrender_sets
(dataset : ModelParams, iteration : int, pipeline : PipelineParams, skip_train : bool, skip_test : bool)
render.py:51
↓ 1 callersFunctionrender_with_mask
(viewpoint_camera, pc: GaussianModel, pipe, bg_color: torch.Tensor, person_mask: torch.Tensor,
gaussian_renderer/__init__.py:149
↓ 1 callersMethodreplace_tensor_to_optimizer
(self, tensor, name)
scene/gaussian_model.py:247
↓ 1 callersFunctionreproject_with_depth
(depth_ref, intrinsics_ref, extrinsics_ref, depth_src, intrinsics_src, extrinsics_src)
utils/graphics_utils.py:320
next →1–100 of 207, ranked by callers