MCPcopy Create free account

hub / github.com/NVlabs/InstantSplat / functions

Functions864 in github.com/NVlabs/InstantSplat

↓ 3 callersFunctionestimate_focal_knowing_depth
Reprojection method, for when the absolute depth is known: 1) estimate the camera focal using a robust estimator 2) reproject points
dust3r/post_process.py:12
↓ 3 callersMethodflush
(self)
utils/general_utils.py:125
↓ 3 callersMethodget_all_pts3d
(self, gt1, gt2, pred1, pred2, dist_clip=None)
mast3r/losses.py:43
↓ 3 callersMethodget_all_pts3d
(self, gt1, gt2, pred1, pred2, dist_clip=None)
dust3r/losses.py:158
↓ 3 callersMethodget_focals
(self)
dust3r/cloud_opt/pair_viewer.py:95
↓ 3 callersMethodget_im_poses
(self)
dust3r/cloud_opt/pair_viewer.py:114
↓ 3 callersMethodget_intrinsics
(self)
dust3r/cloud_opt/pair_viewer.py:104
↓ 3 callersMethodget_pw_poses
(self)
dust3r/cloud_opt/base_opt.py:191
↓ 3 callersFunctionget_tensor_from_camera
Convert transformation matrix to quaternion and translation.
utils/pose_utils.py:183
↓ 3 callersFunctionglobal_aligner
(dust3r_output, device, mode=GlobalAlignerMode.PointCloudOptimizer, **optim_kw)
dust3r/cloud_opt/__init__.py:20
↓ 3 callersFunctioninference
(pairs, model, device, batch_size=8, verbose=True)
dust3r/inference.py:56
↓ 3 callersFunctioninterp
Runs multidimensional B-spline interpolation on the input points.
utils/pose_utils.py:519
↓ 3 callersFunctionloss_of_one_batch
(batch, model, criterion, device, symmetrize_batch=False, use_amp=False, ret=None)
dust3r/inference.py:32
↓ 3 callersFunctionmake_K_cam_depth
(log_focals, pps, trans, quats, log_sizes, core_depth)
mast3r/cloud_opt/sparse_ga.py:231
↓ 3 callersFunctionmake_pts3d
(anchors, K, cam2w, depthmaps, base_focals=None, ret_depth=False)
mast3r/cloud_opt/sparse_ga.py:475
↓ 3 callersFunctionmerge_corres
(idx1, idx2, shape1=None, shape2=None, ret_xy=True, ret_index=False)
mast3r/fast_nn.py:87
↓ 3 callersFunctionnormalize
Normalization helper function.
utils/camera_utils.py:193
↓ 3 callersFunctionnormalize
(x)
utils/pose_utils.py:217
↓ 3 callersFunctionpad_poses
Pad [..., 3, 4] pose matrices with a homogeneous bottom row [0,0,0,1].
utils/pose_utils.py:248
↓ 3 callersFunctionpoints_to_poses
Converts from (position, lookat, up) format to pose matrices.
utils/pose_utils.py:473
↓ 3 callersFunctionqvec2rotmat
(qvec)
scene/colmap_loader.py:43
↓ 3 callersFunctionreg_dense_conf
extract confidence from prediction head output
dust3r/heads/postprocess.py:49
↓ 3 callersFunctionrigid_points_registration
(pts1, pts2, conf)
dust3r/cloud_opt/init_im_poses.py:232
↓ 3 callersFunctionrotate_90
(views, k=1)
mast3r/datasets/base/mast3r_base_stereo_view_dataset.py:315
↓ 3 callersFunctionsave_extrinsic
(sparse_path, extrinsics_w2c, img_files, image_suffix)
utils/sfm_utils.py:202
↓ 3 callersFunctionsave_intrinsics
(sparse_path, focals, org_imgs_shape, imgs_shape, save_focals=False)
utils/sfm_utils.py:227
↓ 3 callersFunctionsave_model
(epoch, fname, best_so_far)
dust3r/training.py:177
↓ 3 callersFunctionsel
(x)
dust3r/datasets/base/base_stereo_view_dataset.py:195
↓ 3 callersMethodset_requires_grad
(self, state: bool)
lpipsPyTorch/modules/networks.py:46
↓ 3 callersFunctionsplit_train_test
(image_files, llffhold=8, n_views=None, verbose=True)
utils/sfm_utils.py:53
↓ 3 callersMethodstate_dict
(self, trainable=True)
dust3r/cloud_opt/base_opt.py:123
↓ 3 callersMethodto_pil
(self)
dust3r/datasets/utils/cropping.py:37
↓ 2 callersFunctionSH2RGB
(sh)
utils/sh_utils.py:117
↓ 2 callersFunctionSum
(*losses_and_masks)
dust3r/losses.py:16
↓ 2 callersMethod_decoder
(self, f1, pos1, f2, pos2)
dust3r/model.py:171
↓ 2 callersMethod_dispatch
(self, func, *args, **kwargs)
dust3r/datasets/utils/cropping.py:52
↓ 2 callersFunction_dotmv
(Trf, pts, ncol=None, norm=False)
mast3r/datasets/utils/cropping.py:119
↓ 2 callersFunction_filter_edges_seq
(edges, seq_dis_thr, cyclic=False)
dust3r/image_pairs.py:80
↓ 2 callersFunction_interleave_imgs
(img1, img2)
dust3r/inference.py:14
↓ 2 callersFunction_make_overlapping_grid
(H, W, size, overlap)
mast3r/utils/coarse_to_fine.py:33
↓ 2 callersFunction_ravel_hw
(tensor, fill=0)
dust3r/cloud_opt/optimizer.py:231
↓ 2 callersMethod_refine_depths_with_TSDF
Leverage TSDF to post-process estimated depths for each pixel, find zero level of TSDF along ray (or closest to 0)
mast3r/cloud_opt/tsdf_optimizer.py:34
↓ 2 callersFunction_rotation_origin_to_pt
Align the origin (0,0,1) with the target point (x,y,1) in projective space. Method: rotate z to put target on (x'+,0,1), then rotate on Y to get
mast3r/datasets/utils/cropping.py:107
↓ 2 callersFunction_score_cell
(cell1, H2, W2, p1, p2, min_corres=10, forced_resolution=None)
mast3r/utils/coarse_to_fine.py:111
↓ 2 callersMethod_set_depthmap
(self, idx, depth, force=False)
dust3r/cloud_opt/pair_viewer.py:83
↓ 2 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average=True)
utils/loss_utils.py:65
↓ 2 callersFunction_start_pos
(total_size, win_size, overlap)
mast3r/utils/coarse_to_fine.py:18
↓ 2 callersFunction_weight_pixels
(cell, pix, assigned, gauss_var=2)
mast3r/utils/coarse_to_fine.py:91
↓ 2 callersFunctionadjust_learning_rate_by_lr
(optimizer, lr)
dust3r/optim_factory.py:9
↓ 2 callersFunctionalignTrajectory
calculate s, R, t so that: gt = R * s * est + t method can be: sim3, se3, posyaw, none; n_aligned: -1 means using all the frames
utils/utils_poses/ATE/align_utils.py:120
↓ 2 callersFunctionalign_multiple_poses
(src_poses, target_poses)
dust3r/cloud_opt/init_im_poses.py:320
↓ 2 callersFunctionarcball_map_to_sphere
Return unit sphere coordinates from window coordinates.
utils/utils_poses/ATE/transformations.py:1740
↓ 2 callersFunctionbruteforce_reciprocal_nns
(A, B, device='cuda', block_size=None, dist='l2')
mast3r/fast_nn.py:17
↓ 2 callersFunctionbuild_dataset
(dataset, batch_size, num_workers, test=False)
dust3r/training.py:253
↓ 2 callersFunctionbuild_rotation
(r)
utils/general_utils.py:78
↓ 2 callersFunctioncameraList_from_camInfos
(cam_infos, resolution_scale, args)
utils/camera_utils.py:57
↓ 2 callersFunctioncat_3d
(vecs)
dust3r/viz.py:23
↓ 2 callersFunctioncat_meshes
(meshes)
dust3r/viz.py:78
↓ 2 callersFunctioncenter_and_z
(poses)
dust3r/cloud_opt/init_im_poses.py:324
↓ 2 callersFunctioncollate_with_cat
(whatever, lists=False)
dust3r/utils/device.py:47
↓ 2 callersFunctionconvert_im_matches_pairs
(img0, img1, image_to_colmap, im_keypoints, matches_im0, matches_im1, viz)
mast3r/colmap/database.py:25
↓ 2 callersFunctioncreate_window
(window_size, channel)
utils/loss_utils.py:49
↓ 2 callersFunctioncrop_tag
(cell)
mast3r/utils/coarse_to_fine.py:10
↓ 2 callersFunctioncumulative_sum
(input_list)
utils/graphics_utils.py:99
↓ 2 callersFunctiondecoder
(feat1, feat2, pos1, pos2, shape1, shape2)
mast3r/cloud_opt/sparse_ga.py:580
↓ 2 callersMethoddensification_postfix
(self, new_xyz, new_features_dc, new_features_rest, new_opacities, new_scaling, new_rotation)
scene/gaussian_model.py:400
↓ 2 callersFunctiondepthmap_to_camera_coordinates
Args: - depthmap (HxW array): - camera_intrinsics: a 3x3 matrix Returns: pointmap of absolute coordinates (HxWx3 arra
dust3r/utils/geometry.py:165
↓ 2 callersFunctiondist_func
(A, B)
mast3r/fast_nn.py:30
↓ 2 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
↓ 2 callersFunctionextract_correspondences_from_pts3d
(view1, view2, target_n_corres, rng=np.random, ret_xy=True, nneg=0)
mast3r/datasets/utils/cropping.py:22
↓ 2 callersFunctionextract_correspondences_nonsym
(A, B, confA, confB, subsample=8, device=None, ptmap_key='pred_desc', pixel_tol=0)
mast3r/fast_nn.py:191
↓ 2 callersFunctionfast_pnp
(pts3d, focal, msk, device, pp=None, niter_PnP=10)
dust3r/cloud_opt/init_im_poses.py:259
↓ 2 callersFunctionfetchPly
(path)
scene/dataset_readers.py:214
↓ 2 callersFunctionfilter_pairs_seq
(pairs, seq_dis_thr, cyclic=False)
dust3r/image_pairs.py:94
↓ 2 callersMethodfrom_pretrained
(cls, pretrained_model_name_or_path, **kw)
mast3r/model.py:48
↓ 2 callersFunctiongenerate_interpolated_path
Creates a smooth spline path between input keyframe camera poses. Spline is calculated with poses in format (position, lookat-point, up-point).
utils/camera_utils.py:127
↓ 2 callersFunctiongetNerfppNorm
(cam_info)
scene/dataset_readers.py:50
↓ 2 callersMethodgetTestCameras
(self, scale=1.0)
scene/__init__.py:105
↓ 2 callersMethodget_RT
(self, idx)
scene/gaussian_model.py:134
↓ 2 callersMethodget_adaptors
(self)
dust3r/cloud_opt/base_opt.py:143
↓ 2 callersFunctionget_camera_from_tensor
Convert quaternion and translation to transformation matrix.
utils/pose_utils.py:57
↓ 2 callersFunctionget_conf_trf
(mode)
dust3r/cloud_opt/commons.py:48
↓ 2 callersMethodget_covariance
(self, scaling_modifier = 1)
scene/gaussian_model.py:123
↓ 2 callersMethodget_dense_pts3d
(self, clean_depth=True, subsample=8)
mast3r/cloud_opt/sparse_ga.py:70
↓ 2 callersMethodget_dense_pts3d
(self, clean_depth=True)
mast3r/cloud_opt/tsdf_optimizer.py:269
↓ 2 callersMethodget_focals
(self)
dust3r/cloud_opt/optimizer.py:127
↓ 2 callersMethodget_focals
(self)
dust3r/cloud_opt/modular_optimizer.py:94
↓ 2 callersMethodget_im_poses
(self)
dust3r/cloud_opt/base_opt.py:225
↓ 2 callersFunctionget_known_poses
(self)
dust3r/cloud_opt/init_im_poses.py:302
↓ 2 callersFunctionget_med_dist_between_poses
(poses)
dust3r/utils/geometry.py:364
↓ 2 callersMethodget_name
(self)
dust3r/losses.py:97
↓ 2 callersFunctionget_positions
(theta)
utils/pose_utils.py:329
↓ 2 callersMethodget_principal_points
(self)
dust3r/cloud_opt/optimizer.py:141
↓ 2 callersMethodget_principal_points
(self)
dust3r/cloud_opt/pair_viewer.py:101
↓ 2 callersMethodget_principal_points
(self)
dust3r/cloud_opt/modular_optimizer.py:105
↓ 2 callersMethodget_pw_norm_scale_factor
(self)
dust3r/cloud_opt/base_opt.py:178
↓ 2 callersFunctionget_similarities
(desc1, desc2, euc=False)
mast3r/losses.py:202
↓ 2 callersFunctionget_sorted_image_files
Get sorted image files from the given directory. Args: image_dir (str): Path to the directory containing images. Returns:
utils/sfm_utils.py:71
↓ 2 callersFunctionglobal_alignment_iter
(net, cur_iter, niter, lr_base, lr_min, optimizer, schedule)
dust3r/cloud_opt/base_opt.py:352
↓ 2 callersFunctionhead_factory
build a prediction head for the decoder
dust3r/heads/__init__.py:11
← previousnext →101–200 of 864, ranked by callers