MCPcopy Create free account

hub / github.com/NVlabs/InstantSplat / functions

Functions864 in github.com/NVlabs/InstantSplat

↓ 1 callersMethodloss
(self, a, b, valid_matches=None)
mast3r/losses.py:233
↓ 1 callersFunctionloss_dust3r
(cam2w, pts3d, pix_loss)
mast3r/cloud_opt/sparse_ga.py:333
↓ 1 callersFunctionloss_func
(x, y)
mast3r/cloud_opt/utils/losses.py:26
↓ 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 callersFunctionmain
(source_path, model_path, ckpt_path, device, batch_size, image_size, schedule, lr, niter, min_conf_t
init_geo.py:24
↓ 1 callersFunctionmain
(source_path, model_path, ckpt_path, device, batch_size, image_size, schedule, lr, niter, min_conf_t
init_test_pose.py:24
↓ 1 callersFunctionmake_batch_symmetric
(batch)
dust3r/inference.py:26
↓ 1 callersMethodmake_sampler
(self, batch_size, shuffle=True, world_size=1, rank=0, drop_last=True)
dust3r/datasets/base/easy_dataset.py:34
↓ 1 callersFunctionmask110
(device, dtype)
mast3r/cloud_opt/sparse_ga.py:465
↓ 1 callersMethodmask_sky
(self)
dust3r/cloud_opt/base_opt.py:290
↓ 1 callersFunctionmatching_check
(x)
mast3r/cloud_opt/sparse_ga.py:305
↓ 1 callersFunctionmax_dilate
Dilate (via max-pooling) a non-negative step function.
utils/stepfun.py:75
↓ 1 callersFunctionminimum_spanning_tree
(imshapes, edges, pred_i, pred_j, conf_i, conf_j, im_conf, min_conf_thr, device, has
dust3r/cloud_opt/init_im_poses.py:135
↓ 1 callersFunctionmkdir_p
(folder_path)
utils/system_utils.py:16
↓ 1 callersFunctionnormalize_activation
(x, eps=1e-10)
lpipsPyTorch/modules/utils.py:6
↓ 1 callersMethodoneupSHdegree
(self)
scene/gaussian_model.py:142
↓ 1 callersFunctionopencv_to_colmap_intrinsics
Modify camera intrinsics to follow a different convention. Coordinates of the center of the top-left pixels are by default: - (0.5, 0.5)
dust3r/utils/geometry.py:236
↓ 1 callersFunctionpad_poses
Pad [..., 3, 4] pose matrices with a homogeneous bottom row [0,0,0,1].
utils/camera_utils.py:197
↓ 1 callersFunctionparallel_threads
tqdm but with parallel execution. Will essentially return res = [ function(arg) # default function(*arg) # if star_args is
dust3r/utils/parallel.py:12
↓ 1 callersFunctionpdf_to_weight
Turn a PDF that integrates to 1 into a vector of weights that sums to 1.
utils/stepfun.py:70
↓ 1 callersFunctionpixel_grid
(H, W)
dust3r/cloud_opt/init_im_poses.py:255
↓ 1 callersFunctionplot_pose
(ref_poses, est_poses, output_path, args, vid=False)
utils/utils_poses/vis_pose_utils.py:177
↓ 1 callersFunctionpoints_to_poses
Converts from (position, lookat, up) format to pose matrices.
utils/camera_utils.py:151
↓ 1 callersFunctionpos2d_in_rect
(p1, cell1)
mast3r/utils/coarse_to_fine.py:104
↓ 1 callersFunctionposes_avg
(poses)
utils/utils_poses/vis_pose_utils.py:48
↓ 1 callersFunctionposes_to_points
Converts from pose matrices to (position, lookat, up) format.
utils/camera_utils.py:144
↓ 1 callersFunctionposes_to_points
Converts from pose matrices to (position, lookat, up) format.
utils/pose_utils.py:466
↓ 1 callersFunctionpostprocess
extract 3D points/confidence from prediction head output
dust3r/heads/postprocess.py:10
↓ 1 callersFunctionprepare_canonical_data
(imgs, tmp_pairs, subsample, order_imgs=False, min_conf_thr=0, cache_path=None, dev
mast3r/cloud_opt/sparse_ga.py:634
↓ 1 callersFunctionprepare_output_and_logger
(args)
train.py:234
↓ 1 callersMethodpreset_focal
(self, known_focals, msk=None)
dust3r/cloud_opt/modular_optimizer.py:58
↓ 1 callersMethodpreset_principal_point
(self, known_pp, msk=None)
dust3r/cloud_opt/modular_optimizer.py:64
↓ 1 callersFunctionproj3d
(inv_K, pixels, z)
mast3r/cloud_opt/sparse_ga.py:469
↓ 1 callersFunctionproject_points
(points_3d, intrinsics, extrinsics)
utils/sfm_utils.py:418
↓ 1 callersFunctionquad2rotation
Convert quaternion to rotation in batch. Since all operation in pytorch, support gradient passing. Args: quad (tensor, batch_size*4)
utils/pose_utils.py:10
↓ 1 callersFunctionquadmultiply
Multiply two quaternions together using quaternion arithmetic
utils/pose_utils.py:86
↓ 1 callersFunctionquaternion_conjugate
Return conjugate of quaternion. >>> q0 = random_quaternion() >>> q1 = quaternion_conjugate(q0) >>> q1[3] == q0[3] and all(q1[:3] == -q0[:
utils/utils_poses/ATE/transformations.py:1500
↓ 1 callersFunctionquaternion_from_matrix
Return quaternion from rotation matrix. >>> R = rotation_matrix(0.123, (1, 2, 3)) >>> q = quaternion_from_matrix(R) >>> numpy.allclose(q,
utils/utils_poses/ATE/transformations.py:1451
↓ 1 callersFunctionquaternion_multiply
Return multiplication of two quaternions. >>> q = quaternion_multiply([1, -2, 3, 4], [-5, 6, 7, 8]) >>> numpy.allclose(q, [-44, -14, 48, 28])
utils/utils_poses/ATE/transformations.py:1483
↓ 1 callersFunctionquaternion_slerp
Return spherical linear interpolation between two quaternions. >>> q0 = random_quaternion() >>> q1 = random_quaternion() >>> q = quaterni
utils/utils_poses/ATE/transformations.py:1525
↓ 1 callersFunctionrandom_quaternion
Return uniform random unit quaternion. rand: array like or None Three independent random variables that are uniformly distributed
utils/utils_poses/ATE/transformations.py:1566
↓ 1 callersFunctionravel_xy
(pos, shape)
mast3r/datasets/utils/cropping.py:94
↓ 1 callersFunctionread
()
gaussian_renderer/network_gui.py:43
↓ 1 callersFunctionreadColmapCameras
(cam_extrinsics, cam_intrinsics, images_folder)
scene/dataset_readers.py:106
↓ 1 callersFunctionreadImages
(renders_dir, gt_dir)
utils/sfm_utils.py:452
↓ 1 callersFunctionread_colmap_gt_pose
(gt_pose_path, llffhold=8)
utils/sfm_utils.py:434
↓ 1 callersFunctionread_extrinsics_binary
see: src/base/reconstruction.cc void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBi
scene/colmap_loader.py:184
↓ 1 callersFunctionread_extrinsics_text
Taken from https://github.com/colmap/colmap/blob/dev/scripts/python/read_write_model.py
scene/colmap_loader.py:248
↓ 1 callersFunctionread_intrinsics_text
Taken from https://github.com/colmap/colmap/blob/dev/scripts/python/read_write_model.py
scene/colmap_loader.py:159
↓ 1 callersFunctionread_points3D_binary
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints
scene/colmap_loader.py:127
↓ 1 callersFunctionread_points3D_text
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3D
scene/colmap_loader.py:85
↓ 1 callersFunctionrecenter_poses
Recenter poses around the origin.
utils/pose_utils.py:295
↓ 1 callersFunctionreciprocal_1d
(corres_1_to_2, corres_2_to_1, ret_recip=False)
mast3r/datasets/utils/cropping.py:13
↓ 1 callersFunctionreg_desc
(desc, mode)
mast3r/catmlp_dpt_head.py:17
↓ 1 callersFunctionremove_buffer_poses
(poses, u, n_frames, u_keyframes, n_buffer)
utils/pose_utils.py:511
↓ 1 callersFunctionrender_path_spiral
(c2w, up, rads, focal, zdelta, zrate, rots, N)
utils/utils_poses/vis_pose_utils.py:74
↓ 1 callersFunctionrender_set_optimize
(model_path, name, iteration, views, gaussians, pipeline, background)
render.py:99
↓ 1 callersFunctionrender_sets
( dataset: ModelParams, iteration: int, pipeline: PipelineParams, skip_train: bool, skip_t
render.py:189
↓ 1 callersMethodreplace_tensor_to_optimizer
(self, tensor, name)
scene/gaussian_model.py:328
↓ 1 callersFunctionreproj2d
(Trf, pts3d)
mast3r/cloud_opt/sparse_ga.py:977
↓ 1 callersFunctionreproject
(pts3d, K, world2cam, shape)
mast3r/datasets/utils/cropping.py:82
↓ 1 callersMethodrestore
(self, model_args, training_args)
scene/gaussian_model.py:82
↓ 1 callersFunctionrigid_points_registration
(pts1, pts2, conf=None)
utils/sfm_utils.py:101
↓ 1 callersFunctionrot_z
(theta)
utils/utils_poses/ATE/align_trajectory.py:30
↓ 1 callersFunctionrotation2quad
Convert rotations given as rotation matrices to quaternions. Args: matrix: Rotation matrices as tensor of shape (..., 3, 3). Re
utils/pose_utils.py:117
↓ 1 callersFunctionrotation_error
Compute rotation error Args: pose_error (4x4 array): relative pose error Returns: rot_error (float): rotation error
utils/utils_poses/comp_ate.py:13
↓ 1 callersFunctionrotmat2qvec
(R)
scene/colmap_loader.py:55
↓ 1 callersFunctionround_by
(total, multiple, up=False)
dust3r/datasets/base/batched_sampler.py:71
↓ 1 callersFunctionsafe_state
(silent)
utils/general_utils.py:112
↓ 1 callersFunctionsample_np
numpy version of sample()
utils/stepfun.py:221
↓ 1 callersFunctionsave_final_model
(args, epoch, model_without_ddp, best_so_far=None)
dust3r/training.py:239
↓ 1 callersFunctionsave_images_and_masks
(sparse_0_path, n_views, imgs, overlapping_masks, image_files, image_suffix)
utils/sfm_utils.py:319
↓ 1 callersFunctionsave_interpolate_pose
(model_path, iter, n_views)
render.py:36
↓ 1 callersMethodsave_ply
(self, path)
scene/gaussian_model.py:261
↓ 1 callersFunctionsave_points3D
(sparse_path, imgs, pts3d, confs, masks=None, use_masks=True, save_all_pts=False, save_txt_path=None, depth_th
utils/sfm_utils.py:250
↓ 1 callersFunctionscipy_bspline
Calculate n samples on a bspline cv : Array ov control vertices n : Number of samples to return degree: Curve d
utils/utils_poses/vis_pose_utils.py:89
↓ 1 callersFunctionsearchForMaxIteration
(folder)
utils/system_utils.py:26
↓ 1 callersFunctionsegment_sky
(image)
dust3r/viz.py:345
↓ 1 callersMethodset_downstream_head
(self, output_mode, head_type, landscape_only, depth_mode, conf_mode, patch_size, img_size,
dust3r/model.py:112
↓ 1 callersMethodset_freeze
(self, freeze)
dust3r/model.py:99
↓ 1 callersMethodsetup_functions
(self)
scene/gaussian_model.py:31
↓ 1 callersFunctionshow_raw_pointcloud
(pts3d, colors, point_size=2)
dust3r/viz.py:29
↓ 1 callersFunctionshow_reconstruction
(shapes_or_imgs, K, cam2w, pts3d, gt_cam2w=None, gt_K=None, cam_size=None, masks=None, **kw)
mast3r/cloud_opt/sparse_ga.py:1012
↓ 1 callersFunctionsigned_expm1
(x)
dust3r/cloud_opt/commons.py:78
↓ 1 callersFunctionsigned_log1p
(x)
dust3r/cloud_opt/commons.py:73
↓ 1 callersFunctionsim_func
(p1, p2, gamma)
mast3r/cloud_opt/sparse_ga.py:904
↓ 1 callersFunctionskew
Returns the skew-symmetric matrix of a vector cfo, 2015/08/13
utils/utils_poses/ATE/transformations.py:180
↓ 1 callersFunctionsparse_global_alignment
Sparse alignment with MASt3R imgs: list of image paths cache_path: path where to dump temporary files (str) lr1, niter1: lea
mast3r/cloud_opt/sparse_ga.py:118
↓ 1 callersFunctionsparse_scene_optimizer
(imgs, subsample, imsizes, pps, base_focals, core_depth, anchors, corres, corres2d,
mast3r/cloud_opt/sparse_ga.py:158
↓ 1 callersFunctionspectral_clustering
(graph, k=None, normalized_cuts=False)
mast3r/cloud_opt/sparse_ga.py:889
↓ 1 callersFunctionspectral_projection_depth
(K, depthmap, subsample, k=64, cache_path='', normalized_cuts=True, gamma=7, min
mast3r/cloud_opt/sparse_ga.py:919
↓ 1 callersFunctionspectral_projection_of_depthmaps
(imgs, intrinsics, depthmaps, subsample, cache_path=None, **kw)
mast3r/cloud_opt/sparse_ga.py:962
↓ 1 callersFunctionstorePly
(path, xyz, rgb)
utils/sfm_utils.py:495
↓ 1 callersFunctionstrip_lowerdiag
(L)
utils/general_utils.py:64
↓ 1 callersFunctionstrip_symmetric
(sym)
utils/general_utils.py:75
↓ 1 callersFunctionsymmetric_inference
(model, img1, img2, device)
mast3r/cloud_opt/sparse_ga.py:571
↓ 1 callersFunctiontest_one_epoch
(model: torch.nn.Module, criterion: torch.nn.Module, data_loader: Sized, device: torch.devi
dust3r/training.py:342
↓ 1 callersFunctiontrain_one_epoch
(model: torch.nn.Module, criterion: torch.nn.Module, data_loader: Sized, optimizer: torch.
dust3r/training.py:267
↓ 1 callersFunctiontraining
(dataset, opt, pipe, testing_iterations, saving_iterations, checkpoint_iterations, checkpoint, debug_from)
train.py:88
↓ 1 callersFunctiontraining_report
(tb_writer, iteration, Ll1, loss, l1_loss, elapsed, testing_iterations, scene : Scene, renderFunc, renderArgs)
train.py:256
← previousnext →401–500 of 864, ranked by callers